fixes tests

Author: pcrespov

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

@@ -110,7 +110,7 @@ def fake_service_details(mocks_dir: Path) -> ServiceMetaDataPublished:
 
 @pytest.fixture
 def fake_service_extras() -> ServiceExtras:
-    extra_example = ServiceExtras.model_config["json_schema_extra"]["examples"][2]  # type: ignore
+    extra_example = ServiceExtras.model_json_schema()["examples"][2]  # type: ignore
     random_extras = ServiceExtras(**extra_example)  # type: ignore
     assert random_extras is not None
     return random_extras
@@ -281,14 +281,14 @@ def _mocked_services_details(
 
 
 assert "json_schema_extra" in RutPricingPlanGet.model_config
-assert isinstance(RutPricingPlanGet.model_config["json_schema_extra"], dict)
-assert isinstance(RutPricingPlanGet.model_config["json_schema_extra"]["examples"], list)
+assert isinstance(RutPricingPlanGet.model_json_schema(), dict)
+assert isinstance(RutPricingPlanGet.model_json_schema()["examples"], list)
 
 
 @pytest.fixture(
     params=[
-        RutPricingPlanGet.model_config["json_schema_extra"]["examples"][0],
-        RutPricingPlanGet.model_config["json_schema_extra"]["examples"][1],
+        RutPricingPlanGet.model_json_schema()["examples"][0],
+        RutPricingPlanGet.model_json_schema()["examples"][1],
     ],
     ids=["with ec2 restriction", "without"],
 )
@@ -329,19 +329,15 @@ def _mocked_service_default_pricing_plan(
 
     def _mocked_get_pricing_unit(request, pricing_plan_id: int) -> httpx.Response:
         assert "json_schema_extra" in RutPricingUnitGet.model_config
-        assert isinstance(RutPricingUnitGet.model_config["json_schema_extra"], dict)
-        assert isinstance(
-            RutPricingUnitGet.model_config["json_schema_extra"]["examples"], list
-        )
+        assert isinstance(RutPricingUnitGet.model_json_schema(), dict)
+        assert isinstance(RutPricingUnitGet.model_json_schema()["examples"], list)
         return httpx.Response(
             200,
             json=jsonable_encoder(
                 (
                     default_pricing_plan.pricing_units[0]
                     if default_pricing_plan.pricing_units
-                    else RutPricingUnitGet.model_config["json_schema_extra"][
-                        "examples"
-                    ][0]
+                    else RutPricingUnitGet.model_json_schema()["examples"][0]
                 ),
                 by_alias=True,
             ),
@@ -596,7 +592,7 @@ async def test_create_computation_with_wallet(
     "default_pricing_plan",
     [
         RutPricingPlanGet.model_validate(
-            RutPricingPlanGet.model_config["json_schema_extra"]["examples"][0]
+            RutPricingPlanGet.model_json_schema()["examples"][0]
         )
     ],
 )
@@ -638,7 +634,7 @@ async def test_create_computation_with_wallet_with_invalid_pricing_unit_name_rai
     "default_pricing_plan",
     [
         RutPricingPlanGet(
-            **RutPricingPlanGet.model_config["json_schema_extra"]["examples"][0]  # type: ignore
+            **RutPricingPlanGet.model_json_schema()["examples"][0]  # type: ignore
         )
     ],
 )