Change to join with timeout, add logging

Author: pkittenis

tests/test_pssh_client.py

@@ -5,11 +5,13 @@
 import unittest
 import gevent
 from pssh import ParallelSSHClient, UnknownHostException, \
-    AuthenticationException, ConnectionErrorException
+    AuthenticationException, ConnectionErrorException, _setup_logger
 from fake_server.fake_server import listen, logger as server_logger
 import random
 import logging
 
+_setup_logger(server_logger)
+
 class ParallelSSHClientTest(unittest.TestCase):
 
     def setUp(self):
@@ -19,7 +21,7 @@ def setUp(self):
     def test_pssh_client_exec_command(self):
         listen_port = random.randint(1026, 65534)
         server = gevent.spawn(listen, { self.fake_cmd : self.fake_resp }, listen_port = listen_port)
-        gevent.sleep(1)
+        server.join(5)
         client = ParallelSSHClient(['localhost'], port=listen_port)
         cmd = client.exec_command(self.fake_cmd)[0]
         output = client.get_stdout(cmd)
@@ -33,7 +35,7 @@ def test_pssh_client_auth_failure(self):
         listen_port = random.randint(2048, 65534)
         server = gevent.spawn(listen, { self.fake_cmd : self.fake_resp },
                               listen_port=listen_port, fail_auth=True, )# link_callback=)
-        gevent.sleep(1)
+        server.join(5)
         client = ParallelSSHClient(['localhost'], port=listen_port)
         cmd = client.exec_command(self.fake_cmd)[0]
         # Handle exception