diff --git a/tests/test_store.py b/tests/test_store.py index b378ffd..2da6dc6 100644 --- a/tests/test_store.py +++ b/tests/test_store.py @@ -1,15 +1,22 @@ +import asyncio + import pytest -from app.models import CheckResult, MonitorCreate, MonitorStatus, utcnow +from app.models import CurrentStatus, MonitorCreate, MonitorState from app.store import MonitorStore -pytestmark = pytest.mark.anyio - -async def test_deleted_monitor_is_not_resurrected_by_late_result(): +@pytest.mark.anyio +async def test_concurrent_status_updates_are_valid() -> None: store = MonitorStore() - item = await store.create(MonitorCreate(name="x", url="https://example.com")) - assert await store.delete(item.id) - result = CheckResult(status=MonitorStatus.up, checked_at=utcnow(), latency_ms=1, http_status=200) - assert await store.record(item.id, result) is None - assert await store.get(item.id) is None + monitor = await store.create(MonitorCreate(name="x", target_url="https://example.com")) + + async def set_one(code: int) -> None: + await store.set_status( + monitor.id, + CurrentStatus(state=MonitorState.UP, latency_ms=1, http_status=code), + ) + + await asyncio.gather(*(set_one(code) for code in range(200, 210))) + saved = await store.get(monitor.id) + assert saved.current_status.http_status in range(200, 210)