tcr 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 9eb63ae66e63fd83d042f819294f1490e4194e52
-  data.tar.gz: b63f2673ed97cb1e59a245c9fa6931abf8abd06a
+  metadata.gz: 565fc7b37f7758c3b70c3edba26a8dace04d4ced
+  data.tar.gz: 6ddd25facb69f62f6c4756b6d18d0d133dea3dd6
 SHA512:
-  metadata.gz: 6a672d5b5d03361c7cbbd491814b313905fee72905ccee3ee7a811210ada82d7ae92bdb3f66951c5417db4fca9553b0bc237f69401b76e4272c22462cfdc22e9
-  data.tar.gz: 8bb20e98750081a6faa8fb275181dff36c073dcf4253c42a41cc8a9a4f61c0f6f10330e32a507ba81301fc85285c037258b512a92a2c28d1b99bdd13f6479ee0
+  metadata.gz: cd8fec962f5fc58784118afd0801c60789b47aef67953f34bf7a9c42c5ffc1614de7633f11a59353ced13ca18347315a1cf7eac081e11cc862117fe27d8a2581
+  data.tar.gz: fa8cd68d5104df3fa8aa7490c6f3f52ebae1440efd39390597989c5ead538bf6203d9da1d605696d8d3530b50c77d07017d7912e9e6ebf44712cab4492639ac2

data/lib/tcr/cassette.rb

@@ -8,7 +8,7 @@ module TCR
       if File.exists?(filename)
         @recording = false
         @contents = File.open(filename) { |f| f.read }
-        @sessions = JSON.parse(@contents)
+        @sessions = unmarshal(@contents)
       else
         @recording = true
         @sessions = []
@@ -31,14 +31,36 @@ module TCR
 
     def save
       if recording?
-        File.open(filename, "w") { |f| f.write(JSON.pretty_generate(@sessions)) }
+        File.open(filename, "w") { |f| f.write(marshal(@sessions)) }
       end
     end
 
     protected
 
+    def unmarshal(content)
+      case TCR.configuration.format
+      when "json"
+        JSON.parse(content)
+      when "yaml"
+        YAML.load(content)
+      else
+        raise RuntimeError.new "unrecognized format #{TCR.configuration.format}, please use `json` or `yaml`"
+      end
+    end
+
+    def marshal(content)
+      case TCR.configuration.format
+      when "json"
+        JSON.pretty_generate(content)
+      when "yaml"
+        YAML.dump(content)
+      else
+        raise RuntimeError.new "unrecognized format #{TCR.configuration.format}, please use `json` or `yaml`"
+      end
+    end
+
     def filename
-      "#{TCR.configuration.cassette_library_dir}/#{name}.json"
+      "#{TCR.configuration.cassette_library_dir}/#{name}.#{TCR.configuration.format}"
     end
   end
 end

data/lib/tcr/configuration.rb

@@ -1,6 +1,6 @@
 module TCR
   class Configuration
-    attr_accessor :cassette_library_dir, :hook_tcp_ports, :block_for_reads
+    attr_accessor :cassette_library_dir, :hook_tcp_ports, :block_for_reads, :format
 
     def initialize
       reset_defaults!
@@ -10,6 +10,7 @@ module TCR
       @cassette_library_dir = "fixtures/tcr_cassettes"
       @hook_tcp_ports = []
       @block_for_reads = false
+      @format = "json"
     end
   end
 end

data/lib/tcr/version.rb

@@ -1,3 +1,3 @@
 module TCR
-  VERSION = "0.1.1"
+  VERSION = "0.1.2"
 end

data/spec/tcr_spec.rb

@@ -15,8 +15,10 @@ describe TCR do
 
   around(:each) do |example|
     File.unlink("test.json") if File.exists?("test.json")
+    File.unlink("test.yaml") if File.exists?("test.yaml")
     example.run
     File.unlink("test.json") if File.exists?("test.json")
+    File.unlink("test.yaml") if File.exists?("test.yaml")
   end
 
   describe ".configuration" do
@@ -157,6 +159,16 @@ describe TCR do
       }.to change{ File.exists?("./test.json") }.from(false).to(true)
     end
 
+    it "creates a cassette file on use with yaml" do
+      TCR.configure { |c| c.format = "yaml" }
+
+      expect {
+        TCR.use_cassette("test") do
+          tcp_socket = TCPSocket.open("smtp.mandrillapp.com", 2525)
+        end
+      }.to change{ File.exists?("./test.yaml") }.from(false).to(true)
+    end
+
     it "records the tcp session data into the file" do
       TCR.use_cassette("test") do
         tcp_socket = TCPSocket.open("smtp.mandrillapp.com", 2525)
@@ -168,6 +180,20 @@ describe TCR do
       cassette_contents.include?("220 smtp.mandrillapp.com ESMTP").should == true
     end
 
+    it "records the tcp session data into the yaml file" do
+      TCR.configure { |c| c.format = "yaml" }
+
+      TCR.use_cassette("test") do
+        tcp_socket = TCPSocket.open("smtp.mandrillapp.com", 2525)
+        io = Net::InternetMessageIO.new(tcp_socket)
+        line = io.readline
+        tcp_socket.close
+      end
+      cassette_contents = File.open("test.yaml") { |f| f.read }
+      cassette_contents.include?("---").should == true
+      cassette_contents.include?("220 smtp.mandrillapp.com ESMTP").should == true
+    end
+
     it "plays back tcp sessions without opening a real connection" do
       expect(TCPSocket).to_not receive(:real_open)
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: tcr
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - Rob Forman
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2017-02-02 00:00:00.000000000 Z
+date: 2017-03-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec