23 lines
967 B
Python
23 lines
967 B
Python
|
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
|