def _default_process_message(self, message): """Map ERROR message to a specific exception.""" if message.type == protocol_pb2.Message.ERROR: error_class = error_to_exception(message.error.type) else: error_class = StorageRequestError self.error(error_class(self, message))
def test_mapping(self): """Protocol's specific exceptions are correct.""" for code_error, proto_error in errors._error_mapping.iteritems(): self.assertTrue(isinstance(proto_error(**REQ_ARGS), errors.StorageRequestError), "%r must inherit from StorageRequestError" % proto_error) self.assertEqual(proto_error, errors.error_to_exception(code_error))