74 lines
2.5 KiB
Python
74 lines
2.5 KiB
Python
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
|