Skip to content

Commit 71dc461

Browse files
committed
dv2 fix
1 parent 4e3603b commit 71dc461

File tree

1 file changed

+2
-9
lines changed

1 file changed

+2
-9
lines changed

services/director-v2/tests/unit/with_dbs/comp_scheduler/test_api_route_computations.py

Lines changed: 2 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -169,22 +169,15 @@ def _mocked_service_resources(request) -> httpx.Response:
169169
def _mocked_services_details(
170170
request, service_key: str, service_version: str
171171
) -> httpx.Response:
172-
assert "json_schema_extra" in ServiceGet.model_config
173-
assert isinstance(ServiceGet.model_config["json_schema_extra"], dict)
174-
assert isinstance(
175-
ServiceGet.model_config["json_schema_extra"]["examples"], list
176-
)
177-
assert isinstance(
178-
ServiceGet.model_config["json_schema_extra"]["examples"][0], dict
179-
)
172+
180173
data_published = fake_service_details.model_copy(
181174
update={
182175
"key": urllib.parse.unquote(service_key),
183176
"version": service_version,
184177
}
185178
).model_dump(by_alias=True)
186179
data = {
187-
**ServiceGet.model_config["json_schema_extra"]["examples"][0],
180+
**ServiceGet.model_json_schema()["examples"][0],
188181
**data_published,
189182
}
190183
payload = ServiceGet.model_validate(data)

0 commit comments

Comments
 (0)