Example #1
0
 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))