Phase 11

tests/test_voice_prompt_controller.py

@@ -0,0 +1,73 @@
+from unittest.mock import Mock
+
+from src.config_loader import VoicePromptSettings
+from src.models import BoundingBox, FrameInfo, TrackEvent, TrackedPerson, TurnFrameResult
+from src.voice_prompt_controller import VoicePromptController
+
+
+def cfg(mode: str = "prompt", **kw: object) -> VoicePromptSettings:
+    values = dict(enabled=True, mode=mode, trigger_strategy="trigger_line_crossing", require_confirmed_track=True,
+                  require_inside_region=True, skip_if_already_turned=True, one_prompt_per_track=True,
+                  global_cooldown_sec=3., minimum_track_age_sec=0., response_window_sec=3.,
+                  fixed_position_axis="x", fixed_position_value=640., fixed_position_direction="decreasing")
+    values.update(kw)
+    return VoicePromptSettings(**values)
+
+
+def info(frame: int, elapsed: float) -> FrameInfo:
+    return FrameInfo(frame, f"t{frame}", elapsed, 100, 100, 30.)
+
+
+def person(track_id: int = 1) -> TrackedPerson:
+    return TrackedPerson(track_id, BoundingBox(1, 1, 20, 40), .9, 0, "person", 10, 20, 10, 40, True, True, 10, 0)
+
+
+def crossing(track_id: int = 1) -> TrackEvent:
+    return TrackEvent("trigger_crossed", track_id, 1, "t", 1., 10, 40)
+
+
+def turn(frame: int, elapsed: float, confirmed: bool = True) -> TurnFrameResult:
+    return TurnFrameResult(1, frame, f"t{frame}", elapsed, True, True, True, 0., 25., 25., 25., "positive",
+                           True, confirmed, "weak", .4, "")
+
+
+def test_prompt_success_duplicate_and_response() -> None:
+    audio = Mock(audio_file="a.wav")
+    audio.play.return_value = True
+    controller = VoicePromptController(cfg(), audio)
+    decisions, events = controller.update([person()], [crossing()], [], info(1, 1.))
+    assert decisions[0].prompt_played
+    assert {e.event_type for e in events} >= {"prompt_played", "response_window_started"}
+    controller.update([person()], [crossing()], [], info(2, 1.1))
+    assert audio.play.call_count == 1
+    _, events = controller.update([person()], [], [turn(3, 1.8)], info(3, 1.8))
+    response = next(e for e in events if e.event_type == "response_detected")
+    assert response.reaction_time_sec == .8
+
+
+def test_prompt_failure_control_and_expiration() -> None:
+    failed = Mock(audio_file="missing.wav")
+    failed.play.return_value = False
+    controller = VoicePromptController(cfg(), failed)
+    _, events = controller.update([person()], [crossing()], [], info(1, 1.))
+    assert "prompt_failed" in {event.event_type for event in events}
+    control_audio = Mock(audio_file="a.wav")
+    control = VoicePromptController(cfg("control", response_window_sec=1.), control_audio)
+    decisions, events = control.update([person()], [crossing()], [], info(1, 1.))
+    assert decisions[0].pseudo_prompt and not decisions[0].prompt_played
+    control_audio.play.assert_not_called()
+    _, events = control.update([person()], [], [], info(2, 2.1))
+    assert "response_window_expired" in {event.event_type for event in events}
+
+
+def test_cooldown_and_finalize_all() -> None:
+    audio = Mock(audio_file="a.wav")
+    audio.play.return_value = True
+    controller = VoicePromptController(cfg(), audio)
+    controller.update([person(1)], [crossing(1)], [], info(1, 1.))
+    decisions, _ = controller.update([person(2)], [crossing(2)], [], info(2, 2.))
+    assert decisions[0].reason == "global_cooldown"
+    states, events = controller.finalize_all(info(3, 5.))
+    assert len(states) == 2 and all(state.finalized for state in states)
+    assert all(event.event_type == "voice_track_finalized" for event in events)
+