def server_mock(reader, writer): try: packet = PublishPacket.build('/topic', b'test_data', rand_packet_id(), 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) except Exception as ae: print(ae) future.set_exception(ae)
def server_mock(reader, writer): try: packet = PublishPacket.build("/topic", b"test_data", rand_packet_id(), 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) except Exception as ae: print(ae) future.set_exception(ae)
def test_from_stream(self): data = b'\x40\x02\x00\x0a' stream = BufferReader(data) message = self.loop.run_until_complete( PubackPacket.from_stream(stream)) self.assertEqual(message.variable_header.packet_id, 10)
def test_from_stream(self): data = b'\x40\x02\x00\x0a' stream = asyncio.StreamReader(loop=self.loop) stream.feed_data(data) message = self.loop.run_until_complete(PubackPacket.from_stream(stream)) self.assertEqual(message.variable_header.packet_id, 10)
def test_from_stream(self): data = b'\x40\x02\x00\x0a' stream = BufferReader(data) message = self.loop.run_until_complete(PubackPacket.from_stream(stream)) self.assertEqual(message.variable_header.packet_id, 10)