Initial commit

backend/tests/test_stats.py

@@ -0,0 +1,73 @@
+import pytest
+from httpx import AsyncClient, ASGITransport
+from app.main import app
+from app.database import engine, init_db
+from app.models import SQLModel
+from datetime import datetime, timezone, timedelta
+from sqlmodel.ext.asyncio.session import AsyncSession
+
+
+@pytest.fixture
+async def setup_db():
+    async with engine.begin() as conn:
+        await conn.run_sync(SQLModel.metadata.drop_all)
+        await conn.run_sync(SQLModel.metadata.create_all)
+
+
+@pytest.mark.asyncio
+async def test_daily_summary(setup_db):
+    from datetime import date
+
+    async with AsyncClient(
+        transport=ASGITransport(app=app), base_url="http://test"
+    ) as ac:
+        # Create a session spanning today
+        start = datetime.now(timezone.utc) - timedelta(hours=5)
+        end = datetime.now(timezone.utc)
+        # We need to manually insert since the API only allows POST for start_time
+        from app.database import get_session
+        from app.models import SleepSession
+        from sqlalchemy import select
+
+        async with AsyncSession(engine) as session:
+            new_s = SleepSession(start_time=start, end_time=end)
+            session.add(new_s)
+            await session.commit()
+
+        from_date = date.today().isoformat()
+        # Use the next day as to_date to cover the whole current day
+        to_date = (date.today() + timedelta(days=1)).isoformat()
+        response = await ac.get(
+            f"/api/stats/daily-summary?from_date={from_date}&to_date={to_date}"
+        )
+
+    assert response.status_code == 200
+    data = response.json()
+    assert len(data["days"]) >= 1
+
+
+@pytest.mark.asyncio
+async def test_timetable(setup_db):
+    from datetime import date
+
+    async with AsyncClient(
+        transport=ASGITransport(app=app), base_url="http://test"
+    ) as ac:
+        from app.database import get_session
+        from app.models import SleepSession
+        from sqlalchemy import select
+
+        async with AsyncSession(engine) as session:
+            start = datetime.now(timezone.utc) - timedelta(hours=1)
+            new_s = SleepSession(start_time=start, end_time=datetime.now(timezone.utc))
+            session.add(new_s)
+            await session.commit()
+
+        from_date = date.today().isoformat()
+        to_date = (date.today() + timedelta(days=1)).isoformat()
+        response = await ac.get(
+            f"/api/stats/timetable?from_date={from_date}&to_date={to_date}"
+        )
+
+    assert response.status_code == 200
+    assert len(response.json()["slots"]) > 0