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)
def test_to_wire(self): opcode, code, message = (ErrorPacket.opcode, 1, 'message ελληνικά'.encode()) raw_data = b''.join((struct.pack('!HH', opcode, code), message, b'\x00')) packet = ErrorPacket(code, message) self.assertEqual(packet.to_wire(), raw_data)