97 lines
3.5 KiB
Python
97 lines
3.5 KiB
Python
"""Прогресс звонящего — чистые данные, из которых собирается живой звонящий.
|
||
|
||
В снимок занятия прогресс пока не пишется; модель заложена, чтобы включение
|
||
сохранения было одним полем сохраняемой части и передачей в пересборку.
|
||
"""
|
||
|
||
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)
|