Created
December 7, 2011 11:34
-
-
Save hodgestar/1442491 to your computer and use it in GitHub Desktop.
Telnet test fix
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
diff --git a/vumi/transports/telnet/tests/test_telnet.py b/vumi/transports/telnet/tests/test_telnet.py | |
index e5a72d0..4e407fc 100644 | |
--- a/vumi/transports/telnet/tests/test_telnet.py | |
+++ b/vumi/transports/telnet/tests/test_telnet.py | |
@@ -35,6 +35,7 @@ class TelnetServerTransportTestCase(TransportTestCase): | |
super(TelnetServerTransportTestCase, self).setUp() | |
self.worker = yield self.get_transport({'telnet_port': 0}) | |
self.client = yield self.make_client() | |
+ yield self.wait_for_client_start() | |
def tearDown(self): | |
super(TelnetServerTransportTestCase, self).tearDown() | |
@@ -47,6 +48,10 @@ class TelnetServerTransportTestCase(TransportTestCase): | |
for content in contents] | |
return messages | |
+ def wait_for_client_start(self): | |
+ """Wait for first message from client to be ready.""" | |
+ return self._amqp.wait_messages('vumi', 'test.inbound', 1) | |
+ | |
@inlineCallbacks | |
def make_client(self): | |
addr = self.worker.telnet_server.getHost() |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment