fixed tests

ITISFoundation · Nov 1, 2024 · 830c88b · 830c88b
1 parent 9401b7b
commit 830c88b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/services/dynamic-scheduler/tests/unit/service_tracker/test__models.py b/services/dynamic-scheduler/tests/unit/service_tracker/test__models.py
@@ -12,6 +12,7 @@
     DynamicServiceStart,
 )
 from models_library.projects import ProjectID
+from pydantic import TypeAdapter
 from servicelib.deferred_tasks import TaskUID
 from simcore_service_dynamic_scheduler.services.service_tracker._models import (
     SchedulerServiceState,
@@ -51,8 +52,8 @@ def test_serialization(
     "dynamic_service_start",
     [
         None,
-        DynamicServiceStart.parse_obj(
-            DynamicServiceStart.Config.schema_extra["example"]
+        TypeAdapter(DynamicServiceStart).validate_python(
+            DynamicServiceStart.model_config["json_schema_extra"]["example"]
         ),
     ],
 )

diff --git a/services/web/server/tests/unit/isolated/test_dynamic_scheduler.py b/services/web/server/tests/unit/isolated/test_dynamic_scheduler.py
@@ -15,6 +15,7 @@
     NodeGetUnknown,
 )
 from models_library.projects_nodes_io import NodeID
+from pydantic import TypeAdapter
 from pytest_mock import MockerFixture
 from servicelib.rabbitmq import RPCServerError
 from simcore_service_webserver.dynamic_scheduler.api import (
@@ -47,8 +48,10 @@ def mock_rpc_client(
 
 @pytest.fixture
 def dynamic_service_start() -> DynamicServiceStart:
-    return DynamicServiceStart.parse_obj(
-        DynamicServiceStart.Config.schema_extra["example"]
+    return (
+        TypeAdapter(DynamicServiceStart).validate_python(
+            DynamicServiceStart.model_config["json_schema_extra"]["example"]
+        ),
     )