Assert that the return value from ssh block is returned

1 files changed, 9 insertions, 0 deletions

diff --git a/test/start/test_connection.rb b/test/start/test_connection.rb
index 94b68c4..43cbf0e 100644
--- a/test/start/test_connection.rb
+++ b/test/start/test_connection.rb
@@ -39,6 +39,15 @@ module NetSSH
         # We aren't interested in the exception
       end
     end
+    
+    def test_return_value_is_returned
+      @connection_session.expects(:closed?).returns(false)
+      @connection_session.expects(:close).once
+      
+      val = 1
+      retval = Net::SSH.start('localhost', 'testuser') { val }
+      assert_equal(val, retval)
+    end
   end
 end