def test_set_packet_type(self): message = Message() message.set_packet_type(Message.LIST) self.assertEqual(message.get_packet_type(), Message.NOTHING) message.set_packet_type(Message.MEMBER) self.assertEqual(message.get_packet_type(), Message.NOTHING) message.set_packet_type(Message.TRANSACTION) self.assertEqual(message.get_packet_type(), Message.NOTHING) message.set_packet_type(Message.CHEESE) self.assertEqual(message.get_packet_type(), Message.NOTHING) message.set_packet_type(Message.RESPONSE) self.assertEqual(message.get_packet_type(), Message.RESPONSE) message.set_packet_type(Message.REQUEST) self.assertEqual(message.get_packet_type(), Message.REQUEST) message.set_packet_type(Message.REPORT) self.assertEqual(message.get_packet_type(), Message.REPORT) message.set_packet_type(Message.ERROR) self.assertEqual(message.get_packet_type(), Message.ERROR)
def test_get_packet_type(self): message = Message() self.assertEqual(message.packet_type, message.get_packet_type())