def test_bad_header_direction(self, *args): c = self.make_connection() c._callbacks = Mock() c.defunct = Mock() # read in a SupportedMessage response header = six.binary_type().join(uint8_pack(i) for i in ( 0xff & (HEADER_DIRECTION_FROM_CLIENT | self.protocol_version), 0, # flags (compression) 0, SupportedMessage.opcode # opcode )) options = self.make_options_body() message = self.make_msg(header, options) c.process_msg(message, len(message) - 8) # make sure it errored correctly c.defunct.assert_called_once_with(ANY) args, kwargs = c.defunct.call_args self.assertIsInstance(args[0], ProtocolError)
def write_byte(f, b): f.write(uint8_pack(b))