Updated mock socket to expect the new protocol negotiation behavior.

1 files changed, 3 insertions, 2 deletions

diff --git a/test/transport/test_server_version.rb b/test/transport/test_server_version.rb
index 44670d1..e8b2740 100644
--- a/test/transport/test_server_version.rb
+++ b/test/transport/test_server_version.rb
@@ -38,10 +38,11 @@ module Transport
 
       def socket(good, *lines)
         socket = mock("socket")
-        socket.expects(:readline).times(lines.length).returns(*lines)
+        socket.expects(:recv).with(1).times(lines.join.length).returns(*lines.join.split(''))
 
         if good
           socket.expects(:write).with("#{Net::SSH::Transport::ServerVersion::PROTO_VERSION}\r\n")
+          socket.expects(:flush)
         else
           socket.expects(:write).never
         end
@@ -54,4 +55,4 @@ module Transport
       end
   end
 
-end
\ No newline at end of file
+end