cleanup before and after using Redis

Author: Andrei Neagu

services/dynamic-scheduler/tests/conftest.py

@@ -130,10 +130,15 @@ async def app(
 
 
 @pytest.fixture
-async def remove_redis_data(redis_service: RedisSettings) -> None:
-    async with RedisClientsManager(
-        {RedisManagerDBConfig(x) for x in RedisDatabase}, redis_service
-    ) as manager:
-        await logged_gather(
-            *[manager.client(d).redis.flushall() for d in RedisDatabase]
-        )
+async def remove_redis_data(redis_service: RedisSettings) -> AsyncIterator[None]:
+    async def _remove_all_data() -> None:
+        async with RedisClientsManager(
+            {RedisManagerDBConfig(x) for x in RedisDatabase}, redis_service
+        ) as manager:
+            await logged_gather(
+                *[manager.client(d).redis.flushall() for d in RedisDatabase]
+            )
+
+    await _remove_all_data()
+    yield None
+    await _remove_all_data()