Skip to content

Commit 60db2ca

Browse files
tests: add rabbitmq client
1 parent 50ea6f9 commit 60db2ca

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

services/notifications/tests/conftest.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33

44

55
import datetime
6-
from collections.abc import AsyncIterator
6+
from collections.abc import AsyncIterator, Awaitable, Callable
77
from functools import partial
88
from pathlib import Path
99
from typing import Any
@@ -20,6 +20,7 @@
2020
from models_library.basic_types import BootModeEnum
2121
from pytest_simcore.helpers.monkeypatch_envs import EnvVarsDict, setenvs_from_dict
2222
from servicelib.fastapi.celery.app_server import FastAPIAppServer
23+
from servicelib.rabbitmq import RabbitMQRPCClient
2324
from simcore_service_notifications.core.application import create_app
2425
from simcore_service_notifications.core.settings import ApplicationSettings
2526
from simcore_service_notifications.modules.celery.tasks import (
@@ -112,3 +113,12 @@ def _on_worker_init_wrapper(sender: WorkController, **_kwargs):
112113
queues=",".join(queue.value for queue in TaskQueue),
113114
) as worker:
114115
yield worker
116+
117+
118+
@pytest.fixture
119+
async def notifications_rabbitmq_rpc_client(
120+
rabbitmq_rpc_client: Callable[[str], Awaitable[RabbitMQRPCClient]],
121+
) -> RabbitMQRPCClient:
122+
rpc_client = await rabbitmq_rpc_client("pytest_notifications_rpc_client")
123+
assert rpc_client
124+
return rpc_client

0 commit comments

Comments
 (0)