ValidationErrorWrapper wrapper

Author: p1c2u

openapi_spec_validator/decorators.py

@@ -1,4 +1,5 @@
 """OpenAPI spec validator decorators module."""
+from functools import wraps
 import logging
 
 from openapi_spec_validator.managers import VisitingManager
@@ -43,3 +44,21 @@ def _attach_scope(self, instance):
             return
 
         instance['x-scope'] = list(self.instance_resolver._scopes_stack)
+
+
+class ValidationErrorWrapper(object):
+
+    def __init__(self, error_class):
+        self.error_class = error_class
+
+    def __call__(self, f):
+        @wraps(f)
+        def wrapper(*args, **kwds):
+            errors = f(*args, **kwds)
+            for err in errors:
+                if not isinstance(err, self.error_class):
+                    # wrap other exceptions with library specific version
+                    yield self.error_class.create_from(err)
+                else:
+                    yield err
+        return wrapper

openapi_spec_validator/validators.py

@@ -2,18 +2,20 @@
 import string
 
 from jsonschema.validators import RefResolver
-from jsonschema.exceptions import ValidationError
-from six import iteritems, raise_from
+from six import iteritems
 
 from openapi_spec_validator.exceptions import (
     ParameterDuplicateError, ExtraParametersError, UnresolvableParameterError,
     OpenAPIValidationError
 )
+from openapi_spec_validator.decorators import ValidationErrorWrapper
 from openapi_spec_validator.factories import Draft4ExtendedValidatorFactory
 from openapi_spec_validator.managers import ResolverManager
 
 log = logging.getLogger(__name__)
 
+wraps_errors = ValidationErrorWrapper(OpenAPIValidationError)
+
 
 def is_ref(spec):
     return isinstance(spec, dict) and '$ref' in spec
@@ -43,12 +45,9 @@ def __init__(self, validator_factory, resolver_handlers):
 
     def validate(self, spec, spec_url=''):
         for err in self.iter_errors(spec, spec_url=spec_url):
-            if isinstance(err, ValidationError):
-                # wrap jsonschema exceptions with library specific version
-                raise raise_from(OpenAPIValidationError.create_from(err), err)
-            else:
-                raise err
+            raise err
 
+    @wraps_errors
     def iter_errors(self, spec, spec_url=''):
         spec_resolver = self._get_resolver(spec_url, spec)
         dereferencer = self._get_dereferencer(spec_resolver)
@@ -87,6 +86,7 @@ class ComponentsValidator(object):
     def __init__(self, dereferencer):
         self.dereferencer = dereferencer
 
+    @wraps_errors
     def iter_errors(self, components):
         components_deref = self.dereferencer.dereference(components)
 
@@ -103,6 +103,7 @@ class SchemasValidator(object):
     def __init__(self, dereferencer):
         self.dereferencer = dereferencer
 
+    @wraps_errors
     def iter_errors(self, schemas):
         schemas_deref = self.dereferencer.dereference(schemas)
         for name, schema in iteritems(schemas_deref):
@@ -118,6 +119,7 @@ class SchemaValidator(object):
     def __init__(self, dereferencer):
         self.dereferencer = dereferencer
 
+    @wraps_errors
     def iter_errors(self, schema, require_properties=True):
         schema_deref = self.dereferencer.dereference(schema)
 
@@ -158,6 +160,7 @@ class PathsValidator(object):
     def __init__(self, dereferencer):
         self.dereferencer = dereferencer
 
+    @wraps_errors
     def iter_errors(self, paths):
         paths_deref = self.dereferencer.dereference(paths)
         for url, path_item in iteritems(paths_deref):
@@ -173,6 +176,7 @@ class PathValidator(object):
     def __init__(self, dereferencer):
         self.dereferencer = dereferencer
 
+    @wraps_errors
     def iter_errors(self, url, path_item):
         path_item_deref = self.dereferencer.dereference(path_item)
 
@@ -192,6 +196,7 @@ class PathItemValidator(object):
     def __init__(self, dereferencer):
         self.dereferencer = dereferencer
 
+    @wraps_errors
     def iter_errors(self, url, path_item):
         path_item_deref = self.dereferencer.dereference(path_item)
 
@@ -220,6 +225,7 @@ class OperationValidator(object):
     def __init__(self, dereferencer):
         self.dereferencer = dereferencer
 
+    @wraps_errors
     def iter_errors(self, url, name, operation, path_parameters=None):
         path_parameters = path_parameters or []
         operation_deref = self.dereferencer.dereference(operation)
@@ -261,6 +267,7 @@ class ParametersValidator(object):
     def __init__(self, dereferencer):
         self.dereferencer = dereferencer
 
+    @wraps_errors
     def iter_errors(self, parameters):
         seen = set()
         for parameter in parameters:
@@ -284,6 +291,7 @@ class ParameterValidator(object):
     def __init__(self, dereferencer):
         self.dereferencer = dereferencer
 
+    @wraps_errors
     def iter_errors(self, parameter):
         if 'schema' in parameter:
             schema = parameter['schema']

tests/integration/test_validators.py

@@ -1,7 +1,5 @@
-from jsonschema.exceptions import ValidationError
-
 from openapi_spec_validator.exceptions import (
-    ExtraParametersError, UnresolvableParameterError,
+    ExtraParametersError, UnresolvableParameterError, OpenAPIValidationError,
 )
 
 
@@ -13,11 +11,11 @@ def test_empty(self, validator):
         errors = validator.iter_errors(spec)
 
         errors_list = list(errors)
-        assert errors_list[0].__class__ == ValidationError
+        assert errors_list[0].__class__ == OpenAPIValidationError
         assert errors_list[0].message == "'openapi' is a required property"
-        assert errors_list[1].__class__ == ValidationError
+        assert errors_list[1].__class__ == OpenAPIValidationError
         assert errors_list[1].message == "'info' is a required property"
-        assert errors_list[2].__class__ == ValidationError
+        assert errors_list[2].__class__ == OpenAPIValidationError
         assert errors_list[2].message == "'paths' is a required property"
 
     def test_info_empty(self, validator):
@@ -30,7 +28,7 @@ def test_info_empty(self, validator):
         errors = validator.iter_errors(spec)
 
         errors_list = list(errors)
-        assert errors_list[0].__class__ == ValidationError
+        assert errors_list[0].__class__ == OpenAPIValidationError
         assert errors_list[0].message == "'title' is a required property"
 
     def test_minimalistic(self, validator):
@@ -200,7 +198,7 @@ def test_default_value_wrong_type(self, validator):
 
         errors_list = list(errors)
         assert len(errors_list) == 1
-        assert errors_list[0].__class__ == ValidationError
+        assert errors_list[0].__class__ == OpenAPIValidationError
         assert errors_list[0].message == (
             "'invaldtype' is not of type 'integer'"
         )
@@ -235,7 +233,7 @@ def test_parameter_default_value_wrong_type(self, validator):
 
         errors_list = list(errors)
         assert len(errors_list) == 1
-        assert errors_list[0].__class__ == ValidationError
+        assert errors_list[0].__class__ == OpenAPIValidationError
         assert errors_list[0].message == (
             "'invaldtype' is not of type 'integer'"
         )