Skip to content

Commit 8a52ec3

Browse files
committed
fix unit api-server tests
1 parent f80853a commit 8a52ec3

File tree

5 files changed

+19
-24
lines changed

5 files changed

+19
-24
lines changed

services/api-server/src/simcore_service_api_server/core/application.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,12 +10,10 @@
1010

1111
from .. import exceptions
1212
from .._meta import API_VERSION, API_VTAG, APP_NAME
13-
from ..api.dependencies.rabbitmq import get_rabbitmq_rpc_client
1413
from ..api.root import create_router
1514
from ..api.routes.health import router as health_router
1615
from ..services_http import catalog, director_v2, storage, webserver
1716
from ..services_http.rabbitmq import setup_rabbitmq
18-
from ..services_rpc import wb_api_server
1917
from ._prometheus_instrumentation import setup_prometheus_instrumentation
2018
from .events import create_start_app_handler, create_stop_app_handler
2119
from .openapi import override_openapi_method, use_route_names_as_operation_ids
@@ -94,7 +92,6 @@ def init_app(settings: ApplicationSettings | None = None) -> FastAPI:
9492
settings.API_SERVER_WEBSERVER,
9593
tracing_settings=settings.API_SERVER_TRACING,
9694
)
97-
wb_api_server.setup(app, get_rabbitmq_rpc_client(app))
9895

9996
if settings.API_SERVER_CATALOG:
10097
catalog.setup(

services/api-server/src/simcore_service_api_server/services_http/rabbitmq.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,8 @@
44
from servicelib.rabbitmq import RabbitMQClient, wait_till_rabbitmq_responsive
55
from servicelib.rabbitmq._client_rpc import RabbitMQRPCClient
66
from settings_library.rabbit import RabbitSettings
7+
from simcore_service_api_server.api.dependencies.rabbitmq import get_rabbitmq_rpc_client
8+
from simcore_service_api_server.services_rpc import wb_api_server
79

810
from ..core.health_checker import ApiServerHealthChecker
911
from ..services_http.log_streaming import LogDistributor
@@ -36,6 +38,7 @@ async def _on_startup() -> None:
3638
await app.state.health_checker.setup(
3739
app.state.settings.API_SERVER_HEALTH_CHECK_TASK_PERIOD_SECONDS
3840
)
41+
wb_api_server.setup(app, get_rabbitmq_rpc_client(app))
3942

4043
async def _on_shutdown() -> None:
4144
if app.state.health_checker:

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@ def fake_project_for_streaming(
7979
fake_project = ProjectGet.model_validate(data)
8080
fake_project.workbench = {faker.uuid4(): faker.uuid4()}
8181
mocker.patch(
82-
"simcore_service_api_server.api.dependencies.webserver.AuthSession.get_project",
82+
"simcore_service_api_server.api.dependencies.webserver_http.AuthSession.get_project",
8383
return_value=fake_project,
8484
)
8585

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

Lines changed: 14 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
import asyncio
66

77
import pytest
8-
from fastapi import status
8+
from fastapi import FastAPI, status
99
from httpx import AsyncClient, BasicAuth
1010
from models_library.api_schemas_webserver.licensed_items import (
1111
LicensedItemGet as _LicensedItemGet,
@@ -18,28 +18,17 @@
1818
from servicelib.rabbitmq._client_rpc import RabbitMQRPCClient
1919
from servicelib.rabbitmq._errors import RemoteMethodNotRegisteredError
2020
from simcore_service_api_server._meta import API_VTAG
21+
from simcore_service_api_server.api.dependencies.webserver_rpc import (
22+
get_wb_api_rpc_client,
23+
)
2124
from simcore_service_api_server.models.pagination import Page
2225
from simcore_service_api_server.models.schemas.model_adapter import LicensedItemGet
23-
24-
25-
@pytest.fixture
26-
async def mock_rabbitmq_rpc_client(mocker: MockerFixture) -> MockerFixture:
27-
class DummyRabbitMqRpcClient:
28-
pass
29-
30-
def _get_dummy_rpc_client():
31-
return DummyRabbitMqRpcClient
32-
33-
mocker.patch(
34-
"simcore_service_api_server.core.application.get_rabbitmq_rpc_client",
35-
sideeffect=_get_dummy_rpc_client,
36-
)
37-
return mocker
26+
from simcore_service_api_server.services_rpc.wb_api_server import WbApiRpcClient
3827

3928

4029
@pytest.fixture
4130
async def mock_wb_api_server_rcp(
42-
mock_rabbitmq_rpc_client: MockerFixture, exception_to_raise: Exception | None
31+
app: FastAPI, mocker: MockerFixture, exception_to_raise: Exception | None
4332
) -> MockerFixture:
4433
async def _get_backend_licensed_items(
4534
rabbitmq_rpc_client: RabbitMQRPCClient,
@@ -59,12 +48,18 @@ async def _get_backend_licensed_items(
5948
total=len(examples),
6049
)
6150

62-
mock_rabbitmq_rpc_client.patch(
51+
class DummyRpcClient:
52+
pass
53+
54+
app.dependency_overrides[get_wb_api_rpc_client] = lambda: WbApiRpcClient(
55+
_client=DummyRpcClient()
56+
)
57+
mocker.patch(
6358
"simcore_service_api_server.services_rpc.wb_api_server._get_licensed_items",
6459
_get_backend_licensed_items,
6560
)
6661

67-
return mock_rabbitmq_rpc_client
62+
return mocker
6863

6964

7065
@pytest.mark.parametrize("exception_to_raise", [None])

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -441,7 +441,7 @@ async def deregister(self, job_id: None):
441441

442442
async def test_log_generator(mocker: MockFixture, faker: Faker):
443443
mocker.patch(
444-
"simcore_service_api_server.services.log_streaming.LogStreamer._project_done",
444+
"simcore_service_api_server.services_http.log_streaming.LogStreamer._project_done",
445445
return_value=True,
446446
)
447447
log_streamer = LogStreamer(user_id=3, director2_api=None, job_id=None, log_distributor=_MockLogDistributor(), log_check_timeout=1) # type: ignore

0 commit comments

Comments
 (0)