refactor: прогресс звонящего вынесен в модели SlotProgress и PersonaProgress

backend/tests/test_caller_progress.py

@@ -0,0 +1,97 @@
+"""Прогресс звонящего — чистые данные, из которых собирается живой звонящий.
+
+В снимок занятия прогресс пока не пишется; модель заложена, чтобы включение
+сохранения было одним полем сохраняемой части и передачей в пересборку.
+"""
+
+from uuid import uuid4
+
+from app.dialog.persona import PersonaProgress, PersonaState
+from app.dialog.slots import SlotMachine, SlotProgress
+from app.domain.events import Exercise, Mood, SessionMode
+from app.session import checkpoint
+from app.session.checkpoint import CallerProgress, dump_state, rebuild_live
+from app.session.state import SessionState
+from tests.test_refinement import SCENARIO
+from tests.test_slots import StemEmbedder
+
+
+def slot_progress() -> SlotProgress:
+    return SlotProgress(asked=["q_address", "q_city"], revealed=["f_address"],
+                        refined=["f_address"])
+
+
+def persona_progress() -> PersonaProgress:
+    return PersonaProgress(stage="dispatch", repeats=2, directive="panic_rises",
+                           history=[Mood.CALM, Mood.PANIC])
+
+
+def test_slot_machine_resumes_given_progress():
+    progress = slot_progress()
+
+    slots = SlotMachine(SCENARIO, StemEmbedder(), progress=progress)
+
+    assert slots.progress == progress
+    assert [fact.id for fact in slots.revealed_facts()] == ["f_address"]
+    assert slots.refined == ["f_address"]
+    assert slots.value_of("f_address") == "Королёв, улица Станционная, дом 28"
+    assert [item.id for item in slots.unasked()] == []
+
+
+def test_slot_machine_without_progress_starts_clean():
+    slots = SlotMachine(SCENARIO, StemEmbedder())
+
+    assert slots.progress == SlotProgress()
+    slots.hear("Назовите адрес")
+    assert slots.progress.revealed == ["f_address"]
+
+
+def test_persona_resumes_given_progress():
+    progress = persona_progress()
+
+    persona = PersonaState(SCENARIO.persona, progress=progress)
+
+    assert persona.progress == progress
+    assert persona.mood is Mood.PANIC
+    assert persona.repeats == 2
+
+
+def call_state() -> SessionState:
+    return SessionState(
+        session_id=uuid4(), scenario_id=SCENARIO.id, scenario_title=SCENARIO.title,
+        level=SCENARIO.level.value, mode=SessionMode.TRAINING,
+        exercise=Exercise.CALL, scenario=SCENARIO,
+    )
+
+
+def test_rebuild_live_uses_caller_progress(monkeypatch):
+    monkeypatch.setattr(checkpoint, "get_embedder", StemEmbedder)
+    state = call_state()
+    progress = CallerProgress(slots=slot_progress(), persona=persona_progress())
+
+    rebuild_live(state, progress)
+
+    assert state.slots.progress == progress.slots
+    assert state.persona.progress == progress.persona
+    assert state.caller is not None
+
+
+def test_rebuild_live_without_progress_starts_clean(monkeypatch):
+    monkeypatch.setattr(checkpoint, "get_embedder", StemEmbedder)
+    state = call_state()
+
+    rebuild_live(state)
+
+    assert state.slots.progress == SlotProgress()
+    assert state.persona.progress == PersonaProgress()
+
+
+def test_snapshot_has_no_caller_progress(monkeypatch):
+    monkeypatch.setattr(checkpoint, "get_embedder", StemEmbedder)
+    state = call_state()
+    rebuild_live(state, CallerProgress(slots=slot_progress(), persona=persona_progress()))
+
+    payload = dump_state(state)
+
+    assert not {"slots", "persona", "caller", "caller_progress"} & set(payload)
+    assert "panic_rises" not in str(payload)