|
1 | 1 | import pytest |
2 | 2 |
|
| 3 | +from openapi_python_client.parser.errors import ParameterError |
| 4 | +from openapi_python_client.parser.properties import Class, Parameters |
| 5 | +from openapi_python_client.parser.properties.schemas import parameter_from_reference |
| 6 | +from openapi_python_client.schema import Parameter, Reference |
| 7 | + |
| 8 | +MODULE_NAME = "openapi_python_client.parser.properties.schemas" |
| 9 | + |
3 | 10 |
|
4 | 11 | def test_class_from_string_default_config(): |
5 | 12 | from openapi_python_client import Config |
@@ -32,3 +39,109 @@ def test_class_from_string(class_override, module_override, expected_class, expe |
32 | 39 | result = Class.from_string(string=ref, config=config) |
33 | 40 | assert result.name == expected_class |
34 | 41 | assert result.module_name == expected_module |
| 42 | + |
| 43 | + |
| 44 | +class TestParameterFromData: |
| 45 | + def test_cannot_parse_parameters_by_reference(self): |
| 46 | + from openapi_python_client.parser.properties import Parameters |
| 47 | + from openapi_python_client.parser.properties.schemas import parameter_from_data |
| 48 | + |
| 49 | + ref = Reference.construct(ref="#/components/parameters/a_param") |
| 50 | + parameters = Parameters() |
| 51 | + param_or_error, new_parameters = parameter_from_data( |
| 52 | + name="a_param", required=True, data=ref, parameters=parameters |
| 53 | + ) |
| 54 | + assert param_or_error == ParameterError("Unable to resolve another reference") |
| 55 | + assert new_parameters == parameters |
| 56 | + |
| 57 | + def test_parameters_without_schema_are_ignored(self): |
| 58 | + from openapi_python_client.parser.properties import Parameters |
| 59 | + from openapi_python_client.parser.properties.schemas import parameter_from_data |
| 60 | + from openapi_python_client.schema import ParameterLocation, Schema |
| 61 | + |
| 62 | + param = Parameter(name="a_schemaless_param", param_in=ParameterLocation.QUERY) |
| 63 | + parameters = Parameters() |
| 64 | + param_or_error, new_parameters = parameter_from_data( |
| 65 | + name=param.name, required=param.required, data=param, parameters=parameters |
| 66 | + ) |
| 67 | + assert param_or_error == ParameterError("Parameter has no schema") |
| 68 | + assert new_parameters == parameters |
| 69 | + |
| 70 | + def test_registers_new_parameters(self): |
| 71 | + from openapi_python_client.parser.properties import Parameters |
| 72 | + from openapi_python_client.parser.properties.schemas import parameter_from_data |
| 73 | + from openapi_python_client.schema import ParameterLocation, Schema |
| 74 | + |
| 75 | + param = Parameter.construct(name="a_param", param_in=ParameterLocation.QUERY, param_schema=Schema.construct()) |
| 76 | + parameters = Parameters() |
| 77 | + param_or_error, new_parameters = parameter_from_data( |
| 78 | + name=param.name, required=param.required, data=param, parameters=parameters |
| 79 | + ) |
| 80 | + assert param_or_error == param |
| 81 | + assert new_parameters.classes_by_name[param.name] == param |
| 82 | + |
| 83 | + |
| 84 | +class TestParameterFromReference: |
| 85 | + def test_returns_parameter_if_parameter_provided(self): |
| 86 | + param = Parameter.construct() |
| 87 | + params = Parameters() |
| 88 | + param_or_error = parameter_from_reference(param=param, parameters=params) |
| 89 | + assert param_or_error == param |
| 90 | + |
| 91 | + def test_errors_out_if_reference_not_in_parameters(self): |
| 92 | + ref = Reference.construct(ref="#/components/parameters/a_param") |
| 93 | + class_info = Class(name="a_param", module_name="module_name") |
| 94 | + existing_param = Parameter.construct(name="a_param") |
| 95 | + param_by_ref = Reference.construct(ref="#/components/parameters/another_param") |
| 96 | + params = Parameters( |
| 97 | + classes_by_name={class_info.name: existing_param}, classes_by_reference={ref.ref: existing_param} |
| 98 | + ) |
| 99 | + param_or_error = parameter_from_reference(param=param_by_ref, parameters=params) |
| 100 | + assert param_or_error == ParameterError( |
| 101 | + detail="Reference `/components/parameters/another_param` not found.", |
| 102 | + ) |
| 103 | + |
| 104 | + def test_returns_reference_from_registry(self): |
| 105 | + existing_param = Parameter.construct(name="a_param") |
| 106 | + class_info = Class(name="MyParameter", module_name="module_name") |
| 107 | + params = Parameters( |
| 108 | + classes_by_name={class_info.name: existing_param}, |
| 109 | + classes_by_reference={"/components/parameters/a_param": existing_param}, |
| 110 | + ) |
| 111 | + |
| 112 | + param_by_ref = Reference.construct(ref="#/components/parameters/a_param") |
| 113 | + param_or_error = parameter_from_reference(param=param_by_ref, parameters=params) |
| 114 | + assert param_or_error == existing_param |
| 115 | + |
| 116 | + |
| 117 | +class TestUpdateParametersFromData: |
| 118 | + def test_reports_parameters_with_errors(self, mocker): |
| 119 | + from openapi_python_client.parser.properties.schemas import update_parameters_with_data |
| 120 | + from openapi_python_client.schema import ParameterLocation, Schema |
| 121 | + |
| 122 | + parameters = Parameters() |
| 123 | + param = Parameter.construct(name="a_param", param_in=ParameterLocation.QUERY, param_schema=Schema.construct()) |
| 124 | + parameter_from_data = mocker.patch( |
| 125 | + f"{MODULE_NAME}.parameter_from_data", side_effect=[(ParameterError(), parameters)] |
| 126 | + ) |
| 127 | + ref_path = Reference.construct(ref="#/components/parameters/a_param") |
| 128 | + new_parameters_or_error = update_parameters_with_data(ref_path=ref_path.ref, data=param, parameters=parameters) |
| 129 | + |
| 130 | + parameter_from_data.assert_called_once() |
| 131 | + assert new_parameters_or_error == ParameterError( |
| 132 | + detail="Unable to parse this part of your OpenAPI document: : None", |
| 133 | + header="Unable to parse parameter #/components/parameters/a_param", |
| 134 | + ) |
| 135 | + |
| 136 | + def test_records_references_to_parameters(self, mocker): |
| 137 | + from openapi_python_client.parser.properties.schemas import update_parameters_with_data |
| 138 | + from openapi_python_client.schema import ParameterLocation, Schema |
| 139 | + |
| 140 | + parameters = Parameters() |
| 141 | + param = Parameter.construct(name="a_param", param_in=ParameterLocation.QUERY, param_schema=Schema.construct()) |
| 142 | + parameter_from_data = mocker.patch(f"{MODULE_NAME}.parameter_from_data", side_effect=[(param, parameters)]) |
| 143 | + ref_path = "#/components/parameters/a_param" |
| 144 | + new_parameters = update_parameters_with_data(ref_path=ref_path, data=param, parameters=parameters) |
| 145 | + |
| 146 | + parameter_from_data.assert_called_once() |
| 147 | + assert new_parameters.classes_by_reference[ref_path] == param |
0 commit comments