Skip to content

Commit 036f372

Browse files
committed
Parametrize test_build_model_property to test various additional_properties values
1 parent daf9628 commit 036f372

File tree

1 file changed

+16
-3
lines changed

1 file changed

+16
-3
lines changed

tests/test_parser/test_properties/test_init.py

Lines changed: 16 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1041,7 +1041,20 @@ def test_build_enums(mocker):
10411041
build_model_property.assert_not_called()
10421042

10431043

1044-
def test_build_model_property():
1044+
@pytest.mark.parametrize(
1045+
"additional_properties_schema, expected_additional_properties",
1046+
[
1047+
(True, True),
1048+
(oai.Schema.construct(), True),
1049+
(None, True),
1050+
(False, False),
1051+
(
1052+
oai.Schema.construct(type="string"),
1053+
StringProperty(name="AdditionalProperties", required=True, nullable=False, default=None),
1054+
),
1055+
],
1056+
)
1057+
def test_build_model_property(additional_properties_schema, expected_additional_properties):
10451058
from openapi_python_client.parser.properties import Schemas, build_model_property
10461059

10471060
data = oai.Schema.construct(
@@ -1053,7 +1066,7 @@ def test_build_model_property():
10531066
},
10541067
description="A class called MyModel",
10551068
nullable=False,
1056-
additional_properties=oai.Schema.construct(),
1069+
additionalProperties=additional_properties_schema,
10571070
)
10581071
schemas = Schemas(models={"OtherModel": None})
10591072

@@ -1086,7 +1099,7 @@ def test_build_model_property():
10861099
"from ..types import UNSET, Unset",
10871100
"from typing import Union",
10881101
},
1089-
additional_properties=True,
1102+
additional_properties=expected_additional_properties,
10901103
)
10911104

10921105

0 commit comments

Comments
 (0)