mirror
/
amqtt
kopia lustrzana https://github.com/Yakifo/amqtt
1
0
Forkuj 0
Kod Zgłoszenia Packages Projekty Wydania Wiki Aktywność

try a fix for blocking methods.

pull/8/head
Nico 2015-09-15 23:38:24 +02:00
rodzic 417bb01148
commit eb9defa5cc
1 zmienionych plików z 4 dodań i 6 usunięć
Pokaż statystyki Ściągnij plik aktualizacji Ściągnij plik porównania Expand all files Collapse all files

10
tests/mqtt/protocol/test_handler.py
Wyświetl plik

@ -242,17 +242,16 @@ class ProtocolHandlerTest(unittest.TestCase):
@asyncio.coroutine @asyncio.coroutine
def server_mock(reader, writer): def server_mock(reader, writer):
try: try:
yield from self.event.wait() #yield from self.event.wait()
#self.event.clear()
packet = PublishPacket.build('/topic', b'test_data', 1, False, QOS_1, False) packet = PublishPacket.build('/topic', b'test_data', 1, False, QOS_1, False)
yield from packet.to_stream(writer) yield from packet.to_stream(writer)
puback = yield from PubackPacket.from_stream(reader) puback = yield from PubackPacket.from_stream(reader)
self.assertIsNotNone(puback) self.assertIsNotNone(puback)
self.assertEqual(packet.packet_id, puback.packet_id) self.assertEqual(packet.packet_id, puback.packet_id)
self.assertEquals(self.session.delivered_message_queue.qsize(), 1) #self.assertEquals(self.session.delivered_message_queue.qsize(), 1)
#yield from self.event.wait()
#writer.close() #writer.close()
except Exception as ae: except Exception as ae:
print(ae)
future.set_exception(ae) future.set_exception(ae)
@asyncio.coroutine @asyncio.coroutine
@ -262,14 +261,13 @@ class ProtocolHandlerTest(unittest.TestCase):
self.session.reader, self.session.writer = adapt(reader, writer) self.session.reader, self.session.writer = adapt(reader, writer)
self.handler = ProtocolHandler(self.session, self.plugin_manager, loop=self.loop) self.handler = ProtocolHandler(self.session, self.plugin_manager, loop=self.loop)
yield from self.start_handler(self.handler, self.session) yield from self.start_handler(self.handler, self.session)
self.event.set() # self.event.set()
message = yield from self.handler.mqtt_deliver_next_message() message = yield from self.handler.mqtt_deliver_next_message()
self.assertIsInstance(message, IncomingApplicationMessage) self.assertIsInstance(message, IncomingApplicationMessage)
self.assertIsNotNone(message.publish_packet) self.assertIsNotNone(message.publish_packet)
self.assertIsNotNone(message.puback_packet) self.assertIsNotNone(message.puback_packet)
yield from self.stop_handler(self.handler, self.session) yield from self.stop_handler(self.handler, self.session)
future.set_result(True) future.set_result(True)
#self.event.set()
except Exception as ae: except Exception as ae:
future.set_exception(ae) future.set_exception(ae)