Added workaround for exit code being available for remote commands that exit immediately

Author: Dan

pssh/ssh_client.py

@@ -229,7 +229,7 @@ def exec_command(self, command, sudo=False, user=None, **kwargs):
         logger.debug("Running command %s on %s", command, self.host)
         channel.exec_command(command, **kwargs)
         logger.debug("Command started")
-        gevent.sleep(0)
+        gevent.sleep(.2)
         return channel, self.host, stdout, stderr
 
     def _read_output_buffer(self, output_buffer, prefix=''):

tests/test_pssh_client.py

@@ -70,10 +70,9 @@ def test_pssh_client_exec_command(self):
                         msg="No output for host")
         self.assertTrue(output[self.host]['exit_code'] == 0)
 
-    def test_pssh_client_no_stdout_non_zero_exit_code(self):
+    def test_pssh_client_no_stdout_non_zero_exit_code_immediate_exit(self):
         output = self.client.run_command('exit 1')
         expected_exit_code = 1
-        self.client.join(output)
         exit_code = output[self.host]['exit_code']
         self.assertEqual(expected_exit_code, exit_code,
                          msg="Got unexpected exit code - %s, expected %s" %
@@ -113,9 +112,9 @@ def test_pssh_client_run_command_get_output(self):
         expected_exit_code = 0
         expected_stdout = [self.fake_resp]
         expected_stderr = []
+        exit_code = output[self.host]['exit_code']
         stdout = list(output[self.host]['stdout'])
         stderr = list(output[self.host]['stderr'])
-        exit_code = output[self.host]['exit_code']
         self.assertEqual(expected_exit_code, exit_code,
                          msg="Got unexpected exit code - %s, expected %s" %
                          (exit_code,