69 lines
2.4 KiB
Python
69 lines
2.4 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
|
|
|
|
|
|
@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_create_session(setup_db):
|
|
async with AsyncClient(
|
|
transport=ASGITransport(app=app), base_url="http://test"
|
|
) as ac:
|
|
start_time = datetime.now(timezone.utc).isoformat()
|
|
payload = {"start_time": start_time}
|
|
response = await ac.post("/api/sessions", json=payload)
|
|
assert response.status_code == 201
|
|
data = response.json()
|
|
assert datetime.fromisoformat(data["start_time"]).replace(
|
|
tzinfo=None
|
|
) == datetime.fromisoformat(start_time).replace(tzinfo=None)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_create_session_already_open(setup_db):
|
|
async with AsyncClient(
|
|
transport=ASGITransport(app=app), base_url="http://test"
|
|
) as ac:
|
|
start_time = datetime.now(timezone.utc).isoformat()
|
|
await ac.post("/api/sessions", json={"start_time": start_time})
|
|
response = await ac.post("/api/sessions", json={"start_time": start_time})
|
|
assert response.status_code == 409
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_end_session(setup_db):
|
|
async with AsyncClient(
|
|
transport=ASGITransport(app=app), base_url="http://test"
|
|
) as ac:
|
|
start_time = datetime.now(timezone.utc).isoformat()
|
|
await ac.post("/api/sessions", json={"start_time": start_time})
|
|
|
|
end_time = datetime.now(timezone.utc).isoformat()
|
|
response = await ac.patch("/api/sessions/1/end", json={"end_time": end_time})
|
|
|
|
assert response.status_code == 200
|
|
assert datetime.fromisoformat(response.json()["end_time"]).replace(
|
|
tzinfo=None
|
|
) == datetime.fromisoformat(end_time).replace(tzinfo=None)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_list_sessions(setup_db):
|
|
async with AsyncClient(
|
|
transport=ASGITransport(app=app), base_url="http://test"
|
|
) as ac:
|
|
start_time = datetime.now(timezone.utc).isoformat()
|
|
await ac.post("/api/sessions", json={"start_time": start_time})
|
|
response = await ac.get("/api/sessions")
|
|
assert response.status_code == 200
|
|
assert len(response.json()["sessions"]) >= 1
|