changed mock

Author: Andrei Neagu

packages/pytest-simcore/src/pytest_simcore/long_running_tasks.py

@@ -1,8 +1,9 @@
 import pytest
+from pytest_mock import MockerFixture
 from servicelib.long_running_tasks import task
 from servicelib.long_running_tasks._store.in_memory import InMemoryStore
 
 
 @pytest.fixture
-def uese_in_memory_lonng_running_tasks_storage() -> None:
-    task._StorageClass = InMemoryStore  # pylint:disable=protected-access
+def uese_in_memory_lonng_running_tasks_storage(mocker: MockerFixture) -> None:
+    mocker.patch.object(task, "RedisStore", InMemoryStore)

packages/service-library/src/servicelib/long_running_tasks/task.py

@@ -38,7 +38,6 @@
 
 _CANCEL_TASKS_CHECK_INTERVAL: Final[datetime.timedelta] = datetime.timedelta(seconds=5)
 
-_StorageClass: Final[type[BaseStore]] = RedisStore
 
 RegisteredTaskName: TypeAlias = str
 Namespace: TypeAlias = str
@@ -112,7 +111,7 @@ def __init__(
         namespace: Namespace,
     ):
         # Task groups: Every taskname maps to multiple asyncio.Task within TrackedTask model
-        self._tasks_data: BaseStore = _StorageClass(
+        self._tasks_data: BaseStore = RedisStore(
             redis_settings, namespace
         )  # type:ignore[call-arg]
         self._created_tasks: dict[TaskId, asyncio.Task] = {}