test: test alt json decoder

Author: jgoodson

tests/test_config.py

@@ -7,7 +7,7 @@
 import pytest
 from ruamel.yaml import YAML as _YAML
 
-from openapi_python_client.config import ConfigFile
+from openapi_python_client.config import Config, ConfigFile, JSONDecoder, MetaType
 
 
 class YAML(_YAML):
@@ -59,3 +59,22 @@ def test_load_from_path(tmp_path: Path, filename, dump, relative) -> None:
     assert config.project_name_override == "project-name"
     assert config.package_name_override == "package_name"
     assert config.package_version_override == "package_version"
+
+
+@pytest.mark.parametrize("json", [None, "orjson", "ujson"])
+def test_config_with_alt_json(json):
+    config = Config.from_sources(
+        ConfigFile(alt_json_decoder=json),
+        MetaType.POETRY,
+        document_source=Path("openapi.yaml"),
+        file_encoding="utf-8",
+        overwrite=False,
+        output_path=None,
+    )
+    config.alt_json_decoder = json
+    if json is None:
+            assert config.alt_json_decoder is None
+    elif json == "orjson":
+            assert config.alt_json_decoder == JSONDecoder.ORJSON
+    elif json == "ujson":
+            assert config.alt_json_decoder == JSONDecoder.UJSON

tests/test_parser/test_responses.py

@@ -3,10 +3,12 @@
 import pytest
 
 import openapi_python_client.schema as oai
+from openapi_python_client.config import JSONDecoder
 from openapi_python_client.parser import responses
 from openapi_python_client.parser.errors import ParseError, PropertyError
 from openapi_python_client.parser.properties import Schemas
 from openapi_python_client.parser.responses import (
+    ALT_JSON_SOURCE,
     JSON_SOURCE,
     NONE_SOURCE,
     HTTPStatusPattern,
@@ -184,6 +186,43 @@ def test_response_from_data_reference(mocker, any_property_factory):
     )
 
 
+def test_response_with_alt_decoder(mocker, any_property_factory):
+    prop = any_property_factory()
+    property_from_data = mocker.patch.object(responses, "property_from_data", return_value=(prop, Schemas()))
+    data = oai.Response.model_construct(
+        description="",
+        content={"application/json": oai.MediaType.model_construct(media_type_schema="something")},
+    )
+    config = MagicMock()
+    config.content_type_overrides = {}
+    config.alt_json_decoder = JSONDecoder.ORJSON
+    status_code = HTTPStatusPattern(pattern="400", code_range=(400, 400))
+
+    response, schemas = responses.response_from_data(
+        status_code=status_code,
+        data=data,
+        schemas=Schemas(),
+        responses={},
+        parent_name="parent",
+        config=config,
+    )
+
+    assert response == responses.Response(
+        status_code=status_code,
+        prop=prop,
+        source=ALT_JSON_SOURCE,
+        data=data,
+    )
+    property_from_data.assert_called_once_with(
+        name="response_400",
+        required=True,
+        data="something",
+        schemas=Schemas(),
+        parent_name="parent",
+        config=config,
+    )
+
+
 @pytest.mark.parametrize(
     "ref_string,expected_error_string",
     [