Skip to content

Commit 80dbbd3

Browse files
remove user warnings
1 parent 5075fe0 commit 80dbbd3

File tree

1 file changed

+28
-8
lines changed

1 file changed

+28
-8
lines changed

packages/pytest-simcore/src/pytest_simcore/services_api_mocks_for_aiohttp_clients.py

Lines changed: 28 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -105,10 +105,20 @@ def create_computation_cb(url, **kwargs) -> CallbackResult:
105105
"62237c33-8d6c-4709-aa92-c3cf693dd6d2",
106106
],
107107
}
108-
returned_computation = ComputationTask.model_validate(
109-
ComputationTask.model_config["json_schema_extra"]["examples"][0]
110-
).model_copy(
111-
update={
108+
109+
assert "json_schema_extra" in ComputationTask.model_config
110+
assert isinstance(ComputationTask.model_config["json_schema_extra"], dict)
111+
assert isinstance(
112+
ComputationTask.model_config["json_schema_extra"]["examples"], list
113+
)
114+
assert isinstance(
115+
ComputationTask.model_config["json_schema_extra"]["examples"][0], dict
116+
)
117+
computation: dict[str, Any] = ComputationTask.model_config["json_schema_extra"][
118+
"examples"
119+
][0].copy()
120+
computation.update(
121+
{
112122
"id": f"{kwargs['json']['project_id']}",
113123
"state": state,
114124
"pipeline_details": {
@@ -118,6 +128,10 @@ def create_computation_cb(url, **kwargs) -> CallbackResult:
118128
},
119129
}
120130
)
131+
returned_computation = ComputationTask.model_validate(
132+
computation
133+
)
134+
121135
return CallbackResult(
122136
status=201,
123137
# NOTE: aioresponses uses json.dump which does NOT encode serialization of UUIDs
@@ -134,10 +148,15 @@ def get_computation_cb(url, **kwargs) -> CallbackResult:
134148
assert isinstance(
135149
ComputationGet.model_config["json_schema_extra"]["examples"], list
136150
)
137-
returned_computation = ComputationGet.model_validate(
138-
ComputationGet.model_config["json_schema_extra"]["examples"][0]
139-
).model_copy(
140-
update={
151+
assert isinstance(
152+
ComputationGet.model_config["json_schema_extra"]["examples"][0], dict
153+
)
154+
155+
computation: dict[str, Any] = ComputationGet.model_config["json_schema_extra"][
156+
"examples"
157+
][0].copy()
158+
computation.update(
159+
{
141160
"id": Path(url.path).name,
142161
"state": state,
143162
"pipeline_details": {
@@ -147,6 +166,7 @@ def get_computation_cb(url, **kwargs) -> CallbackResult:
147166
},
148167
}
149168
)
169+
returned_computation = ComputationGet.model_validate(computation)
150170

151171
return CallbackResult(
152172
status=200,

0 commit comments

Comments
 (0)