Added expire to add_memes_to_queue_by_key

src/redis.py

@@ -60,10 +60,15 @@ async def get_meme_queue_length_by_key(key: str) -> int:
     return await redis_client.scard(key)
 
 
-async def add_memes_to_queue_by_key(key: str, memes: list[dict]) -> int:
+async def add_memes_to_queue_by_key(
+        key: str, memes: list[dict],
+        expire: int = 3600
+    ) -> None:
     jsoned_memes = [orjson.dumps(meme) for meme in memes]
-    # TODO: add ttl, probably using redis transactions
-    return await redis_client.sadd(key, *jsoned_memes)
+    p = await redis_client.pipeline(transaction=True)
+    await p.sadd(key, *jsoned_memes)
+    await p.expire(key, expire)
+    await p.execute(raise_on_error=True)
 
 
 def get_user_info_key(user_id: int) -> str:

tests/conftest.py

@@ -5,8 +5,6 @@
 import pytest_asyncio
 from async_asgi_testclient import TestClient
 
-from src.main import app
-
 
 @pytest.fixture(autouse=True, scope="session")
 def run_migrations() -> None:
@@ -23,12 +21,3 @@ def event_loop() -> Generator[asyncio.AbstractEventLoop, None, None]:
     loop = asyncio.get_event_loop_policy().new_event_loop()
     yield loop
     loop.close()
-
-
-@pytest_asyncio.fixture
-async def client() -> AsyncGenerator[TestClient, None]:
-    host, port = "127.0.0.1", "9000"
-    scope = {"client": (host, port)}
-
-    async with TestClient(app, scope=scope) as client:
-        yield client

tests/test_redis.py

@@ -0,0 +1,29 @@
+from datetime import datetime
+
+import asyncio
+
+import pytest
+import pytest_asyncio
+
+from src import redis
+
+
+@pytest.mark.asyncio
+async def test_add_memes_to_queue_by_key_ok():
+    user_id = 999
+    queue_key = redis.get_meme_queue_key(user_id)
+
+    memes = [
+        {'id': 1, 'recommended_by': 'best_algo'},
+        {'id': 2, 'recommended_by': 'best_algo'},
+    ]
+
+    await redis.add_memes_to_queue_by_key(queue_key, memes, expire=1)
+
+    stored = await redis.redis_client.smembers(queue_key)
+    assert len(stored) == 2
+
+    await asyncio.sleep(3)
+
+    stored = await redis.redis_client.smembers(queue_key)
+    assert len(stored) == 0