type annotations

Author: pcrespov

services/api-server/tests/unit/api_solvers/test_api_routers_solvers_jobs.py

@@ -27,7 +27,7 @@
 
 
 @pytest.fixture
-def bucket_name():
+def bucket_name() -> str:
     return "test-bucket"
 
 
@@ -87,7 +87,7 @@ def mocked_directorv2_service_api(
     presigned_download_link: AnyUrl,
     mocked_directorv2_rest_api_base: MockRouter,
     directorv2_service_openapi_specs: dict[str, Any],
-):
+) -> MockRouter:
     settings: ApplicationSettings = app.state.settings
     assert settings.API_SERVER_DIRECTOR_V2
     oas = directorv2_service_openapi_specs

services/api-server/tests/unit/test_api_solver_jobs.py

@@ -21,6 +21,7 @@
     HttpApiCallCaptureModel,
     SideEffectCallback,
 )
+from respx import MockRouter
 from simcore_service_api_server._meta import API_VTAG
 from simcore_service_api_server.models.schemas.jobs import Job, JobStatus
 from simcore_service_api_server.models.schemas.model_adapter import (
@@ -59,7 +60,7 @@ def _inspect_job_side_effect(
 )
 async def test_get_solver_job_wallet(
     client: AsyncClient,
-    mocked_webserver_rest_api_base,
+    mocked_webserver_rest_api_base: MockRouter,
     create_respx_mock_from_capture: CreateRespxMockCallback,
     auth: httpx.BasicAuth,
     project_tests_dir: Path,
@@ -131,7 +132,7 @@ def _get_wallet_side_effect(
 )
 async def test_get_solver_job_pricing_unit(
     client: AsyncClient,
-    mocked_webserver_rest_api_base,
+    mocked_webserver_rest_api_base: MockRouter,
     create_respx_mock_from_capture: CreateRespxMockCallback,
     auth: httpx.BasicAuth,
     project_tests_dir: Path,
@@ -202,8 +203,8 @@ def _get_pricing_unit_side_effect(
 )
 async def test_start_solver_job_pricing_unit_with_payment(
     client: AsyncClient,
-    mocked_webserver_rest_api_base,
-    mocked_directorv2_rest_api_base,
+    mocked_webserver_rest_api_base: MockRouter,
+    mocked_directorv2_rest_api_base: MockRouter,
     create_respx_mock_from_capture: CreateRespxMockCallback,
     auth: httpx.BasicAuth,
     project_tests_dir: Path,
@@ -277,8 +278,8 @@ def _put_pricing_plan_and_unit_side_effect(
 
 async def test_get_solver_job_pricing_unit_no_payment(
     client: AsyncClient,
-    mocked_webserver_rest_api_base,
-    mocked_directorv2_rest_api_base,
+    mocked_webserver_rest_api_base: MockRouter,
+    mocked_directorv2_rest_api_base: MockRouter,
     create_respx_mock_from_capture: CreateRespxMockCallback,
     auth: httpx.BasicAuth,
     project_tests_dir: Path,
@@ -310,8 +311,8 @@ async def test_get_solver_job_pricing_unit_no_payment(
 
 async def test_start_solver_job_conflict(
     client: AsyncClient,
-    mocked_webserver_rest_api_base,
-    mocked_directorv2_rest_api_base,
+    mocked_webserver_rest_api_base: MockRouter,
+    mocked_directorv2_rest_api_base: MockRouter,
     create_respx_mock_from_capture: CreateRespxMockCallback,
     auth: httpx.BasicAuth,
     project_tests_dir: Path,
@@ -344,7 +345,7 @@ async def test_start_solver_job_conflict(
 
 async def test_stop_job(
     client: AsyncClient,
-    mocked_directorv2_rest_api_base,
+    mocked_directorv2_rest_api_base: MockRouter,
     create_respx_mock_from_capture: CreateRespxMockCallback,
     auth: httpx.BasicAuth,
     project_tests_dir: Path,
@@ -389,9 +390,9 @@ def _stop_job_side_effect(
 )
 async def test_get_solver_job_outputs(
     client: AsyncClient,
-    mocked_webserver_rest_api_base,
-    mocked_storage_rest_api_base,
-    mocked_solver_job_outputs,
+    mocked_webserver_rest_api_base: MockRouter,
+    mocked_storage_rest_api_base: MockRouter,
+    mocked_solver_job_outputs: None,
     create_respx_mock_from_capture: CreateRespxMockCallback,
     auth: httpx.BasicAuth,
     project_tests_dir: Path,