fixed tests

Author: Andrei Neagu

packages/service-library/tests/long_running_tasks/conftest.py

@@ -1,7 +1,7 @@
 # pylint: disable=redefined-outer-name
 # pylint: disable=unused-argument
 import logging
-from collections.abc import AsyncIterator, Awaitable, Callable
+from collections.abc import AsyncIterable, AsyncIterator, Awaitable, Callable
 from datetime import timedelta
 
 import pytest
@@ -12,6 +12,7 @@
     RedisNamespace,
     TasksManager,
 )
+from servicelib.rabbitmq._client_rpc import RabbitMQRPCClient
 from settings_library.rabbit import RabbitSettings
 from settings_library.redis import RedisSettings
 from utils import TEST_CHECK_STALE_INTERVAL_S, NoWebAppLongRunningManager
@@ -79,3 +80,14 @@ async def _(
     for manager in managers:
         with log_catch(_logger, reraise=False):
             await manager.teardown()
+
+
+@pytest.fixture
+async def rabbitmq_rpc_client(
+    rabbit_service: RabbitSettings,
+) -> AsyncIterable[RabbitMQRPCClient]:
+    client = await RabbitMQRPCClient.create(
+        client_name="test-lrt-rpc-client", settings=rabbit_service
+    )
+    yield client
+    await client.close()

packages/service-library/tests/long_running_tasks/test_long_running_tasks_lrt_api.py

@@ -4,7 +4,7 @@
 
 import asyncio
 import secrets
-from collections.abc import AsyncIterable, Awaitable, Callable
+from collections.abc import Awaitable, Callable
 from typing import Any, Final
 
 import pytest
@@ -78,17 +78,6 @@ async def _to_replace(self: TasksManager) -> None:
     mocker.patch.object(TasksManager, "_stale_tasks_monitor", _to_replace)
 
 
-@pytest.fixture
-async def rabbitmq_rpc_client(
-    rabbit_service: RabbitSettings,
-) -> AsyncIterable[RabbitMQRPCClient]:
-    client = await RabbitMQRPCClient.create(
-        client_name="test-lrt-rpc-client", settings=rabbit_service
-    )
-    yield client
-    await client.close()
-
-
 @pytest.fixture
 async def long_running_managers(
     disable_stale_tasks_monitor: None,