def test_set_packet(self):
message = Message()
message.set_packet(Message.LIST)
self.assertEqual(message.get_packet(), Message.LIST)
message.set_packet(Message.MEMBER)
self.assertEqual(message.get_packet(), Message.MEMBER)
message.set_packet(Message.TRANSACTION)
self.assertEqual(message.get_packet(), Message.TRANSACTION)
message.set_packet(Message.CHEESE)
self.assertEqual(message.get_packet(), Message.CHEESE)
message.set_packet(Message.RESPONSE)
self.assertEqual(message.get_packet(), Message.NOTHING)
message.set_packet(Message.REQUEST)
self.assertEqual(message.get_packet(), Message.NOTHING)
message.set_packet(Message.REPORT)
self.assertEqual(message.get_packet(), Message.NOTHING)
message.set_packet(Message.ERROR)
self.assertEqual(message.get_packet(), Message.NOTHING)
def test_get_packet(self):
message = Message()
self.assertEqual(message.packet, message.get_packet())