127 lines
4.9 KiB
Python
127 lines
4.9 KiB
Python
"""Фикстуры тестов: изолированная in-memory БД + TestClient + помощники."""
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
|
|
# Тесты НИКОГДА не работают с БД дева/прода. Форсируем тестовое окружение и
|
|
# in-memory БД для ГЛОБАЛЬНОГО движка ещё ДО импорта приложения — даже случайное
|
|
# обращение к app.db.session.engine не затронет файлы дева/прода, какой бы
|
|
# APP_ENV ни был унаследован из окружения.
|
|
os.environ["APP_ENV"] = "development"
|
|
os.environ["DEV_DATABASE_URL"] = "sqlite://"
|
|
os.environ["FS_STARTUP_BOOTSTRAP"] = "0" # тесты сами поднимают схему/данные
|
|
|
|
import pytest # noqa: E402
|
|
from fastapi.testclient import TestClient # noqa: E402
|
|
from sqlalchemy.pool import StaticPool # noqa: E402
|
|
from sqlmodel import Session, SQLModel, create_engine # noqa: E402
|
|
|
|
import app.models # noqa: F401,E402 (регистрация моделей)
|
|
from app.core.security import hash_password # noqa: E402
|
|
from app.db.session import get_session # noqa: E402
|
|
from app.main import app # noqa: E402
|
|
from app.models import User # noqa: E402
|
|
from app.seed.reference_data import seed_reference_data # noqa: E402
|
|
|
|
|
|
@pytest.fixture()
|
|
def engine():
|
|
eng = create_engine(
|
|
"sqlite://",
|
|
connect_args={"check_same_thread": False},
|
|
poolclass=StaticPool,
|
|
)
|
|
SQLModel.metadata.create_all(eng)
|
|
with Session(eng) as s:
|
|
seed_reference_data(s)
|
|
yield eng
|
|
# In-memory БД исчезает с закрытием соединения StaticPool; drop_all не нужен
|
|
# (и невозможен из-за циклической FK users↔groups в SQLite).
|
|
eng.dispose()
|
|
|
|
|
|
@pytest.fixture()
|
|
def client(engine):
|
|
def _get_session():
|
|
with Session(engine) as s:
|
|
yield s
|
|
|
|
app.dependency_overrides[get_session] = _get_session
|
|
with TestClient(app) as c:
|
|
yield c
|
|
app.dependency_overrides.clear()
|
|
|
|
|
|
@pytest.fixture()
|
|
def make_admin(engine):
|
|
def _make(username: str = "admin", password: str = "secret123") -> int:
|
|
with Session(engine) as s:
|
|
admin = User(
|
|
nickname=username,
|
|
role="admin",
|
|
auth_provider="local",
|
|
password_hash=hash_password(password),
|
|
)
|
|
s.add(admin)
|
|
s.commit()
|
|
s.refresh(admin)
|
|
return admin.id
|
|
|
|
return _make
|
|
|
|
|
|
# ─── Помощники ────────────────────────────────────────────────────────────────
|
|
|
|
def csrf_headers(client: TestClient) -> dict:
|
|
token = client.cookies.get("csrf_token")
|
|
return {"X-CSRF-Token": token} if token else {}
|
|
|
|
|
|
def login(client: TestClient, nickname: str) -> dict:
|
|
# csrf_headers пуст при первом входе и содержит токен при смене пользователя
|
|
# (когда уже есть сессионная cookie и middleware требует заголовок).
|
|
r = client.post("/api/auth/dev/login", json={"nickname": nickname}, headers=csrf_headers(client))
|
|
assert r.status_code == 200, r.text
|
|
return r.json()
|
|
|
|
|
|
def start_match(client: TestClient, group_id: int, roster: list[dict]):
|
|
"""roster: [{user_id, faction_id, was_random?}] → ответ старта (in_progress)."""
|
|
return client.post(
|
|
"/api/matches",
|
|
json={"group_id": group_id, "participants": roster},
|
|
headers=csrf_headers(client),
|
|
)
|
|
|
|
|
|
def finish_match(
|
|
client: TestClient,
|
|
match_id: int,
|
|
results: list[dict],
|
|
win_reason: str = "objectives",
|
|
overall_comment: str | None = None,
|
|
):
|
|
"""results: [{user_id, place, comment?, faction_id?}]."""
|
|
body: dict = {"participants": results, "win_reason": win_reason}
|
|
if overall_comment is not None:
|
|
body["overall_comment"] = overall_comment
|
|
return client.post(
|
|
f"/api/matches/{match_id}/finish", json=body, headers=csrf_headers(client)
|
|
)
|
|
|
|
|
|
def create_finished_match(
|
|
client: TestClient, group_id: int, players: list[dict], win_reason: str = "objectives"
|
|
) -> dict:
|
|
"""players: [{user_id, faction_id, place, comment?}] → завершённая партия (для статистики)."""
|
|
roster = [{"user_id": p["user_id"], "faction_id": p["faction_id"]} for p in players]
|
|
started = start_match(client, group_id, roster)
|
|
assert started.status_code == 200, started.text
|
|
results = [
|
|
{"user_id": p["user_id"], "place": p["place"], "comment": p.get("comment")}
|
|
for p in players
|
|
]
|
|
r = finish_match(client, started.json()["id"], results, win_reason)
|
|
assert r.status_code == 200, r.text
|
|
return r.json()
|