test/unit/test_connection_manager.rb in sshkit-0.0.2 vs test/unit/test_connection_manager.rb in sshkit-0.0.3

@@ -1,15 +1,27 @@
+require 'time'
 require 'helper'
 
 module SSHKit
 
   class TestConnectionManager < UnitTest
 
     def setup
-      SSHKit.config.backend = SSHKit::Backend::Abstract
+      @s = String.new
+      SSHKit.config.backend = SSHKit::Backend::Printer
     end
 
+    def tearddown
+      @s = nil
+    end
+
+    def block_to_run
+      lambda do |host|
+        execute "echo #{Time.now}"
+      end
+    end
+
     def test_connection_manager_handles_a_single_argument
       h = Host.new('1.example.com')
       Host.expects(:new).with('1.example.com').once().returns(h)
       ConnectionManager.new '1.example.com'
     end
@@ -24,54 +36,55 @@
       cm = ConnectionManager.new %w{[e-mail]:22 [e-mail]}
       assert_equal ['[e-mail]:22'], cm.hosts.map(&:to_s)
     end
 
     def test_the_connection_manager_yields_the_host_to_each_connection_instance
-      spy = lambda do |host, connection|
-        assert_equal host, Host.new("1.example.com")
+      spy = lambda do |host|
+        execute "echo #{host.hostname}"
       end
-      ConnectionManager.new(%w{1.example.com}).each &spy
+      String.new.tap do |str|
+        SSHKit.capture_output str do
+          ConnectionManager.new(%w{1.example.com}).each &spy
+        end
+        assert_equal "echo 1.example.com", str.strip
+      end
     end
 
     def test_the_connection_manaager_runs_things_in_parallel_by_default
-      results = []
-      command = lambda do |host,connection|
-        results << Time.now
+      SSHKit.capture_output @s do
+        ConnectionManager.new(%w{1.example.com 2.example.com}).each &block_to_run
       end
-      ConnectionManager.new(%w{1.example.com 2.example.com}).each &command
       assert_equal 2, results.length
       assert_equal *results.map(&:to_i)
     end
 
     def test_the_connection_manager_can_run_things_in_sequence
-      results = []
-      command = lambda do |host,connection|
-        results << Time.now
+      SSHKit.capture_output @s do
+        ConnectionManager.new(%w{1.example.com 2.example.com}).each in: :sequence, &block_to_run
       end
-      ConnectionManager.new(%w{1.example.com 2.example.com}).each(in: :sequence, &command)
       assert_equal 2, results.length
       assert_operator results.first.to_i, :<, results.last.to_i
     end
 
     def test_the_connection_manager_can_run_things_in_groups
-      results = []
-      command = lambda do |host,connection|
-        debugger
-        results << Time.now
+      SSHKit.capture_output @s do
+        ConnectionManager.new(%w{1.example.com 2.example.com 3.example.com
+                                 4.example.com 5.example.com 6.example.com}).each in: :groups, &block_to_run
       end
-      ConnectionManager.new(%w{1.example.com 2.example.com 3.example.com
-                               4.example.com 5.example.com 6.example.com}).each(in: :groups, &command)
       assert_equal 6, results.length
       assert_equal *results[0..1].map(&:to_i)
       assert_equal *results[2..3].map(&:to_i)
       assert_equal *results[4..5].map(&:to_i)
       assert_operator results[0].to_i, :<, results[2].to_i
       assert_operator results[3].to_i, :<, results[4].to_i
     end
 
-    def test_slow_host_timeout
-      # Ensure that we throw an error and rollback if one host takes an
-      # exceptional length of time longer than the others
+    private
+
+    def results
+      @s.lines.collect do |line|
+        Time.parse(line.split[1..-1].join(' '))
+      end
     end
 
   end
 
 end