52 lines
1.3 KiB
Python
52 lines
1.3 KiB
Python
|
import json
|
||
|
|
from unittest.mock import AsyncMock
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from app.core.task_tracker import TaskTracker
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.asyncio
|
||
|
|
async def test_update_task_status_refreshes_ttl(monkeypatch: pytest.MonkeyPatch) -> None:
|
||
|
|
client = AsyncMock()
|
||
|
|
client.hget.return_value = json.dumps({"task_id": "t1", "status": "pending"})
|
||
|
|
client.exists.return_value = 1
|
||
|
|
|
||
|
|
async def fake_get_client():
|
||
|
|
return client
|
||
|
|
|
||
|
|
monkeypatch.setattr(
|
||
|
|
"app.core.task_tracker.redis_service.get_client",
|
||
|
|
fake_get_client,
|
||
|
|
)
|
||
|
|
|
||
|
|
tracker = TaskTracker()
|
||
|
|
ok = await tracker.update_task_status("user-1", "t1", "running")
|
||
|
|
|
||
|
|
assert ok is True
|
||
|
|
client.hset.assert_awaited_once()
|
||
|
|
client.expire.assert_awaited_once_with("task:user:user-1:tasks", tracker.task_ttl)
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.asyncio
|
||
|
|
async def test_remove_task_refreshes_ttl_when_hash_remains(
|
||
|
|
monkeypatch: pytest.MonkeyPatch,
|
||
|
|
) -> None:
|
||
|
|
client = AsyncMock()
|
||
|
|
client.exists.return_value = 1
|
||
|
|
|
||
|
|
async def fake_get_client():
|
||
|
|
return client
|
||
|
|
|
||
|
|
monkeypatch.setattr(
|
||
|
|
"app.core.task_tracker.redis_service.get_client",
|
||
|
|
fake_get_client,
|
||
|
|
)
|
||
|
|
|
||
|
|
tracker = TaskTracker()
|
||
|
|
ok = await tracker.remove_task("user-1", "t1")
|
||
|
|
|
||
|
|
assert ok is True
|
||
|
|
client.hdel.assert_awaited_once()
|
||
|
|
client.expire.assert_awaited_once_with("task:user:user-1:tasks", tracker.task_ttl)
|