first pass at tests

1 files changed, 20 insertions, 0 deletions

diff --git a/test/multi_test.rb b/test/multi_test.rb
new file mode 100644
index 0000000..0a5d5bb
--- /dev/null
+++ b/test/multi_test.rb
@@ -0,0 +1,20 @@
+require 'common'
+require 'net/ssh/multi'
+
+class MultiTest < Test::Unit::TestCase
+  def test_start_with_block_should_yield_session_and_then_close
+    Net::SSH::Multi::Session.any_instance.expects(:loop)
+    Net::SSH::Multi::Session.any_instance.expects(:close)
+    yielded = false
+    Net::SSH::Multi.start do |session|
+      yielded = true
+      assert_instance_of Net::SSH::Multi::Session, session
+    end
+  end
+
+  def test_start_without_block_should_return_open_session
+    Net::SSH::Multi::Session.any_instance.expects(:loop).never
+    Net::SSH::Multi::Session.any_instance.expects(:close).never
+    assert_instance_of Net::SSH::Multi::Session, Net::SSH::Multi.start
+  end
+end
\ No newline at end of file