Merge pull request #64 from danjones1618/bugfix/null-parsing

Add NULL type parsing

Author: manchenkoff

src/openapi_parser/builders/schema.py

@@ -1,14 +1,28 @@
 import logging
 from typing import Any, Callable, Dict
 
-from .common import extract_extension_attributes, extract_typed_props, merge_schema, PropertyMeta
 from ..enumeration import DataType, IntegerFormat, NumberFormat, StringFormat
 from ..errors import ParserError
-from ..specification import AnyOf, Array, Boolean, Discriminator, Integer, Number, Object, OneOf, Property, Schema, String
-from ..loose_types import (
-    LooseIntegerFormat,
-    LooseNumberFormat,
-    LooseStringFormat,
+from ..loose_types import LooseIntegerFormat, LooseNumberFormat, LooseStringFormat
+from ..specification import (
+    AnyOf,
+    Array,
+    Boolean,
+    Discriminator,
+    Integer,
+    Null,
+    Number,
+    Object,
+    OneOf,
+    Property,
+    Schema,
+    String,
+)
+from .common import (
+    PropertyMeta,
+    extract_extension_attributes,
+    extract_typed_props,
+    merge_schema,
 )
 
 SchemaBuilderMethod = Callable[[dict], Schema]
@@ -89,6 +103,7 @@ class SchemaFactory:
     def __init__(self, strict_enum: bool = True) -> None:
         self.strict_enum = strict_enum
         self._builders = {
+            DataType.NULL: self._null,
             DataType.INTEGER: self._integer,
             DataType.NUMBER: self._number,
             DataType.STRING: self._string,
@@ -128,6 +143,9 @@ def create(self, data: dict) -> Schema:
 
         return builder_func(data)
 
+    def _null(self, data: dict) -> Null:
+        return Null(**extract_attrs(data, {}))
+
     def _integer(self, data: dict) -> Integer:
         format_cast = IntegerFormat if self.strict_enum else LooseIntegerFormat
         attrs_map = {

src/openapi_parser/enumeration.py

@@ -3,6 +3,7 @@
 
 @unique
 class DataType(Enum):
+    NULL = 'null'
     INTEGER = 'integer'
     NUMBER = 'number'
     STRING = 'string'

src/openapi_parser/specification.py

@@ -94,6 +94,11 @@ class String(Schema):
     format: Optional[Union[StringFormat, LooseStringFormat]] = None
 
 
+@dataclass
+class Null(Schema):
+    pass
+
+
 @dataclass
 class Boolean(Schema):
     pass

tests/builders/schema/test_null.py

@@ -0,0 +1,19 @@
+import pytest
+
+from openapi_parser.builders.schema import SchemaFactory
+from openapi_parser.specification import DataType, Null
+
+data_provider = (
+    (
+        {
+            "type": "null",
+        },
+        Null(type=DataType.NULL)
+    ),
+)
+
+
+@pytest.mark.parametrize(['data', 'expected'], data_provider)
+def test_boolean_builder(data: dict, expected: Null) -> None:
+    factory = SchemaFactory()
+    assert expected == factory.create(data)