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)
|