WIP for #25

Author: Dan

pssh.py

@@ -489,9 +489,9 @@ def run_command(self, *args, **kwargs):
                        'cmd'     : <greenlet>}}
 
         """
-        for host in self.hosts:
-            self.pool.spawn(self._exec_command, host, *args, **kwargs)
-        return self.get_output()
+        cmds = [self.pool.spawn(self._exec_command, host, *args, **kwargs)
+                for host in self.hosts]
+        return self.get_output(commands=cmds)
 
     def exec_command(self, *args, **kwargs):
         """Run command on all hosts in parallel, honoring `self.pool_size`

tests/test_pssh_client.py

@@ -311,6 +311,11 @@ def test_pssh_pool_size(self):
                          msg="Expected pool size to be %s, got %s" % (
                              expected, actual,))
 
+    def test_pssh_hosts_more_than_pool_size(self):
+        """Test we can successfully run on more hosts than our pool size and
+        get logs for all hosts"""
+        raise NotImplementedError
+        
     def test_ssh_proxy(self):
         """Test connecting to remote destination via SSH proxy
         client -> proxy -> destination