from v1.agent.router import _is_target_run_event, _is_terminal_run_event


def test_is_target_run_event_matches_expected_run_id() -> None:
    event: dict[str, object] = {"type": "STEP_STARTED", "runId": "run_123"}
    assert _is_target_run_event(event, target_run_id="run_123") is True


def test_is_target_run_event_rejects_other_run_id() -> None:
    event: dict[str, object] = {"type": "STEP_STARTED", "runId": "run_999"}
    assert _is_target_run_event(event, target_run_id="run_123") is False


def test_is_target_run_event_rejects_missing_run_id() -> None:
    event: dict[str, object] = {"type": "STEP_STARTED"}
    assert _is_target_run_event(event, target_run_id="run_123") is False


def test_is_terminal_run_event_only_accepts_terminal_types() -> None:
    assert _is_terminal_run_event({"type": "RUN_FINISHED"}) is True
    assert _is_terminal_run_event({"type": "RUN_ERROR"}) is True
    assert _is_terminal_run_event({"type": "STEP_STARTED"}) is False