Skip to content

Commit 830c88b

Browse files
author
Andrei Neagu
committed
fixed tests
1 parent 9401b7b commit 830c88b

File tree

2 files changed

+8
-4
lines changed

2 files changed

+8
-4
lines changed

services/dynamic-scheduler/tests/unit/service_tracker/test__models.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
DynamicServiceStart,
1313
)
1414
from models_library.projects import ProjectID
15+
from pydantic import TypeAdapter
1516
from servicelib.deferred_tasks import TaskUID
1617
from simcore_service_dynamic_scheduler.services.service_tracker._models import (
1718
SchedulerServiceState,
@@ -51,8 +52,8 @@ def test_serialization(
5152
"dynamic_service_start",
5253
[
5354
None,
54-
DynamicServiceStart.parse_obj(
55-
DynamicServiceStart.Config.schema_extra["example"]
55+
TypeAdapter(DynamicServiceStart).validate_python(
56+
DynamicServiceStart.model_config["json_schema_extra"]["example"]
5657
),
5758
],
5859
)

services/web/server/tests/unit/isolated/test_dynamic_scheduler.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
NodeGetUnknown,
1616
)
1717
from models_library.projects_nodes_io import NodeID
18+
from pydantic import TypeAdapter
1819
from pytest_mock import MockerFixture
1920
from servicelib.rabbitmq import RPCServerError
2021
from simcore_service_webserver.dynamic_scheduler.api import (
@@ -47,8 +48,10 @@ def mock_rpc_client(
4748

4849
@pytest.fixture
4950
def dynamic_service_start() -> DynamicServiceStart:
50-
return DynamicServiceStart.parse_obj(
51-
DynamicServiceStart.Config.schema_extra["example"]
51+
return (
52+
TypeAdapter(DynamicServiceStart).validate_python(
53+
DynamicServiceStart.model_config["json_schema_extra"]["example"]
54+
),
5255
)
5356

5457

0 commit comments

Comments
 (0)