Example #1
0
 def server_mock(reader, writer):
     try:
         packet = PublishPacket.build('/topic', b'test_data', rand_packet_id(), False, QOS_2, False)
         yield from packet.to_stream(writer)
         pubrec = yield from PubrecPacket.from_stream(reader)
         self.assertIsNotNone(pubrec)
         self.assertEqual(packet.packet_id, pubrec.packet_id)
         self.assertIn(packet.packet_id, self.handler._pubrel_waiters)
         pubrel = PubrelPacket.build(packet.packet_id)
         yield from pubrel.to_stream(writer)
         pubcomp = yield from PubcompPacket.from_stream(reader)
         self.assertIsNotNone(pubcomp)
         self.assertEqual(packet.packet_id, pubcomp.packet_id)
     except Exception as ae:
         future.set_exception(ae)
Example #2
0
 def server_mock(reader, writer):
     try:
         packet = PublishPacket.build("/topic", b"test_data", rand_packet_id(), False, QOS_2, False)
         yield from packet.to_stream(writer)
         pubrec = yield from PubrecPacket.from_stream(reader)
         self.assertIsNotNone(pubrec)
         self.assertEqual(packet.packet_id, pubrec.packet_id)
         self.assertIn(packet.packet_id, self.handler._pubrel_waiters)
         pubrel = PubrelPacket.build(packet.packet_id)
         yield from pubrel.to_stream(writer)
         pubcomp = yield from PubcompPacket.from_stream(reader)
         self.assertIsNotNone(pubcomp)
         self.assertEqual(packet.packet_id, pubcomp.packet_id)
     except Exception as ae:
         future.set_exception(ae)
Example #3
0
 def test_from_stream(self):
     data = b'\x50\x02\x00\x0a'
     stream = BufferReader(data)
     message = self.loop.run_until_complete(
         PubrecPacket.from_stream(stream))
     self.assertEqual(message.variable_header.packet_id, 10)
Example #4
0
 def test_from_stream(self):
     data = b"\x50\x02\x00\x0a"
     stream = asyncio.StreamReader(loop=self.loop)
     stream.feed_data(data)
     message = self.loop.run_until_complete(PubrecPacket.from_stream(stream))
     self.assertEqual(message.variable_header.packet_id, 10)
Example #5
0
 def test_from_stream(self):
     data = b'\x50\x02\x00\x0a'
     stream = BufferReader(data)
     message = self.loop.run_until_complete(PubrecPacket.from_stream(stream))
     self.assertEqual(message.variable_header.packet_id, 10)