tcr 0.0.4 → 0.0.5

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: afe506ab1f3f57f65d4b9131d0b90a19f3392e00
+  data.tar.gz: a1b9bab2f8f7692c7be4a641c2d10c06743fccd8
+SHA512:
+  metadata.gz: e6107c9636806831a366f43607651f6613e557026fb9a520cc630d20f327abfc935ded60fa0d410492b018b123a2d83e48843b1d004b6ab82e962b9f9f84725c
+  data.tar.gz: ccdc0808bed751308838b9f992ad00fecaab8917bef9cc3bc29e2eabeed9b49b759d7fa33f4c0b9bd4be31b5c299f3c304110b883825c184b4ec138e6785c4ec

data/.travis.yml

@@ -0,0 +1,6 @@
+language: ruby
+rvm:
+  - "1.9.3"
+  - "2.0.0"
+  - "2.1.2"
+script: bundle exec rspec

data/README.md

@@ -1,5 +1,12 @@
 # TCR (TCP + VCR)
 
+[![Build Status](https://travis-ci.org/robforman/tcr.png?branch=master)](https://travis-ci.org/robforman/tcr)
+
+[ ![Codeship Status for SalesLoft/melody](https://www.codeship.io/projects/9fcbda40-6859-0132-9920-3ad5c353d440/status?branch=master)](https://www.codeship.io/projects/53337)
+
+
+
+
 TCR is a *very* lightweight version of [VCR](https://github.com/vcr/vcr) for TCP sockets.
 
 Currently used for recording 'net/smtp' interactions so only a few of the TCPSocket methods are recorded out.
@@ -55,6 +62,14 @@ Run this test once, and TCR will record the tcp interactions to fixtures/tcr_cas
 
 Run it again, and TCR will replay the interactions from json when the tcp request is made. This test is now fast (no real TCP requests are made anymore), deterministic and accurate.
 
+You can disable TCR hooking TCPSocket ports for a given block via `turned_off`:
+
+```ruby
+TCR.turned_off do
+  tcp_socket = TCPSocket.open("aspmx.l.google.com", 25)
+end
+```
+
 ## Contributing
 
 1. Fork it

data/lib/tcr.rb

@@ -41,6 +41,7 @@ module TCR
     raise ArgumentError, "`TCR.use_cassette` requires a block." unless block
     TCR.cassette = Cassette.new(name)
     yield
+    TCR.cassette.save
     TCR.cassette = nil
   end
 
@@ -59,7 +60,7 @@ class TCPSocket
   class << self
     alias_method :real_open,  :open
 
-    def open(address, port)
+    def open(address, port, *args)
       if TCR.configuration.hook_tcp_ports.include?(port)
         TCR::RecordableTCPSocket.new(address, port, TCR.cassette)
       else
@@ -68,3 +69,15 @@ class TCPSocket
     end
   end
 end
+
+class OpenSSL::SSL::SSLSocket
+  class << self
+    def new(io, *args)
+      if TCR::RecordableTCPSocket === io
+        TCR::RecordableSSLSocket.new(io)
+      else
+        super
+      end
+    end
+  end
+end

data/lib/tcr/cassette.rb

@@ -20,15 +20,19 @@ module TCR
     end
 
     def next_session
-      session = @sessions.shift
-      raise NoMoreSessionsError unless session
-      session
+      if recording?
+        @sessions << []
+        @sessions.last
+      else
+        raise NoMoreSessionsError if @sessions.empty?
+        @sessions.shift
+      end
     end
 
-    def append(session)
-      raise "Can't append session unless recording" unless recording?
-      @sessions << session
-      File.open(filename, "w") { |f| f.write(JSON.pretty_generate(@sessions)) }
+    def save
+      if recording?
+        File.open(filename, "w") { |f| f.write(JSON.pretty_generate(@sessions)) }
+      end
     end
 
     protected
@@ -37,4 +41,4 @@ module TCR
       "#{TCR.configuration.cassette_library_dir}/#{name}.json"
     end
   end
-end
+end

data/lib/tcr/configuration.rb

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

data/lib/tcr/recordable_tcp_socket.rb

@@ -1,45 +1,46 @@
+require 'delegate'
+require 'openssl'
+require 'thread'
+
+
 module TCR
   class RecordableTCPSocket
-    attr_reader :live, :cassette
+    attr_reader :live, :socket
     attr_accessor :recording
 
     def initialize(address, port, cassette)
       raise TCR::NoCassetteError.new unless TCR.cassette
 
+      @read_lock = Queue.new
+
       if cassette.recording?
         @live = true
         @socket = TCPSocket.real_open(address, port)
-        @recording = []
       else
         @live = false
-        @recording = cassette.next_session
       end
-      @cassette = cassette
+      @recording = cassette.next_session
+    end
+
+    def read(bytes)
+      _read(:read, bytes)
+    end
+
+    def gets(*args)
+      _read(:gets, *args)
     end
 
     def read_nonblock(bytes)
-      if live
-        data = @socket.read_nonblock(bytes)
-        recording << ["read", data]
-      else
-        direction, data = recording.shift
-        raise TCR::DirectionMismatchError.new("Expected to 'read' but next in recording was 'write'") unless direction == "read"
-      end
+      _read(:read_nonblock, bytes, blocking: false)
+    end
 
-      data
+    def print(str)
+      _write(:print, str)
     end
 
     def write(str)
-      if live
-        len = @socket.write(str)
-        recording << ["write", str]
-      else
-        direction, data = recording.shift
-        raise TCR::DirectionMismatchError.new("Expected to 'write' but next in recording was 'read'") unless direction == "write"
-        len = data.length
-      end
-
-      len
+      _write(:write, str)
+      str.length
     end
 
     def to_io
@@ -59,8 +60,94 @@ module TCR
     def close
       if live
         @socket.close
-        cassette.append(recording)
       end
     end
+
+    private
+
+    def _intercept_socket
+      if @socket
+        @socket = yield @socket
+      end
+    end
+
+    def _block_for_read_data
+      while recording.first && recording.first.first != "read"
+        @read_lock.pop
+      end
+    end
+
+    def _check_for_blocked_reads
+      @read_lock << 1
+    end
+
+    def _write(method, data)
+      if live
+        @socket.__send__(method, data)
+        recording << ["write", data]
+      else
+        direction, data = recording.shift
+        _ensure_direction("write", direction)
+        _check_for_blocked_reads
+      end
+    end
+
+    def _read(method, *args, blocking: true)
+      if live
+          data = @socket.__send__(method, *args)
+          recording << ["read", data]
+      else
+        _block_for_read_data if blocking && TCR.configuration.block_for_reads
+        raise EOFError if recording.empty?
+        direction, data = recording.shift
+        _ensure_direction("read", direction)
+      end
+      data
+    end
+
+    def _ensure_direction(desired, actual)
+      raise TCR::DirectionMismatchError.new("Expected to '#{desired}' but next in recording was '#{actual}'") unless desired == actual
+    end
+  end
+
+  class RecordableSSLSocket < SimpleDelegator
+    def initialize(tcr_socket)
+      super(tcr_socket)
+      tcr_socket.send(:_intercept_socket) do |sock|
+        socket = OpenSSL::SSL::SSLSocket.new(sock, OpenSSL::SSL::SSLContext.new)
+        socket.sync_close = true
+        socket.connect
+        socket
+      end
+    end
+
+    def sync_close=(arg)
+      true
+    end
+
+    def session
+      self
+    end
+
+    def session=(args)
+    end
+
+    def io
+      self
+    end
+
+    def shutdown
+      if live
+        socket.io.shutdown
+      end
+    end
+
+    def connect
+      self
+    end
+
+    def post_connection_check(*args)
+      true
+    end
   end
 end

data/lib/tcr/version.rb

@@ -1,3 +1,3 @@
 module TCR
-  VERSION = "0.0.4"
+  VERSION = "0.0.5"
 end

data/spec/fixtures/block_for_reads.json

@@ -0,0 +1,12 @@
+[
+  [
+    [
+      "write",
+      "hello\n"
+    ],
+    [
+      "read",
+      "world!\n"
+    ]
+  ]
+]

data/spec/fixtures/google_https.json

@@ -0,0 +1,48 @@
+[
+  [
+    [
+      "write",
+      "GET / HTTP/1.1\r\nAccept-Encoding: gzip;q=1.0,deflate;q=0.6,identity;q=0.3\r\nAccept: */*\r\nUser-Agent: Ruby\r\nConnection: close\r\nHost: www.google.com\r\n\r\n"
+    ],
+    [
+      "read",
+      "HTTP/1.1 200 OK\r\nDate: Thu, 15 May 2014 15:04:03 GMT\r\nExpires: -1\r\nCache-Control: private, max-age=0\r\nContent-Type: text/html; charset=ISO-8859-1\r\nSet-Cookie: PREF=ID=46c9c47b277ecf04:FF=0:TM=1400166243:LM=1400166243:S=godo17lIPQRSk5e6; expires=Sat, 14-May-2016 15:04:03 GMT; path=/; domain=.google.com\r\nSet-Cookie: NID=67=hkRuWOEJ-mwgTBbt5NpMcYlD7G38o1tTA_mF_l2CvgY9BPVHzf3wom6KaSnvy5Ln6EqfhpD5t0KSia4b-yBj0AUFJbjfZpf_6487tcmyQMw9PtdewdXO1ZtMDD7McNeB; expires=Fri, 14-Nov-2014 15:04:03 GMT; path=/; domain=.google.com; HttpOnly\r\nP3P: CP=\"This is not a P3P policy! See http://www.google.com/support/accounts/bin/answer.py?hl=en&answer=151657 for more info.\"\r\nServer: gws\r\nX-XSS-Protection: 1; mode=block\r\nX-Frame-Options: SAMEORIGIN\r\nAlternate-Protocol: 443:quic\r\nConnection: close\r\n\r\n<!doctype html><html itemscope=\"\" itemtype=\"http://schema.org/WebPage\" lang=\"en\"><head><meta content=\"Search the world's information, including webpages, images, videos and more. Google has many special features to help you find exactly what you're looking for.\" name=\"description\"><meta content=\"noodp\" name=\"robots\"><meta content=\"/images/google_favicon_128.png\" itemprop=\"image\"><title>Google</title><script>(function(){\nwindow.google={kEI:\"Y9d0U_rDLo-jqAaU0oDoDA\",getEI:function(a){for(var b;a&&(!a.getAttribute||!(b=a.getAttribute(\"eid\")));)a=a.pare"
+    ],
+    [
+      "read",
+      "ntNode;return b||google.kEI},https:function(){return\"https:\"==window.location.protocol},kEXPI:\"17259,4000116,4007661,4007830,4008142,4009033,4009641,4010806,4010858,4010899,4011228,4011258,4011679,4011960,4012373,4012504,4013414,4013591,4013723,4013758,4013787,4013823,4013941,4013967,4013979,4014016,4014093,4014237,4014431,4014515,4014636,4014671,4014789,4014810,4014813,4014991,4015155,4015234,4015260,4015299,4015301,4015519,4015550,4015587,4015638,4015639,4015772,4016013,4016127,4016309,4016363,4016367,4016373,4016446,4016487,4016703,4016730,4016800,4016824,4016855,4016933,4016976,4017261,4017278,4017280,4017285,4017596,4017612,4017639,4017666,4017681,4017694,4017710,4017712,4017742,4017788,4017801,4017818,4017856,4017862,4017913,4017922,4017960,4017962,4018019,8300012,8300015,8500223,8500240,8500252,8500264,8500272,8500306,8500314,8500332,8500365,8500378,8500389,8500394,10200002,10200012,10200014,10200029,10200038,10200040,10200048,10200053,10200066,10200084,10200120,10200134,10200136,10200155,10200157,10200159,10200164,10200176,10200184,10200211,10200221,10200226,10200242,10200246,10200250,10200254,10200261,10200263,10200271,10200293,10200295,10200299\",kCSI:{e:\"17259,4000116,4007661,4007830,4008142,4009033,4009641,4010806,4010858,4010899,4011228,4011258,4011679,4011960,4012373,4012504,4013414,4013591,4013723,401375"
+    ],
+    [
+      "read",
+      "8,4013787,4013823,4013941,4013967,4013979,4014016,4014093,4014237,4014431,4014515,4014636,4014671,4014789,4014810,4014813,4014991,4015155,4015234,4015260,4015299,4015301,4015519,4015550,4015587,4015638,4015639,4015772,4016013,4016127,4016309,4016363,4016367,4016373,4016446,4016487,4016703,4016730,4016800,4016824,4016855,4016933,4016976,4017261,4017278,4017280,4017285,4017596,4017612,4017639,4017666,4017681,4017694,4017710,4017712,4017742,4017788,4017801,4017818,4017856,4017862,4017913,4017922,4017960,4017962,4018019,8300012,8300015,8500223,8500240,8500252,8500264,8500272,8500306,8500314,8500332,8500365,8500378,8500389,8500394,10200002,10200012,10200014,10200029,10200038,10200040,10200048,10200053,10200066,10200084,10200120,10200134,10200136,10200155,10200157,10200159,10200164,10200176,10200184,10200211,10200221,10200226,10200242,10200246,10200250,10200254,10200261,10200263,10200271,10200293,10200295,10200299\",ei:\"Y9d0U_rDLo-jqAaU0oDoDA\"},authuser:0,ml:function(){},kHL:\"en\",time:function(){return(new Date).getTime()},log:function(a,b,c,h,k){var d=\nnew Image,f=google.lc,e=google.li,g=\"\";d.onerror=d.onload=d.onabort=function(){delete f[e]};f[e]=d;c||-1!=b.search(\"&ei=\")||(g=\"&ei=\"+google.getEI(h));c=c||\"/\"+(k||\"gen_204\")+\"?atyp=i&ct=\"+a+\"&cad=\"+b+g+\"&zx=\"+google.time();a=/^http:/i;a.test(c)&&google.https()?(google.ml(Err"
+    ],
+    [
+      "read",
+      "or(\"GLMM\"),!1,{src:c}),delete f[e]):(d.src=c,google.li=e+1)},lc:[],li:0,y:{},x:function(a,b){google.y[a.id]=[a,b];return!1},load:function(a,b,c){google.x({id:a+l++},function(){google.load(a,b,c)})}};var l=0;})();\n(function(){google.sn=\"webhp\";google.timers={};google.startTick=function(a,b){var f=google.time();google.timers[a]={t:{start:f},bfr:!!b};};google.tick=function(a,b,f){google.timers[a]||google.startTick(a);google.timers[a].t[b]=f||google.time()};google.startTick(\"load\",!0);\ntry{}catch(d){}})();\nvar _gjwl=location;function _gjuc(){var a=_gjwl.href.indexOf(\"#\");if(0<=a&&(a=_gjwl.href.substring(a),0<a.indexOf(\"&q=\")||0<=a.indexOf(\"#q=\"))&&(a=a.substring(1),-1==a.indexOf(\"#\"))){for(var d=0;d<a.length;){var b=d;\"&\"==a.charAt(b)&&++b;var c=a.indexOf(\"&\",b);-1==c&&(c=a.length);b=a.substring(b,c);if(0==b.indexOf(\"fp=\"))a=a.substring(0,d)+a.substring(c,a.length),c=d;else if(\"cad=h\"==b)return 0;d=c}_gjwl.href=\"/search?\"+a+\"&cad=h\";return 1}return 0}\nfunction _gjh(){!_gjuc()&&window.google&&google.x&&google.x({id:\"GJH\"},function(){google.nav&&google.nav.gjh&&google.nav.gjh()})};\nwindow._gjh&&_gjh();</script><style>#gbar,#guser{font-size:13px;padding-top:1px !important;}#gbar{height:22px}#guser{padding-bottom:7px !important;text-align:right}.gbh,.gbd{border-top:1px solid #c9d7f1;font-size:1px}.gbh{height:0;position:absolu"
+    ],
+    [
+      "read",
+      "te;top:24px;width:100%}@media all{.gb1{height:22px;margin-right:.5em;vertical-align:top}#gbar{float:left}}a.gb1,a.gb4{text-decoration:underline !important}a.gb1,a.gb4{color:#00c !important}.gbi .gb4{color:#dd8e27 !important}.gbf .gb4{color:#900 !important}</style><style>body,td,a,p,.h{font-family:arial,sans-serif}body{margin:0;overflow-y:scroll}#gog{padding:3px 8px 0}td{line-height:.8em}.gac_m td{line-height:17px}form{margin-bottom:20px}.h{color:#36c}.q{color:#00c}.ts td{padding:0}.ts{border-collapse:collapse}em{font-weight:bold;font-style:normal}.lst{height:25px;width:496px}.gsfi,.lst{font:18px arial,sans-serif}.gsfs{font:17px arial,sans-serif}.ds{display:inline-box;display:inline-block;margin:3px 0 4px;margin-left:4px}input{font-family:inherit}a.gb1,a.gb2,a.gb3,a.gb4{color:#11c !important}body{background:#fff;color:black}a{color:#11c;text-decoration:none}a:hover,a:active{text-decoration:underline}.fl a{color:#36c}a:visited{color:#551a8b}a.gb1,a.gb4{text-decoration:underline}a.gb3:hover{text-decoration:none}#ghead a.gb2:hover{color:#fff !important}.sblc{padding-top:5px}.sblc a{display:block;margin:2px 0;margin-left:13px;font-size:11px}.lsbb{background:#eee;border:solid 1px;border-color:#ccc #999 #999 #ccc;height:30px}.lsbb{display:block}.ftl,#fll a{display:inline-block;margin:0 12px}.lsb{background:url(/images/srpr/n"
+    ],
+    [
+      "read",
+      "av_logo80.png) 0 -258px repeat-x;border:none;color:#000;cursor:pointer;height:30px;margin:0;outline:0;font:15px arial,sans-serif;vertical-align:top}.lsb:active{background:#ccc}.lst:focus{outline:none}#addlang a{padding:0 3px}</style><script></script></head><body bgcolor=\"#fff\"><script>(function(){var src='/images/nav_logo176.png';var iesg=false;document.body.onload = function(){window.n && window.n();if (document.images){new Image().src=src;}\nif (!iesg){document.f&&document.f.q.focus();document.gbqf&&document.gbqf.q.focus();}\n}\n})();</script><textarea id=\"csi\" style=\"display:none\"></textarea><div id=\"mngb\"> <div id=gbar><nobr><b class=gb1>Search</b> <a class=gb1 href=\"https://www.google.com/imghp?hl=en&tab=wi\">Images</a> <a class=gb1 href=\"https://maps.google.com/maps?hl=en&tab=wl\">Maps</a> <a class=gb1 href=\"https://play.google.com/?hl=en&tab=w8\">Play</a> <a class=gb1 href=\"https://www.youtube.com/?tab=w1\">YouTube</a> <a class=gb1 href=\"https://news.google.com/nwshp?hl=en&tab=wn\">News</a> <a class=gb1 href=\"https://mail.google.com/mail/?tab=wm\">Gmail</a> <a class=gb1 href=\"https://drive.google.com/?tab=wo\">Drive</a> <a class=gb1 style=\"text-decoration:none\" href=\"http://www.google.com/intl/en/options/\"><u>More</u> &raquo;</a></nobr></div><div id=guser width=100%><nobr><span id=gbn class=gbi></span><span id=gbf class="
+    ],
+    [
+      "read",
+      "gbf></span><span id=gbe></span><a href=\"http://www.google.com/history/optout?hl=en\" class=gb4>Web History</a> | <a  href=\"/preferences?hl=en\" class=gb4>Settings</a> | <a target=_top id=gb_70 href=\"https://accounts.google.com/ServiceLogin?hl=en&continue=https://www.google.com/\" class=gb4>Sign in</a></nobr></div><div class=gbh style=left:0></div><div class=gbh style=right:0></div>  </div><center><br clear=\"all\" id=\"lgpd\"><div id=\"lga\"><img alt=\"Google\" height=\"95\" src=\"/images/srpr/logo9w.png\" style=\"padding:28px 0 14px\" width=\"269\" id=\"hplogo\" onload=\"window.lol&&lol()\"><br><br></div><form action=\"/search\" name=\"f\"><table cellpadding=\"0\" cellspacing=\"0\"><tr valign=\"top\"><td width=\"25%\">&nbsp;</td><td align=\"center\" nowrap=\"\"><input name=\"ie\" value=\"ISO-8859-1\" type=\"hidden\"><input value=\"en\" name=\"hl\" type=\"hidden\"><input name=\"source\" type=\"hidden\" value=\"hp\"><div class=\"ds\" style=\"height:32px;margin:4px 0\"><input style=\"color:#000;margin:0;padding:5px 8px 0 6px;vertical-align:top\" autocomplete=\"off\" class=\"lst\" value=\"\" title=\"Google Search\" maxlength=\"2048\" name=\"q\" size=\"57\"></div><br style=\"line-height:0\"><span class=\"ds\"><span class=\"lsbb\"><input class=\"lsb\" value=\"Google Search\" name=\"btnG\" type=\"submit\"></span></span><span class=\"ds\"><span class=\"lsbb\"><input class=\"lsb\" value=\"I'm Feeling Lucky\" name=\"btnI\" on"
+    ],
+    [
+      "read",
+      "click=\"if(this.form.q.value)this.checked=1; else top.location='/doodles/'\" type=\"submit\"></span></span></td><td class=\"fl sblc\" align=\"left\" nowrap=\"\" width=\"25%\"><a href=\"/advanced_search?hl=en&amp;authuser=0\">Advanced search</a><a href=\"/language_tools?hl=en&amp;authuser=0\">Language tools</a></td></tr></table><input id=\"gbv\" name=\"gbv\" type=\"hidden\" value=\"1\"></form><div id=\"gac_scont\"></div><div style=\"font-size:83%;min-height:3.5em\"><br></div><span id=\"footer\"><div style=\"font-size:10pt\"><div style=\"margin:19px auto;text-align:center\" id=\"fll\"><a href=\"/intl/en/ads/\">Advertising&nbsp;Programs</a><a href=\"/services/\">Business Solutions</a><a href=\"https://plus.google.com/116899029375914044550\" rel=\"publisher\">+Google</a><a href=\"/intl/en/about.html\">About Google</a></div></div><p style=\"color:#767676;font-size:8pt\">&copy; 2013 - <a href=\"/intl/en/policies/\">Privacy & Terms</a></p></span></center><div id=xjsd></div><div id=xjsi data-jiis=\"bp\"><script>if(google.y)google.y.first=[];(function(){function b(a){window.setTimeout(function(){var c=document.createElement(\"script\");c.src=a;document.getElementById(\"xjsd\").appendChild(c)},0)}google.dljp=function(a){google.xjsu=a;b(a)};google.dlj=b;})();\nif(!google.xjs){window._=window._||{};window._._DumpException=function(e){throw e};if(google.timers&&google.timers.load.t){goo"
+    ],
+    [
+      "read",
+      "gle.timers.load.t.xjsls=new Date().getTime();}google.dljp('/xjs/_/js/k\\x3dxjs.hp.en_US.Gh3uPFri1JA.O/m\\x3dsb_he,pcc/rt\\x3dj/d\\x3d1/sv\\x3d1/rs\\x3dAItRSTNl_2-4gdX7sEzEk9xHPbjumUrcCA');google.xjs=1;}google.pmc={\"sb_he\":{\"agen\":true,\"cgen\":true,\"client\":\"heirloom-hp\",\"dh\":true,\"ds\":\"\",\"eqch\":true,\"fl\":true,\"host\":\"google.com\",\"jam\":0,\"jsonp\":true,\"msgs\":{\"cibl\":\"Clear Search\",\"dym\":\"Did you mean:\",\"lcky\":\"I\\u0026#39;m Feeling Lucky\",\"lml\":\"Learn more\",\"oskt\":\"Input tools\",\"psrc\":\"This search was removed from your \\u003Ca href=\\\"/history\\\"\\u003EWeb History\\u003C/a\\u003E\",\"psrl\":\"Remove\",\"sbit\":\"Search by image\",\"srch\":\"Google Search\"},\"ovr\":{},\"pq\":\"\",\"qcpw\":false,\"scd\":10,\"sce\":5,\"stok\":\"PoibB8u7e-sS13448HMOoIqGSu4\"},\"pcc\":{}};google.y.first.push(function(){if(google.med){google.med('init');google.initHistory();google.med('history');}});if(google.j&&google.j.en&&google.j.xi){window.setTimeout(google.j.xi,0);}</script></div><script>(function(){if(google.timers&&google.timers.load.t){var b,c,d,e,g=function(a,f){a.removeEventListener?(a.removeEventListener(\"load\",f,!1),a.removeEventListener(\"error\",f,!1)):(a.detachEvent(\"onload\",f),a.detachEvent(\"onerror\",f))},h=function(a){e=(new Date).getTime();++c;a=a||window.event;a=a.target||a.srcElement;g(a,h)},k=document.getElementsByTagName(\"img\");b=k.length;for(var l=c=0,m;l<b;++l)m"
+    ],
+    [
+      "read",
+      "=k[l],m.complete||\"string\"!=typeof m.src||!m.src?++c:m.addEventListener?(m.addEventListener(\"load\",h,!1),m.addEventListener(\"error\",\nh,!1)):(m.attachEvent(\"onload\",h),m.attachEvent(\"onerror\",h));d=b-c;var n=function(){if(google.timers.load.t){google.timers.load.t.ol=(new Date).getTime();google.timers.load.t.iml=e;google.kCSI.imc=c;google.kCSI.imn=b;google.kCSI.imp=d;void 0!==google.stt&&(google.kCSI.stt=google.stt);google.csiReport&&google.csiReport()}};window.addEventListener?window.addEventListener(\"load\",n,!1):window.attachEvent&&\nwindow.attachEvent(\"onload\",n);google.timers.load.t.prt=e=(new Date).getTime()};})();\n</script></body></html>"
+    ]
+  ]
+]

data/spec/fixtures/google_imap.json

@@ -0,0 +1,92 @@
+[
+  [
+    [
+      "read",
+      "* OK Gimap ready for requests from 72.16.218.22 n58mb218005052yhi\r\n"
+    ],
+    [
+      "write",
+      "RUBY0001 LOGIN"
+    ],
+    [
+      "write",
+      " "
+    ],
+    [
+      "write",
+      "ben.olive@example.net"
+    ],
+    [
+      "write",
+      " "
+    ],
+    [
+      "write",
+      "password"
+    ],
+    [
+      "write",
+      "\r\n"
+    ],
+    [
+      "read",
+      "* CAPABILITY IMAP4rev1 UNSELECT IDLE NAMESPACE QUOTA ID XLIST CHILDREN X-GM-EXT-1 UIDPLUS COMPRESS=DEFLATE ENABLE MOVE CONDSTORE ESEARCH\r\n"
+    ],
+    [
+      "read",
+      "RUBY0001 OK ben.olive@example.net Ben Olive authenticated (Success)\r\n"
+    ],
+    [
+      "write",
+      "RUBY0002 EXAMINE"
+    ],
+    [
+      "write",
+      " "
+    ],
+    [
+      "write",
+      "INBOX"
+    ],
+    [
+      "write",
+      "\r\n"
+    ],
+    [
+      "read",
+      "* FLAGS (\\Answered \\Flagged \\Draft \\Deleted \\Seen $Phishing $NotPhishing)\r\n"
+    ],
+    [
+      "read",
+      "* OK [PERMANENTFLAGS ()] Flags permitted.\r\n"
+    ],
+    [
+      "read",
+      "* OK [UIDVALIDITY 1] UIDs valid.\r\n"
+    ],
+    [
+      "read",
+      "* 206 EXISTS\r\n"
+    ],
+    [
+      "read",
+      "* 0 RECENT\r\n"
+    ],
+    [
+      "read",
+      "* OK [UIDNEXT 5504] Predicted next UID.\r\n"
+    ],
+    [
+      "read",
+      "* OK [HIGHESTMODSEQ 609503]\r\n"
+    ],
+    [
+      "read",
+      "RUBY0002 OK [READ-ONLY] INBOX selected. (Success)\r\n"
+    ],
+    [
+      "read",
+      null
+    ]
+  ]
+]

data/spec/fixtures/starwars_telnet.json

@@ -0,0 +1,8 @@
+[
+  [
+    [
+      "read",
+      "\u001b[H\u001b[J\u001b[H\r\n\r\n\r\n\r\n\r\n\r\n                             "
+    ]
+  ]
+]

data/spec/spec_helper.rb

@@ -3,6 +3,20 @@
 # Require this file using `require "spec_helper"` to ensure that it is only
 # loaded once.
 #
+module SpawnTestServers
+  # Spawn a new server that will send the specified payload when connected to.
+  # The listening port number will be returned
+  def spawn_server(payload)
+    server = TCPServer.new("127.0.0.1", 0)
+    fork do
+      client = server.accept
+      client.write(payload)
+      client.close
+    end
+    server.addr[1]
+  end
+end
+
 # See http://rubydoc.info/gems/rspec-core/RSpec/Core/Configuration
 RSpec.configure do |config|
   config.treat_symbols_as_metadata_keys_with_true_values = true
@@ -14,4 +28,6 @@ RSpec.configure do |config|
   # the seed, which is printed after each run.
   #     --seed 1234
   config.order = 'random'
+
+  config.include(SpawnTestServers)
 end

data/spec/tcr_spec.rb

@@ -1,13 +1,23 @@
 require "spec_helper"
 require "tcr"
 require "net/protocol"
+require "net/http"
+require "net/imap"
 require "net/smtp"
+require 'thread'
+
 
 describe TCR do
   before(:each) do
     TCR.configuration.reset_defaults!
   end
 
+  around(:each) do |example|
+    File.unlink("test.json") if File.exists?("test.json")
+    example.run
+    File.unlink("test.json") if File.exists?("test.json")
+  end
+
   describe ".configuration" do
      it "has a default cassette location configured" do
        TCR.configuration.cassette_library_dir.should == "fixtures/tcr_cassettes"
@@ -16,6 +26,10 @@ describe TCR do
      it "has an empty list of hook ports by default" do
        TCR.configuration.hook_tcp_ports.should == []
      end
+
+     it "defaults to erroring on read/write mismatch access" do
+       TCR.configuration.block_for_reads.should be_false
+     end
    end
 
    describe ".configure" do
@@ -30,6 +44,12 @@ describe TCR do
          TCR.configure { |c| c.hook_tcp_ports = [25] }
        }.to change{ TCR.configuration.hook_tcp_ports }.from([]).to([25])
      end
+
+     it "configures allowing a blocking read mode" do
+       expect {
+         TCR.configure { |c| c.block_for_reads = true }
+       }.to change{ TCR.configuration.block_for_reads }.from(false).to(true)
+     end
    end
 
    it "raises an error if you connect to a hooked port without using a cassette" do
@@ -62,16 +82,50 @@ describe TCR do
     end
   end
 
+  describe "block_for_reads" do
+    before(:each) {
+      TCR.configure { |c|
+        c.hook_tcp_ports = [9999]
+        c.cassette_library_dir = '.'
+      }
+    }
+
+    it "blocks read thread until data is available instead of raising mismatch error" do
+      TCR.configure { |c| c.block_for_reads = true }
+      reads = Queue.new
+
+      TCR.use_cassette("spec/fixtures/block_for_reads") do
+        sock = TCPSocket.open("google.com", 9999)
+
+        t = Thread.new do
+          reads << sock.gets
+        end
+
+        expect(reads.size).to eq(0)
+        sock.print("hello\n")
+        t.value
+        expect(reads.size).to eq(1)
+      end
+    end
+
+    context "when disabled" do
+      it "raises mismatch error" do
+        TCR.use_cassette("spec/fixtures/block_for_reads") do
+          sock = TCPSocket.open("google.com", 9999)
+          expect {
+            Timeout::timeout(1) { sock.gets }
+          }.to raise_error(TCR::DirectionMismatchError)
+        end
+      end
+    end
+  end
+
   describe ".use_cassette" do
     before(:each) {
       TCR.configure { |c|
         c.hook_tcp_ports = [25]
         c.cassette_library_dir = "."
       }
-      File.unlink("test.json") if File.exists?("test.json")
-    }
-    after(:each) {
-      File.unlink("test.json") if File.exists?("test.json")
     }
 
     it "requires a block to call" do
@@ -90,7 +144,6 @@ describe TCR do
       expect {
         TCR.use_cassette("test") do
           tcp_socket = TCPSocket.open("aspmx.l.google.com", 25)
-          tcp_socket.close
         end
       }.to change{ File.exists?("./test.json") }.from(false).to(true)
     end
@@ -126,6 +179,42 @@ describe TCR do
       }.to raise_error(TCR::DirectionMismatchError)
     end
 
+    it "stubs out Socket#gets" do
+      TCR.configure { |c|
+        c.hook_tcp_ports = [993]
+        c.block_for_reads = true
+      }
+      expect {
+        TCR.use_cassette("spec/fixtures/google_imap") do
+          conn = Net::IMAP.new("imap.gmail.com", 993, true)
+          conn.login("ben.olive@example.net", "password")
+          conn.examine(Net::IMAP.encode_utf7("INBOX"))
+          conn.disconnect
+        end
+      }.not_to raise_error
+    end
+
+    it "stubs out Socket#read" do
+      TCR.configure { |c|
+        c.hook_tcp_ports = [23]
+      }
+      TCR.use_cassette("spec/fixtures/starwars_telnet") do
+        sock = TCPSocket.open("towel.blinkenlights.nl", 23)
+        expect(sock.read(50).length).to eq(50)
+        sock.close
+      end
+    end
+
+    it "supports ssl sockets" do
+      TCR.configure { |c| c.hook_tcp_ports = [443] }
+      http = Net::HTTP.new("www.google.com", 443)
+      http.use_ssl = true
+      expect {
+        TCR.use_cassette("spec/fixtures/google_https") do
+          http.request(Net::HTTP::Get.new("/"))
+        end
+      }.not_to raise_error
+    end
 
     context "multiple connections" do
       it "records multiple sessions per cassette" do
@@ -152,6 +241,35 @@ describe TCR do
         end
       end
 
+      context "when a cassette is recorded with connections opens concurrently" do
+        let(:ports) { ["Apple", "Banana"].map { |payload| spawn_server(payload) } }
+        before(:each) do
+          TCR.configure { |c|
+            c.hook_tcp_ports = ports
+            c.cassette_library_dir = "."
+          }
+        end
+        before(:each) do
+          TCR.use_cassette("test") do
+            apple = TCPSocket.open("127.0.0.1", ports.first)
+            banana = TCPSocket.open("127.0.0.1", ports.last)
+
+            banana.gets
+            apple.gets
+
+            banana.close
+            apple.close
+          end
+        end
+
+        it "replays the sessions in the order they were created" do
+          TCR.use_cassette("test") do
+            apple = TCPSocket.open("127.0.0.1", ports.first)
+            expect(apple.gets).to eq("Apple")
+          end
+        end
+      end
+
       it "raises an error if you try to playback more sessions than you previously recorded" do
         expect {
           TCR.use_cassette("spec/fixtures/multitest-smtp") do
@@ -163,4 +281,4 @@ describe TCR do
       end
     end
   end
-end
+end

metadata

@@ -1,46 +1,41 @@
 --- !ruby/object:Gem::Specification
 name: tcr
 version: !ruby/object:Gem::Version
-  version: 0.0.4
-  prerelease: 
+  version: 0.0.5
 platform: ruby
 authors:
 - Rob Forman
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-09-12 00:00:00.000000000 Z
+date: 2015-01-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec
   requirement: !ruby/object:Gem::Requirement
-    none: false
     requirements:
-    - - ! '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
-    none: false
     requirements:
-    - - ! '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
   name: geminabox
   requirement: !ruby/object:Gem::Requirement
-    none: false
     requirements:
-    - - ! '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
-    none: false
     requirements:
-    - - ! '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
 description: TCR is a lightweight VCR for TCP sockets.
@@ -50,8 +45,9 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
-- .gitignore
-- .rvmrc
+- ".gitignore"
+- ".rvmrc"
+- ".travis.yml"
 - Gemfile
 - LICENSE.txt
 - README.md
@@ -62,39 +58,46 @@ files:
 - lib/tcr/errors.rb
 - lib/tcr/recordable_tcp_socket.rb
 - lib/tcr/version.rb
+- spec/fixtures/block_for_reads.json
+- spec/fixtures/google_https.json
+- spec/fixtures/google_imap.json
 - spec/fixtures/google_smtp.json
 - spec/fixtures/multitest-smtp.json
 - spec/fixtures/multitest.json
+- spec/fixtures/starwars_telnet.json
 - spec/spec_helper.rb
 - spec/tcr_spec.rb
 - tcr.gemspec
 homepage: ''
 licenses: []
+metadata: {}
 post_install_message: 
 rdoc_options: []
 require_paths:
 - lib
 required_ruby_version: !ruby/object:Gem::Requirement
-  none: false
   requirements:
-  - - ! '>='
+  - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 required_rubygems_version: !ruby/object:Gem::Requirement
-  none: false
   requirements:
-  - - ! '>='
+  - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 1.8.25
+rubygems_version: 2.2.2
 signing_key: 
-specification_version: 3
+specification_version: 4
 summary: TCR is a lightweight VCR for TCP sockets.
 test_files:
+- spec/fixtures/block_for_reads.json
+- spec/fixtures/google_https.json
+- spec/fixtures/google_imap.json
 - spec/fixtures/google_smtp.json
 - spec/fixtures/multitest-smtp.json
 - spec/fixtures/multitest.json
+- spec/fixtures/starwars_telnet.json
 - spec/spec_helper.rb
 - spec/tcr_spec.rb