Separate schema casting and validation

p1c2u · p1c2u · commit d6c5c762a79d · 2018-08-17T15:59:24.000+01:00
diff --git a/openapi_core/schema/media_types/models.py b/openapi_core/schema/media_types/models.py
@@ -42,6 +42,11 @@ def unmarshal(self, value):
             raise InvalidMediaTypeValue(str(exc))
 
         try:
-            return self.schema.unmarshal(deserialized)
+            unmarshalled = self.schema.unmarshal(deserialized)
+        except InvalidSchemaValue as exc:
+            raise InvalidMediaTypeValue(str(exc))
+
+        try:
+            return self.schema.validate(unmarshalled)
         except InvalidSchemaValue as exc:
             raise InvalidMediaTypeValue(str(exc))
diff --git a/openapi_core/schema/parameters/models.py b/openapi_core/schema/parameters/models.py
@@ -112,6 +112,11 @@ def unmarshal(self, value):
             raise InvalidParameterValue(str(exc))
 
         try:
-            return self.schema.unmarshal(deserialized)
+            unmarshalled = self.schema.unmarshal(deserialized)
+        except InvalidSchemaValue as exc:
+            raise InvalidParameterValue(str(exc))
+
+        try:
+            return self.schema.validate(unmarshalled)
         except InvalidSchemaValue as exc:
             raise InvalidParameterValue(str(exc))
diff --git a/openapi_core/schema/responses/models.py b/openapi_core/schema/responses/models.py
@@ -17,6 +17,9 @@ def __init__(
         self.links = links and dict(links) or {}
 
     def __getitem__(self, mimetype):
+        return self.get_content_type(mimetype)
+
+    def get_content_type(self, mimetype):
         try:
             return self.content[mimetype]
         except MimeTypeNotFound:
diff --git a/openapi_core/schema/schemas/models.py b/openapi_core/schema/schemas/models.py
@@ -3,7 +3,7 @@
 from collections import defaultdict
 import warnings
 
-from six import iteritems
+from six import iteritems, integer_types, binary_type, text_type
 
 from openapi_core.extensions.models.factories import ModelFactory
 from openapi_core.schema.schemas.enums import SchemaFormat, SchemaType
@@ -12,6 +12,9 @@
     OpenAPISchemaError, NoOneOfSchema, MultipleOneOfSchema,
 )
 from openapi_core.schema.schemas.util import forcebool, format_date
+from openapi_core.schema.schemas.validators import (
+    TypeValidator, AttributeValidator,
+)
 
 log = logging.getLogger(__name__)
 
@@ -29,6 +32,16 @@ class Schema(object):
         SchemaFormat.DATE.value: format_date,
     })
 
+    VALIDATOR_CALLABLE_GETTER = {
+        None: lambda x: x,
+        SchemaType.BOOLEAN: TypeValidator(bool),
+        SchemaType.INTEGER: TypeValidator(integer_types),
+        SchemaType.NUMBER: TypeValidator(integer_types, float),
+        SchemaType.STRING: TypeValidator(binary_type, text_type),
+        SchemaType.ARRAY: TypeValidator(list, tuple),
+        SchemaType.OBJECT: AttributeValidator('__class__'),
+    }
+
     def __init__(
             self, schema_type=None, model=None, properties=None, items=None,
             schema_format=None, required=None, default=None, nullable=False,
@@ -222,3 +235,19 @@ def _unmarshal_properties(self, value, one_of_schema=None):
                 prop_value = prop.default
             properties[prop_name] = prop.unmarshal(prop_value)
         return properties
+
+    def validate(self, value):
+        if value is None:
+            if not self.nullable:
+                raise InvalidSchemaValue("Null value for non-nullable schema")
+            return self.default
+
+        validator = self.VALIDATOR_CALLABLE_GETTER[self.type]
+
+        if not validator(value):
+            raise InvalidSchemaValue(
+                "Value of {0} not valid type of {1}".format(
+                    value, self.type.value)
+            )
+
+        return value
diff --git a/openapi_core/schema/schemas/validators.py b/openapi_core/schema/schemas/validators.py
@@ -0,0 +1,22 @@
+class TypeValidator(object):
+
+    def __init__(self, *types):
+        self.types = types
+
+    def __call__(self, value):
+        if not isinstance(value, self.types):
+            return False
+
+        return True
+
+
+class AttributeValidator(object):
+
+    def __init__(self, attribute):
+        self.attribute = attribute
+
+    def __call__(self, value):
+        if not hasattr(value, self.attribute):
+            return False
+
+        return True
