test/yads/test_ssh.rb in yads-0.1.0 vs test/yads/test_ssh.rb in yads-0.1.1

@@ -6,10 +6,16 @@
     connection_mock
 
     Yads::SSH.new(:host => "example.org", :user => "deploy", :forward_agent => true)
   end
 
+  def test_connect_using_non_standard_port
+    connection_mock(nil, :port => 2222)
+
+    Yads::SSH.new(:host => "example.org", :port => 2222, :user => "deploy", :forward_agent => true)
+  end
+
   def test_execute
     session = mock
     session.expects(:exec!).with("mkdir -p /tmp/yads")
     connection_mock(session)
 
@@ -38,9 +44,9 @@
     s.close
   end
 
   private
 
-    def connection_mock(session = nil)
-      Net::SSH.expects(:start).with("example.org", "deploy", :forward_agent => true).returns(session)
-    end
+  def connection_mock(session = nil, options = {})
+    Net::SSH.expects(:start).with("example.org", "deploy", { :forward_agent => true }.merge(options)).returns(session)
+  end
 end