def test_from_wire_bad_input(self): raw_data = 'b'.encode() with self.assertRaises(PayloadParseError): DataPacket.from_wire(raw_data) code, message = (9, 'message ελληνικά'.encode()) raw_data = b''.join((struct.pack('!H', code), message)) with self.assertRaises(InvalidErrorcodeError): ErrorPacket.from_wire(raw_data)
def test_from_wire_good_input(self): code, message = (1, 'message ελληνικά'.encode()) raw_data = b''.join((struct.pack('!H', code), message)) packet = ErrorPacket.from_wire(raw_data) self.assertEqual(code, packet.error_code) self.assertEqual(message, packet.error_msg)