diff options
-rw-r--r-- | gear/__init__.py | 15 | ||||
-rw-r--r-- | gear/tests/test_gear.py | 6 |
2 files changed, 20 insertions, 1 deletions
diff --git a/gear/__init__.py b/gear/__init__.py index fa6b872..fecce32 100644 --- a/gear/__init__.py +++ b/gear/__init__.py @@ -1199,15 +1199,28 @@ class BaseClient(BaseClientServer): finally: self.connections_condition.release() - def waitForServer(self): + def _checkTimeout(self, start_time, timeout): + if time.time() - start_time > timeout: + raise TimeoutError() + + def waitForServer(self, timeout=None): """Wait for at least one server to be connected. Block until at least one gearman server is connected. + + :arg numeric timeout: Number of seconds to wait for a connection. + If None, wait forever (default: no timeout). + :raises TimeoutError: If the timeout is reached before any server + connects. """ + connected = False + start_time = time.time() while self.running: self.connections_condition.acquire() while self.running and not self.active_connections: + if timeout is not None: + self._checkTimeout(start_time, timeout) self.log.debug("Waiting for at least one active connection") self.connections_condition.wait(timeout=1) if self.active_connections: diff --git a/gear/tests/test_gear.py b/gear/tests/test_gear.py index 3791295..546d4c4 100644 --- a/gear/tests/test_gear.py +++ b/gear/tests/test_gear.py @@ -232,6 +232,12 @@ class TestServerConnection(tests.BaseTestCase): class TestClient(tests.BaseTestCase): + def test_wait_for_server_timeout(self): + client = gear.Client('client') + client.addServer('127.0.0.1', 0) + self.assertRaises(gear.TimeoutError, + client.waitForServer, timeout=1) + def test_handleStatusRes_1(self): client = gear.Client() |