Skip to content

Instantly share code, notes, and snippets.

@hodgestar
Created December 7, 2011 11:34
Show Gist options
  • Save hodgestar/1442491 to your computer and use it in GitHub Desktop.
Save hodgestar/1442491 to your computer and use it in GitHub Desktop.
Telnet test fix
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