v20 - boolean schemas, false

commonism · Nov 16, 2024 · 2325ec4 · 2325ec4
1 parent bdcb8bc
commit 2325ec4
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 2 deletions.
diff --git a/aiopenapi3/v20/schemas.py b/aiopenapi3/v20/schemas.py
@@ -40,7 +40,7 @@ class Schema(ObjectExtended, SchemaBase):
     items: Optional[Union[list[Union["Schema", Reference]], Union["Schema", Reference]]] = Field(default=None)
     allOf: list[Union["Schema", Reference]] = Field(default_factory=list)
     properties: dict[str, Union["Schema", Reference]] = Field(default_factory=dict)
-    additionalProperties: Optional[Union["Schema", Reference]] = Field(default=None)
+    additionalProperties: Optional[Union["Schema", Reference, "_Not"]] = Field(default=None)
 
     discriminator: Optional[str] = Field(default=None)  # 'Discriminator'
     readOnly: Optional[bool] = Field(default=None)
@@ -60,3 +60,7 @@ def is_boolean_schema(cls, data: Any) -> Any:
 
     def __getstate__(self):
         return SchemaBase.__getstate__(self)
+
+
+class _Not(Schema):
+    not_: Optional["Schema"] = Field(alias="not")
diff --git a/tests/fixtures/schema-boolean-v20.yaml b/tests/fixtures/schema-boolean-v20.yaml
@@ -15,3 +15,7 @@ definitions:
   A:
     type: object
     additionalProperties: true
+
+  B:
+    type: object
+    additionalProperties: false
diff --git a/tests/schema_test.py b/tests/schema_test.py
@@ -757,4 +757,12 @@ def test_schema_boolean(with_schema_boolean):
 
 
 def test_schema_boolean_v20(with_schema_boolean_v20):
-    OpenAPI("/", with_schema_boolean_v20)
+    api = OpenAPI("/", with_schema_boolean_v20)
+
+    A = api._root.definitions["A"].get_type()
+    B = api._root.definitions["B"].get_type()
+
+    A.model_validate({"a": 1})
+
+    with pytest.raises(ValidationError):
+        B.model_validate({"b": 1})