def test_with_different_message_types(self): msg_list = [messages.Message(20, bytearray(b"\x20")), messages.Message(30, bytearray(b"\x10"))] msg_gen = FlushMessageList(msg_list) with self.assertRaises(AssertionError): msg_gen.generate(None)
def test_with_different_message_types(self): msg_list = [ messages.Message(20, bytearray(b'\x20')), messages.Message(30, bytearray(b'\x10')) ] msg_gen = FlushMessageList(msg_list) with self.assertRaises(AssertionError): msg_gen.generate(None)
def test_with_message_list(self): msg_list = [] msg_gen = FlushMessageList(msg_list) self.assertEqual(msg_list, []) msg_list.append(messages.Message(20, bytearray(b'\x20\x30'))) msg_list.append(messages.Message(20, bytearray(b'\x60\x70'))) msg = msg_gen.generate(None) self.assertEqual(msg.contentType, 20) self.assertEqual(msg.write(), bytearray(b'\x20\x30\x60\x70')) self.assertEqual(msg_list, [])