140 lines
5.4 KiB
Python
140 lines
5.4 KiB
Python
"""Преподавательский цикл: шаблон → черновик → проверка → утверждение."""
|
||
|
||
from pathlib import Path
|
||
|
||
import pytest
|
||
from fastapi.testclient import TestClient
|
||
|
||
from app.api.http import scenarios as scenarios_api
|
||
from app.db.base import get_session
|
||
from app.main import app
|
||
from app.scenarios import store
|
||
from app.scenarios.editor import merge_patch, template_copy, validate
|
||
from app.scenarios.loader import ScenarioError, load_file
|
||
|
||
LIBRARY = Path(__file__).resolve().parents[2] / "scenarios"
|
||
|
||
|
||
class FakeSession:
|
||
"""Минимальная транзакционная замена БД для проверки HTTP без Postgres."""
|
||
|
||
def __init__(self):
|
||
self.rows = {}
|
||
|
||
def add(self, row):
|
||
self.rows[row.id] = row
|
||
|
||
async def get(self, model, key):
|
||
return self.rows.get(key)
|
||
|
||
async def commit(self):
|
||
pass
|
||
|
||
async def scalars(self, query):
|
||
return [row for row in self.rows.values() if row.status == "published"]
|
||
|
||
|
||
@pytest.fixture
|
||
def client(monkeypatch):
|
||
db = FakeSession()
|
||
|
||
async def no_audit(*args, **kwargs):
|
||
pass
|
||
|
||
async def no_restore(*args, **kwargs):
|
||
return 0
|
||
|
||
app.dependency_overrides[get_session] = lambda: db
|
||
monkeypatch.setattr(scenarios_api, "audit", no_audit)
|
||
monkeypatch.setattr(store, "restore_published", no_restore)
|
||
with TestClient(app) as test_client:
|
||
yield test_client
|
||
app.dependency_overrides.clear()
|
||
|
||
|
||
def test_template_copy_is_local_independent_and_valid():
|
||
source = load_file(LIBRARY / "fire-apartment-l2.yaml", LIBRARY)
|
||
body = template_copy(source, "draft-example")
|
||
assert body["id"] == "draft-example"
|
||
assert body["extends"] is None
|
||
assert body["ticket"] is None
|
||
assert body["ground_truth"].keys() == {"address", "victims"}
|
||
assert validate(body).ground_truth.dds == source.ground_truth.dds
|
||
assert source.id != body["id"] # оригинал не меняется
|
||
|
||
|
||
def test_editor_rejects_derived_truth_and_missing_fact():
|
||
source = load_file(LIBRARY / "fire-apartment-l2.yaml", LIBRARY)
|
||
body = template_copy(source, "draft-example")
|
||
with pytest.raises(ScenarioError, match="ground_truth"):
|
||
merge_patch(body, {"ground_truth": {"dds": "02"}})
|
||
with pytest.raises(ScenarioError, match="id сценария"):
|
||
merge_patch(body, {"id": "other"})
|
||
broken = merge_patch(body, {"facts": []})
|
||
with pytest.raises(ScenarioError, match="нужны факты"):
|
||
validate(broken)
|
||
|
||
|
||
def test_draft_is_hidden_until_approval_and_then_available_to_lesson(client):
|
||
source = next(s for s in store.all_scenarios() if s.id == "fire-apartment-l2")
|
||
assert client.post("/api/auth/dev-token").status_code == 200
|
||
response = client.post(
|
||
"/api/scenarios/drafts/from-template",
|
||
json={"source_id": source.id, "title": "Учебная копия"},
|
||
)
|
||
assert response.status_code == 201, response.text
|
||
draft = response.json()
|
||
draft_id = draft["id"]
|
||
assert draft["generation"] == "template_copy"
|
||
assert draft["status"] == "draft"
|
||
assert store.get(draft_id) is None
|
||
assert client.get(f"/api/scenarios/{draft_id}").status_code == 404
|
||
|
||
changed = client.patch(
|
||
f"/api/scenarios/drafts/{draft_id}",
|
||
json={"first_line": "Соседи! В доме дым!"},
|
||
)
|
||
assert changed.status_code == 200, changed.text
|
||
assert changed.json()["body"]["first_line"] == "Соседи! В доме дым!"
|
||
check = client.post(f"/api/scenarios/drafts/{draft_id}/validate")
|
||
assert check.status_code == 200 and check.json()["valid"]
|
||
approved = client.post(f"/api/scenarios/drafts/{draft_id}/approve")
|
||
assert approved.status_code == 200, approved.text
|
||
assert approved.json()["status"] == "published"
|
||
assert store.get(draft_id).first_line == "Соседи! В доме дым!"
|
||
assert client.get(f"/api/scenarios/{draft_id}").status_code == 200
|
||
assert client.patch(f"/api/scenarios/drafts/{draft_id}", json={"title": "нет"}).status_code == 404
|
||
|
||
|
||
def test_invalid_draft_cannot_be_approved(client):
|
||
client.post("/api/auth/dev-token")
|
||
response = client.post(
|
||
"/api/scenarios/drafts/from-template",
|
||
json={"source_id": "fire-apartment-l2"},
|
||
)
|
||
draft_id = response.json()["id"]
|
||
assert client.patch(f"/api/scenarios/drafts/{draft_id}", json={"facts": []}).status_code == 200
|
||
check = client.post(f"/api/scenarios/drafts/{draft_id}/validate")
|
||
assert check.json()["valid"] is False
|
||
assert client.post(f"/api/scenarios/drafts/{draft_id}/approve").status_code == 422
|
||
assert store.get(draft_id) is None
|
||
|
||
|
||
def test_unauthenticated_user_cannot_manage_drafts(client):
|
||
response = client.post(
|
||
"/api/scenarios/drafts/from-template", json={"source_id": "fire-apartment-l2"}
|
||
)
|
||
assert response.status_code == 401
|
||
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_approved_scenario_restores_from_local_database():
|
||
source = load_file(LIBRARY / "fire-apartment-l2.yaml", LIBRARY)
|
||
db = FakeSession()
|
||
row = await store.create_draft(db, source=source)
|
||
approved = await store.approve_draft(db, row)
|
||
store.set_library([source]) # имитация нового процесса после чтения YAML
|
||
assert store.get(approved.id) is None
|
||
assert await store.restore_published(db) == 1
|
||
assert store.get(approved.id).title == approved.title
|
||
assert await store.restore_published(db) == 0
|