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)
