reaplce some more assertEqual

pull/19/head
Florian Ludwig 2021-03-06 18:59:19 +01:00
rodzic 7c56c9caba
commit f439ccd1b1
1 zmienionych plików z 17 dodań i 17 usunięć

Wyświetl plik

@ -20,8 +20,8 @@ class PublishPacketTest(unittest.TestCase):
self.assertEqual(message.variable_header.topic_name, 'topic')
self.assertEqual(message.variable_header.packet_id, None)
self.assertFalse((message.fixed_header.flags >> 1) & 0x03)
self.assertTrue(message.fixed_header.flags & 0x01)
self.assertTrue(message.payload.data, b'0123456789')
assert message.fixed_header.flags & 0x01
assert message.payload.data, b'0123456789'
def test_from_stream_qos_2(self):
data = b'\x37\x13\x00\x05topic\x00\x0a0123456789'
@ -29,9 +29,9 @@ class PublishPacketTest(unittest.TestCase):
message = self.loop.run_until_complete(PublishPacket.from_stream(stream))
self.assertEqual(message.variable_header.topic_name, 'topic')
self.assertEqual(message.variable_header.packet_id, 10)
self.assertTrue((message.fixed_header.flags >> 1) & 0x03)
self.assertTrue(message.fixed_header.flags & 0x01)
self.assertTrue(message.payload.data, b'0123456789')
assert (message.fixed_header.flags >> 1) & 0x03
assert message.fixed_header.flags & 0x01
assert message.payload.data, b'0123456789'
def test_to_stream_no_packet_id(self):
variable_header = PublishVariableHeader('topic', None)
@ -68,19 +68,19 @@ class PublishPacketTest(unittest.TestCase):
packet = PublishPacket.build('/topic', b'data', 1, True, QOS_0, False)
self.assertEqual(packet.packet_id, 1)
self.assertTrue(packet.dup_flag)
assert packet.dup_flag
self.assertEqual(packet.qos, QOS_0)
self.assertFalse(packet.retain_flag)
packet = PublishPacket.build('/topic', b'data', 1, True, QOS_1, False)
self.assertEqual(packet.packet_id, 1)
self.assertTrue(packet.dup_flag)
assert packet.dup_flag
self.assertEqual(packet.qos, QOS_1)
self.assertFalse(packet.retain_flag)
packet = PublishPacket.build('/topic', b'data', 1, True, QOS_2, False)
self.assertEqual(packet.packet_id, 1)
self.assertTrue(packet.dup_flag)
assert packet.dup_flag
self.assertEqual(packet.qos, QOS_2)
self.assertFalse(packet.retain_flag)
@ -88,34 +88,34 @@ class PublishPacketTest(unittest.TestCase):
self.assertEqual(packet.packet_id, 1)
self.assertFalse(packet.dup_flag)
self.assertEqual(packet.qos, QOS_0)
self.assertTrue(packet.retain_flag)
assert packet.retain_flag
packet = PublishPacket.build('/topic', b'data', 1, False, QOS_1, True)
self.assertEqual(packet.packet_id, 1)
self.assertFalse(packet.dup_flag)
self.assertEqual(packet.qos, QOS_1)
self.assertTrue(packet.retain_flag)
assert packet.retain_flag
packet = PublishPacket.build('/topic', b'data', 1, False, QOS_2, True)
self.assertEqual(packet.packet_id, 1)
self.assertFalse(packet.dup_flag)
self.assertEqual(packet.qos, QOS_2)
self.assertTrue(packet.retain_flag)
assert packet.retain_flag
packet = PublishPacket.build('/topic', b'data', 1, True, QOS_0, True)
self.assertEqual(packet.packet_id, 1)
self.assertTrue(packet.dup_flag)
assert packet.dup_flag
self.assertEqual(packet.qos, QOS_0)
self.assertTrue(packet.retain_flag)
assert packet.retain_flag
packet = PublishPacket.build('/topic', b'data', 1, True, QOS_1, True)
self.assertEqual(packet.packet_id, 1)
self.assertTrue(packet.dup_flag)
assert packet.dup_flag
self.assertEqual(packet.qos, QOS_1)
self.assertTrue(packet.retain_flag)
assert packet.retain_flag
packet = PublishPacket.build('/topic', b'data', 1, True, QOS_2, True)
self.assertEqual(packet.packet_id, 1)
self.assertTrue(packet.dup_flag)
assert packet.dup_flag
self.assertEqual(packet.qos, QOS_2)
self.assertTrue(packet.retain_flag)
assert packet.retain_flag