diff options
Diffstat (limited to 'test/multi_test.rb')
-rw-r--r-- | test/multi_test.rb | 20 |
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 |