From eb9defa5cc618038aa41e5eb69c72d7d98c89e4e Mon Sep 17 00:00:00 2001 From: Nico Date: Tue, 15 Sep 2015 23:38:24 +0200 Subject: [PATCH] try a fix for blocking methods. --- tests/mqtt/protocol/test_handler.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/tests/mqtt/protocol/test_handler.py b/tests/mqtt/protocol/test_handler.py index 6d30a60..9aa95b9 100644 --- a/tests/mqtt/protocol/test_handler.py +++ b/tests/mqtt/protocol/test_handler.py @@ -242,17 +242,16 @@ class ProtocolHandlerTest(unittest.TestCase): @asyncio.coroutine def server_mock(reader, writer): try: - yield from self.event.wait() - #self.event.clear() + #yield from self.event.wait() packet = PublishPacket.build('/topic', b'test_data', 1, False, QOS_1, False) yield from packet.to_stream(writer) puback = yield from PubackPacket.from_stream(reader) self.assertIsNotNone(puback) self.assertEqual(packet.packet_id, puback.packet_id) - self.assertEquals(self.session.delivered_message_queue.qsize(), 1) - #yield from self.event.wait() + #self.assertEquals(self.session.delivered_message_queue.qsize(), 1) #writer.close() except Exception as ae: + print(ae) future.set_exception(ae) @asyncio.coroutine @@ -262,14 +261,13 @@ class ProtocolHandlerTest(unittest.TestCase): self.session.reader, self.session.writer = adapt(reader, writer) self.handler = ProtocolHandler(self.session, self.plugin_manager, loop=self.loop) yield from self.start_handler(self.handler, self.session) - self.event.set() +# self.event.set() message = yield from self.handler.mqtt_deliver_next_message() self.assertIsInstance(message, IncomingApplicationMessage) self.assertIsNotNone(message.publish_packet) self.assertIsNotNone(message.puback_packet) yield from self.stop_handler(self.handler, self.session) future.set_result(True) - #self.event.set() except Exception as ae: future.set_exception(ae)