test_tui_state.py 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925
  1. """Tests for the TUI view-state reducer."""
  2. import pytest
  3. from stt_daemon.tui.state import (
  4. MAX_TRANSCRIPTS,
  5. SPEAKER_NAMED_STYLE,
  6. SPEAKER_UNKNOWN_STYLE,
  7. TIME_STYLE,
  8. ViewState,
  9. _fmt_clock,
  10. )
  11. def test_mode_event_updates_mode_and_style():
  12. state = ViewState()
  13. state.ingest({"type": "mode", "mode": "command_armed"})
  14. assert state.mode == "command_armed"
  15. assert state.mode_style() == "yellow"
  16. def test_initial_health_is_not_optimistic():
  17. """Before the first heartbeat arrives, health must read as unknown/down."""
  18. state = ViewState()
  19. assert state.health_ok() is False
  20. state.ingest({"type": "heartbeat", "stt_ok": True, "device_ok": True})
  21. assert state.health_ok() is True
  22. def test_heartbeat_self_corrects_missed_mode():
  23. state = ViewState()
  24. state.ingest(
  25. {
  26. "type": "heartbeat",
  27. "mode": "chat",
  28. "stt_ok": True,
  29. "in_flight": False,
  30. "worker_stall_s": 0.0,
  31. "queue_depth": 2,
  32. "device_ok": True,
  33. }
  34. )
  35. assert state.mode == "chat"
  36. assert state.queue_depth == 2
  37. def test_heartbeat_ingests_inference_depth_and_in_flight():
  38. state = ViewState()
  39. state.ingest(
  40. {
  41. "type": "heartbeat",
  42. "inference_depth": 3,
  43. "inference_in_flight": True,
  44. }
  45. )
  46. assert state.inference_depth == 3
  47. assert state.inference_in_flight is True
  48. def test_heartbeat_omitting_inference_fields_preserves_prior_values():
  49. state = ViewState()
  50. state.inference_depth = 5
  51. state.inference_in_flight = True
  52. state.ingest(
  53. {
  54. "type": "heartbeat",
  55. "mode": "idle",
  56. "stt_ok": True,
  57. }
  58. )
  59. assert state.inference_depth == 5
  60. assert state.inference_in_flight is True
  61. def test_health_red_when_stt_not_ok():
  62. state = ViewState()
  63. state.ingest({"type": "heartbeat", "mode": "idle", "stt_ok": False})
  64. assert state.health_ok() is False
  65. def test_health_red_when_device_down():
  66. state = ViewState()
  67. state.ingest({"type": "heartbeat", "mode": "idle", "stt_ok": True, "device_ok": False})
  68. assert state.health_ok() is False
  69. def test_worker_stall_recorded_and_degrades():
  70. state = ViewState()
  71. state.ingest({"type": "heartbeat", "stt_ok": False, "worker_stall_s": 16.5})
  72. assert state.worker_stall_s == 16.5
  73. assert state.health_ok() is False
  74. def test_command_text_and_edited_flag():
  75. state = ViewState()
  76. state.ingest({"type": "command_text", "text": "hello", "edited": True, "phase": "ready"})
  77. assert state.command_text == "hello"
  78. assert state.command_edited is True
  79. assert state.phase == "ready"
  80. def test_status_populates_fields():
  81. state = ViewState()
  82. state.ingest(
  83. {
  84. "type": "status",
  85. "mode": "command_armed",
  86. "ready": True,
  87. "phase": "ready",
  88. "command_text": "x",
  89. "stt_ok": True,
  90. }
  91. )
  92. assert state.mode == "command_armed"
  93. assert state.ready is True
  94. assert state.command_text == "x"
  95. def test_transcripts_are_bounded():
  96. state = ViewState()
  97. for i in range(MAX_TRANSCRIPTS + 25):
  98. state.ingest({"type": "transcript", "text": f"t{i}", "lang": "en"})
  99. assert len(state.transcripts) == MAX_TRANSCRIPTS
  100. assert state.transcripts[-1]["text"] == f"t{MAX_TRANSCRIPTS + 24}"
  101. last_block = state.transcript_lines()[-1].plain
  102. assert f"t{MAX_TRANSCRIPTS + 24}" in last_block
  103. assert "EN" in last_block
  104. def test_transcript_block_header_has_clock_lang_duration_and_reason():
  105. state = ViewState()
  106. state.ingest(
  107. {
  108. "type": "transcript",
  109. "ts": 1759073916.19,
  110. "lang": "es",
  111. "lang_prob": 0.99,
  112. "duration_s": 7.83,
  113. "end_reason": "end_of_turn",
  114. "text": "hola mundo",
  115. }
  116. )
  117. header, body = state.transcript_lines()[0].plain.split("\n")
  118. assert header[2] == ":" and header[5] == ":" and len(header[:8]) == 8
  119. assert "ES" in header
  120. assert "7.8s" in header
  121. assert "end_of_turn" in header
  122. assert body.strip() == "hola mundo"
  123. def _header_span_style(block, fragment: str) -> str:
  124. for span in block.spans:
  125. if block.plain[span.start : span.end] == fragment:
  126. return str(span.style)
  127. raise AssertionError(f"no span rendering {fragment!r} in {block.plain!r}")
  128. def test_transcript_header_places_named_speaker_between_lang_and_duration():
  129. state = ViewState()
  130. state.ingest(
  131. {
  132. "type": "transcript",
  133. "ts": 1759073916.19,
  134. "lang": "en",
  135. "duration_s": 3.24,
  136. "end_reason": "pause",
  137. "speaker": "Efren",
  138. "speaker_id": "spk_0001",
  139. "speaker_status": "matched",
  140. "speaker_score": 0.87,
  141. "text": "turn on the lights",
  142. }
  143. )
  144. block = state.transcript_lines()[0]
  145. header, body = block.plain.split("\n")
  146. assert header.index("EN") < header.index("Efren") < header.index("3.2s")
  147. assert _header_span_style(block, "Efren") == SPEAKER_NAMED_STYLE == "bold green"
  148. assert body.strip() == "turn on the lights"
  149. def test_transcript_header_renders_matched_but_unnamed_speaker_as_unknown():
  150. state = ViewState()
  151. state.ingest(
  152. {
  153. "type": "transcript",
  154. "ts": 1759073916.19,
  155. "lang": "en",
  156. "duration_s": 3.24,
  157. "speaker": "Unknown",
  158. "speaker_id": "spk_0002",
  159. "speaker_status": "matched",
  160. "text": "hello there",
  161. }
  162. )
  163. block = state.transcript_lines()[0]
  164. header = block.plain.split("\n")[0]
  165. assert header.index("EN") < header.index("Unknown") < header.index("3.2s")
  166. assert _header_span_style(block, "Unknown") == SPEAKER_UNKNOWN_STYLE == "bold red"
  167. @pytest.mark.parametrize("status", ["uncertain", "unknown", "none"])
  168. def test_transcript_header_renders_unmatched_statuses_as_unknown(status):
  169. state = ViewState()
  170. state.ingest(
  171. {
  172. "type": "transcript",
  173. "ts": 1759073916.19,
  174. "lang": "es",
  175. "duration_s": 1.0,
  176. "speaker": "Unknown",
  177. "speaker_status": status,
  178. "text": "hola",
  179. }
  180. )
  181. block = state.transcript_lines()[0]
  182. assert "Unknown" in block.plain
  183. assert _header_span_style(block, "Unknown") == SPEAKER_UNKNOWN_STYLE
  184. def test_transcript_header_without_speaker_field_renders_unknown():
  185. state = ViewState()
  186. state.ingest(
  187. {
  188. "type": "transcript",
  189. "ts": 1759073916.19,
  190. "lang": "en",
  191. "duration_s": 1.0,
  192. "text": "no speaker field",
  193. }
  194. )
  195. block = state.transcript_lines()[0]
  196. assert _header_span_style(block, "Unknown") == SPEAKER_UNKNOWN_STYLE
  197. def test_speakers_event_populates_the_speaker_snapshot():
  198. speakers = [
  199. {
  200. "id": "spk_0001",
  201. "name": "Efren",
  202. "status": "confirmed",
  203. "cumulative_s": 42.5,
  204. "sample_path": "speakers/samples/spk_0001.wav",
  205. "created_at": "2026-09-28T12:00:00",
  206. }
  207. ]
  208. state = ViewState()
  209. state.ingest({"type": "speakers", "speakers": speakers})
  210. assert state.speakers == speakers
  211. def test_speakers_event_without_payload_clears_the_snapshot():
  212. state = ViewState()
  213. state.speakers = [{"id": "spk_0001"}]
  214. state.ingest({"type": "speakers"})
  215. assert state.speakers == []
  216. def test_transcript_time_language_and_text_colors_are_disjoint():
  217. state = ViewState()
  218. for lang in ("es", "en", "fr", "de", "pt", "it", "zz"):
  219. state.ingest(
  220. {
  221. "type": "transcript",
  222. "ts": 1.0,
  223. "lang": lang,
  224. "duration_s": 1.0,
  225. "end_reason": "pause",
  226. "text": "x",
  227. }
  228. )
  229. palette = {
  230. str(span.style).split()[-1]
  231. for block in state.transcript_lines()
  232. for span in block.spans
  233. }
  234. time_color = str(state.transcript_lines()[0].spans[0].style).split()[-1]
  235. lang_color = str(state.transcript_lines()[0].spans[1].style).split()[-1]
  236. text_colors = {
  237. str(block.spans[-1].style).split()[-1] for block in state.transcript_lines()
  238. }
  239. assert not ({time_color, lang_color} & text_colors), palette
  240. assert time_color != lang_color
  241. def test_transcripts_total_keeps_counting_past_the_cap():
  242. """The bounded buffer drops old lines; the total must still advance."""
  243. state = ViewState()
  244. for i in range(MAX_TRANSCRIPTS + 25):
  245. state.ingest({"type": "transcript", "text": f"t{i}", "lang": "en"})
  246. assert state.transcripts_total == MAX_TRANSCRIPTS + 25
  247. def test_command_output_is_collected_for_display():
  248. state = ViewState()
  249. state.ingest({"type": "tool_call", "tool": "get_datetime", "args": {}})
  250. state.ingest(
  251. {"type": "tool_result", "tool": "get_datetime", "ok": True, "result": "2026-09-21T01:00:00"}
  252. )
  253. state.ingest({"type": "error", "code": "not_ready", "message": "command is not ready"})
  254. lines = state.lower_lines()
  255. assert any("tool get_datetime" in line for line in lines)
  256. assert any("ok:" in line and "2026-09-21" in line for line in lines)
  257. assert any("not_ready" in line for line in lines)
  258. def test_chat_turns_also_appear_in_lower_log():
  259. state = ViewState()
  260. state.ingest({"type": "chat_message", "role": "user", "text": "hello"})
  261. assert state.lower_lines() == ["user: hello"]
  262. def test_lower_log_is_bounded():
  263. from stt_daemon.tui.state import MAX_LOWER
  264. state = ViewState()
  265. for i in range(MAX_LOWER + 20):
  266. state.ingest({"type": "error", "code": f"e{i}", "message": "m"})
  267. assert len(state.lower_lines()) == MAX_LOWER
  268. def test_speech_activity_is_tracked():
  269. state = ViewState()
  270. assert state.speech_open is False
  271. state.ingest({"type": "speech_start"})
  272. assert state.speech_open is True
  273. state.ingest({"type": "speech_stop"})
  274. assert state.speech_open is False
  275. state.ingest({"type": "status", "speech_open": True})
  276. assert state.speech_open is True
  277. def test_chat_and_tool_events_recorded():
  278. state = ViewState()
  279. state.ingest({"type": "chat_message", "role": "user", "text": "hi"})
  280. state.ingest({"type": "chat_message", "role": "assistant", "text": "hello"})
  281. state.ingest({"type": "tool_call", "tool": "get_datetime", "args": {}})
  282. state.ingest({"type": "tool_result", "tool": "get_datetime", "ok": True, "result": "x"})
  283. state.ingest({"type": "error", "code": "no_viewer", "message": "m"})
  284. assert state.chat_lines() == ["user: hi", "assistant: hello"]
  285. assert state.tool is not None and state.tool["tool"] == "get_datetime"
  286. assert state.tool_result is not None and state.tool_result["ok"] is True
  287. assert state.last_error is not None and state.last_error["code"] == "no_viewer"
  288. def test_leaving_chat_clears_chat_pane_but_keeps_command_output():
  289. state = ViewState()
  290. state.ingest({"type": "tool_call", "tool": "get_datetime", "args": {}})
  291. state.ingest({"type": "mode", "mode": "chat"})
  292. state.ingest({"type": "chat_message", "role": "user", "text": "hello"})
  293. state.ingest({"type": "chat_message", "role": "assistant", "text": "hi"})
  294. assert state.chat_lines() == ["user: hello", "assistant: hi"]
  295. epoch = state.chat_epoch
  296. state.ingest({"type": "mode", "mode": "idle"})
  297. assert state.chat_lines() == []
  298. assert state.lower_lines() == ["-> tool get_datetime"]
  299. assert state.chat_epoch == epoch + 1
  300. def test_heartbeat_mode_change_clears_chat():
  301. state = ViewState()
  302. state.ingest({"type": "mode", "mode": "chat"})
  303. state.ingest({"type": "chat_message", "role": "user", "text": "x"})
  304. state.ingest({"type": "heartbeat", "mode": "idle", "stt_ok": True})
  305. assert state.chat_lines() == []
  306. assert state.lower_lines() == []
  307. def test_compaction_note_renders_and_clears_on_chat_exit():
  308. state = ViewState()
  309. state.ingest({"type": "mode", "mode": "chat"})
  310. state.ingest(
  311. {
  312. "type": "chat_message",
  313. "role": "note",
  314. "text": "Compacted 4 earlier turns into a summary.",
  315. }
  316. )
  317. assert "note: Compacted 4 earlier turns into a summary." in state.lower_lines()
  318. state.ingest({"type": "mode", "mode": "idle"})
  319. assert state.lower_lines() == []
  320. def test_entering_chat_does_not_clear_rendered_log():
  321. state = ViewState()
  322. state.ingest({"type": "tool_result", "tool": "get_datetime", "ok": True, "result": "x"})
  323. epoch = state.chat_epoch
  324. state.ingest({"type": "mode", "mode": "chat"})
  325. assert state.chat_epoch == epoch
  326. assert any("ok:" in line for line in state.lower_lines())
  327. def test_cleared_event_wipes_output_and_increments_epoch():
  328. state = ViewState()
  329. state.ingest({"type": "chat_message", "role": "user", "text": "hello"})
  330. state.ingest({"type": "chat_message", "role": "assistant", "text": "hi"})
  331. state.ingest({"type": "tool_call", "tool": "get_datetime", "args": {}})
  332. state.ingest(
  333. {
  334. "type": "tool_result",
  335. "tool": "get_datetime",
  336. "ok": True,
  337. "result": "2026-09-21T01:00:00",
  338. }
  339. )
  340. state.ingest({"type": "error", "code": "test_error", "message": "test message"})
  341. state.ingest({"type": "transcript", "text": "preserve this", "lang": "en"})
  342. epoch = state.chat_epoch
  343. assert len(state.chat_lines()) == 2
  344. assert len(state.lower_lines()) >= 4
  345. assert state.tool is not None
  346. assert state.tool_result is not None
  347. assert state.last_error is not None
  348. assert len(state.transcripts) == 1
  349. state.ingest({"type": "cleared"})
  350. assert state.chat_lines() == []
  351. assert state.lower_lines() == []
  352. assert state.tool is None
  353. assert state.tool_result is None
  354. assert state.last_error is None
  355. assert state.chat_epoch == epoch + 1
  356. assert len(state.transcripts) == 1
  357. block = state.transcript_lines()[0].plain
  358. assert "preserve this" in block
  359. assert "EN" in block
  360. def _signals_triage():
  361. return {
  362. "order": ["q_choice", "q_noul", "q_score", "q_unknown"],
  363. "items": {
  364. "q_choice": {
  365. "kind": "choice",
  366. "choice": "yes",
  367. "answer_confidence": 0.91,
  368. "flagged": True,
  369. },
  370. "q_noul": {"kind": "noul", "p": 0.75, "flagged": False},
  371. "q_score": {"kind": "score", "level": 3, "label": "high", "flagged": True},
  372. "q_unknown": {"kind": "mystery", "flagged": False},
  373. },
  374. "escalated": True,
  375. "input_tokens": 42,
  376. }
  377. def test_triage_signals_event_stores_and_bumps_epoch():
  378. state = ViewState()
  379. epoch = state.triage_epoch
  380. state.ingest({"type": "triage", "stage": "signals", "triage": _signals_triage()})
  381. assert state.triage == _signals_triage()
  382. assert state.triage_epoch == epoch + 1
  383. def test_triage_content_event_replaces_triage_and_bumps_epoch():
  384. state = ViewState()
  385. state.ingest({"type": "triage", "stage": "signals", "triage": _signals_triage()})
  386. epoch = state.triage_epoch
  387. merged = {
  388. **_signals_triage(),
  389. "content": {"summary": "s", "opportunities": [], "risks": [], "action_items": []},
  390. }
  391. state.ingest({"type": "triage", "stage": "content", "triage": merged})
  392. assert state.triage == merged
  393. assert state.triage_epoch == epoch + 1
  394. def test_triage_cleared_event_empties_and_bumps_epoch():
  395. state = ViewState()
  396. state.ingest({"type": "triage", "stage": "signals", "triage": _signals_triage()})
  397. epoch = state.triage_epoch
  398. state.ingest({"type": "triage", "stage": "cleared"})
  399. assert state.triage is None
  400. assert state.triage_epoch == epoch + 1
  401. def test_status_adopts_triage_fields():
  402. state = ViewState()
  403. epoch = state.triage_epoch
  404. state.ingest(
  405. {
  406. "type": "status",
  407. "triage": _signals_triage(),
  408. "triage_state": "ok",
  409. "triage_reason": "",
  410. }
  411. )
  412. assert state.triage_state == "ok"
  413. assert state.triage == _signals_triage()
  414. assert state.triage_epoch == epoch + 1
  415. def test_status_without_triage_fields_bumps_nothing():
  416. state = ViewState()
  417. state.ingest({"type": "status", "mode": "idle", "ready": True})
  418. assert state.triage_epoch == 0
  419. assert state.triage_state == "disabled"
  420. def test_triage_lines_choice_noul_score_and_flag_marks():
  421. state = ViewState()
  422. state.triage = _signals_triage()
  423. state.triage_state = "ok"
  424. lines = state.triage_lines()
  425. assert lines[0] == "q_choice: yes (0.91) ⚠"
  426. assert lines[1].startswith("███████ q_noul: 0.75")
  427. assert lines[2].startswith("███") and "q_score: high ⚠" in lines[2]
  428. assert lines[3] == "q_unknown"
  429. def test_triage_lines_content_block_appended():
  430. state = ViewState()
  431. state.triage = {
  432. "order": ["q1"],
  433. "items": {"q1": {"kind": "choice", "choice": "no", "answer_confidence": 0.9}},
  434. "content": {
  435. "summary": "user wants action",
  436. "opportunities": ["opp1", "opp2", "opp3", "opp4"],
  437. "risks": ["risk1"],
  438. "action_items": [{"what": "call", "why": "urgent"}],
  439. },
  440. }
  441. state.triage_state = "ok"
  442. lines = state.triage_lines()
  443. assert "" in lines
  444. assert "summary: user wants action" in lines
  445. assert "opportunitie: opp1" in lines
  446. assert "opp4" not in " ".join(lines)
  447. assert "risk: risk1" in lines
  448. assert "action_item: {'what': 'call', 'why': 'urgent'}" in lines
  449. def test_triage_lines_non_ok_state_renders_reason():
  450. state = ViewState()
  451. state.triage_state = "loading"
  452. state.triage_reason = "model warming up"
  453. assert state.triage_lines() == ["triage: unavailable — model warming up"]
  454. def test_triage_lines_unknown_reason_falls_back_generically():
  455. state = ViewState()
  456. state.triage_state = "service_down"
  457. assert state.triage_lines() == ["triage: unavailable — service_down"]
  458. def test_triage_lines_ok_without_data_renders_no_data():
  459. state = ViewState()
  460. state.triage_state = "ok"
  461. assert state.triage_lines() == ["triage: unavailable — no data"]
  462. def test_triage_lines_malformed_payload_never_raises():
  463. state = ViewState()
  464. state.triage_state = "ok"
  465. state.triage = {
  466. "order": ["a", "b", "c"],
  467. "items": {
  468. "a": {"kind": "choice", "flagged": True},
  469. "b": {"kind": "noul", "p": "not-a-number", "flagged": True},
  470. "c": {"kind": "score", "level": None},
  471. },
  472. }
  473. lines = state.triage_lines()
  474. assert lines[0] == "a: ? (?) ⚠"
  475. assert "b: ? ⚠" in lines[1]
  476. assert "c: ?" in lines[2]
  477. def test_triage_lines_empty_order_renders_no_signals():
  478. state = ViewState()
  479. state.triage_state = "ok"
  480. state.triage = {"order": [], "items": {}}
  481. assert state.triage_lines() == ["triage: no signals"]
  482. def test_clear_output_resets_triage_and_bumps_epoch():
  483. state = ViewState()
  484. state.ingest({"type": "triage", "stage": "signals", "triage": _signals_triage()})
  485. epoch = state.triage_epoch
  486. state.clear_output()
  487. assert state.triage is None
  488. assert state.triage_epoch == epoch + 1
  489. def test_status_adopted_triage_survives_then_cleared_event_empties_panel():
  490. state = ViewState()
  491. state.ingest(
  492. {"type": "status", "triage": _signals_triage(), "triage_state": "ok", "triage_reason": ""}
  493. )
  494. assert state.triage_lines()[0].startswith("q_choice:")
  495. state.ingest({"type": "triage", "stage": "cleared"})
  496. assert state.triage is None
  497. def test_lone_signals_event_renders_without_status_snapshot():
  498. state = ViewState()
  499. state.ingest(
  500. {
  501. "type": "triage",
  502. "stage": "signals",
  503. "triage": {
  504. "order": ["category"],
  505. "items": {
  506. "category": {"kind": "choice", "choice": "question", "answer_confidence": 0.91}
  507. },
  508. },
  509. }
  510. )
  511. assert state.triage_state == "ok"
  512. assert state.triage_lines() == ["category: question (0.91)"]
  513. def test_lone_content_event_renders_without_status_snapshot():
  514. state = ViewState()
  515. state.ingest(
  516. {
  517. "type": "triage",
  518. "stage": "content",
  519. "triage": {
  520. "order": ["category"],
  521. "items": {
  522. "category": {"kind": "choice", "choice": "question", "answer_confidence": 0.91}
  523. },
  524. "content": {"summary": "wants action"},
  525. },
  526. }
  527. )
  528. assert state.triage_state == "ok"
  529. lines = state.triage_lines()
  530. assert lines[0] == "category: question (0.91)"
  531. assert "summary: wants action" in lines
  532. def test_cleared_event_after_lone_signals_leaves_state_usable():
  533. state = ViewState()
  534. state.ingest({"type": "triage", "stage": "signals", "triage": _signals_triage()})
  535. state.ingest({"type": "triage", "stage": "cleared"})
  536. assert state.triage is None
  537. assert state.triage_lines() == ["triage: unavailable — no data"]
  538. def test_triage_event_with_ts_records_last_triage_time():
  539. state = ViewState()
  540. state.ingest(
  541. {"type": "triage", "stage": "signals", "triage": _signals_triage(), "ts": 1759073916.19}
  542. )
  543. assert state.triage_ts == 1759073916.19
  544. def test_triage_event_without_ts_keeps_previous_time():
  545. state = ViewState()
  546. state.ingest(
  547. {"type": "triage", "stage": "signals", "triage": _signals_triage(), "ts": 1759073916.19}
  548. )
  549. state.ingest(
  550. {
  551. "type": "triage",
  552. "stage": "content",
  553. "triage": {**_signals_triage(), "content": {"summary": "s"}},
  554. }
  555. )
  556. assert state.triage_ts == 1759073916.19
  557. def test_triage_renderables_prepends_clock_in_transcript_time_style():
  558. ts = 1759073916.19
  559. state = ViewState()
  560. state.ingest({"type": "triage", "stage": "signals", "triage": _signals_triage(), "ts": ts})
  561. renderables = state.triage_renderables()
  562. assert renderables[0].plain == _fmt_clock(ts)
  563. assert str(renderables[0].spans[0].style) == TIME_STYLE
  564. assert [item.plain for item in renderables[1:]] == state.triage_lines()
  565. def test_triage_cleared_event_and_clear_output_drop_the_clock():
  566. ts = 1759073916.19
  567. state = ViewState()
  568. state.ingest({"type": "triage", "stage": "signals", "triage": _signals_triage(), "ts": ts})
  569. state.ingest({"type": "triage", "stage": "cleared"})
  570. assert state.triage_ts is None
  571. assert all(item.plain != _fmt_clock(ts) for item in state.triage_renderables())
  572. state.ingest({"type": "triage", "stage": "signals", "triage": _signals_triage(), "ts": ts})
  573. state.clear_output()
  574. assert state.triage_ts is None
  575. assert all(item.plain != _fmt_clock(ts) for item in state.triage_renderables())
  576. def test_triage_renderables_clock_only_for_ok_dict_triage():
  577. ts = 1759073916.19
  578. clock = _fmt_clock(ts)
  579. state = ViewState()
  580. state.ingest({"type": "triage", "stage": "signals", "triage": _signals_triage(), "ts": ts})
  581. state.triage_state = "loading"
  582. state.triage_reason = "warming up"
  583. assert state.triage_renderables()[0].plain == "triage: unavailable — warming up"
  584. assert all(item.plain != clock for item in state.triage_renderables())
  585. state.triage_state = "ok"
  586. state.triage = None
  587. assert state.triage_renderables()[0].plain == "triage: unavailable — no data"
  588. assert all(item.plain != clock for item in state.triage_renderables())
  589. def test_status_adoption_preserves_last_triage_time():
  590. ts = 1759073916.19
  591. state = ViewState()
  592. state.ingest({"type": "triage", "stage": "signals", "triage": _signals_triage(), "ts": ts})
  593. state.ingest(
  594. {
  595. "type": "status",
  596. "triage": _signals_triage(),
  597. "triage_state": "ok",
  598. "triage_reason": "",
  599. }
  600. )
  601. assert state.triage_ts == ts
  602. assert state.triage_renderables()[0].plain == _fmt_clock(ts)
  603. def _questions():
  604. return {
  605. "topic": {
  606. "type": "choice",
  607. "instructions": "What is the main theme?",
  608. "criteria": {
  609. "planning": "roadmap or prioritization",
  610. "problem": "a blocker or risk",
  611. },
  612. },
  613. "urgency": {
  614. "type": "choice",
  615. "instructions": "How urgent?",
  616. "criteria": ["low", "high"],
  617. },
  618. "sentiment": {
  619. "type": "score",
  620. "instructions": "Overall tone?",
  621. "criteria": ["negative", "neutral", "positive"],
  622. },
  623. "blocker": {
  624. "type": "noul",
  625. "instructions": "A blocker was raised.",
  626. "criteria": {"true": "an obstacle was raised", "false": "no obstacle"},
  627. },
  628. "owner_named": {
  629. "type": "noul",
  630. "instructions": "A responsible person was named.",
  631. },
  632. }
  633. def test_status_adopts_questions_and_bumps_questions_epoch_only():
  634. state = ViewState()
  635. triage_epoch = state.triage_epoch
  636. questions = _questions()
  637. state.ingest({"type": "status", "questions": questions})
  638. assert state.questions == questions
  639. assert state.questions_epoch == 1
  640. assert state.triage_epoch == triage_epoch
  641. def test_status_without_questions_preserves_loaded_questions():
  642. state = ViewState()
  643. questions = _questions()
  644. state.ingest({"type": "status", "questions": questions})
  645. state.ingest({"type": "status", "mode": "idle", "ready": True})
  646. assert state.questions == questions
  647. assert state.questions_epoch == 1
  648. def test_questions_lines_render_choice_dict_choice_list_score_and_noul():
  649. state = ViewState()
  650. state.questions = _questions()
  651. assert state.questions_lines() == [
  652. "topic [choice]",
  653. " What is the main theme?",
  654. " planning — roadmap or prioritization",
  655. " problem — a blocker or risk",
  656. "urgency [choice]",
  657. " How urgent?",
  658. " low",
  659. " high",
  660. "sentiment [score]",
  661. " Overall tone?",
  662. " 1. negative",
  663. " 2. neutral",
  664. " 3. positive",
  665. "blocker [noul]",
  666. " A blocker was raised.",
  667. " true — an obstacle was raised",
  668. " false — no obstacle",
  669. "owner_named [noul]",
  670. " A responsible person was named.",
  671. ]
  672. def test_questions_render_while_triage_lane_is_loading():
  673. state = ViewState()
  674. state.questions = _questions()
  675. state.triage_state = "loading"
  676. state.triage_reason = "warming up"
  677. assert state.questions_lines()[0] == "topic [choice]"
  678. assert not any("unavailable" in line for line in state.questions_lines())
  679. def test_questions_lines_unavailable_when_missing():
  680. state = ViewState()
  681. state.triage_state = "loading"
  682. state.triage_reason = "model warming up"
  683. assert state.questions_lines() == ["questions: unavailable — model warming up"]
  684. def test_questions_lines_unavailable_falls_back_to_state_then_generic():
  685. state = ViewState()
  686. assert state.questions_lines() == ["questions: unavailable — disabled"]
  687. state.triage_state = ""
  688. state.triage_reason = ""
  689. assert state.questions_lines() == ["questions: unavailable — no questions loaded"]
  690. def test_questions_lines_unavailable_for_empty_or_non_dict_payload():
  691. state = ViewState()
  692. state.triage_state = "no_questions"
  693. state.triage_reason = "no questions loaded from questions.json"
  694. for payload in (None, {}, [], "junk", 7):
  695. state.ingest({"type": "status", "questions": payload})
  696. assert state.questions_lines() == [
  697. "questions: unavailable — no questions loaded from questions.json"
  698. ]
  699. def test_questions_lines_malformed_entries_never_raise():
  700. state = ViewState()
  701. state.questions = {
  702. "good": {"type": "choice", "instructions": "pick", "criteria": {"a": "b"}},
  703. "not_a_dict": "boom",
  704. "no_instructions": {"type": "noul"},
  705. "bad_criteria": {"type": "score", "instructions": "level?", "criteria": {}},
  706. "empty_criteria": {"type": "choice", "instructions": "choose", "criteria": []},
  707. "no_type": {"instructions": "huh"},
  708. }
  709. lines = state.questions_lines()
  710. assert lines == [
  711. "good [choice]",
  712. " pick",
  713. " a — b",
  714. "no_instructions [noul]",
  715. "bad_criteria [score]",
  716. " level?",
  717. "empty_criteria [choice]",
  718. " choose",
  719. "no_type [?]",
  720. " huh",
  721. ]
  722. def test_questions_lines_all_malformed_falls_back_to_unavailable():
  723. state = ViewState()
  724. state.triage_state = "ok"
  725. state.questions = {"broken": "not a dict"}
  726. assert state.questions_lines() == ["questions: unavailable — ok"]
  727. def test_questions_renderables_plain_matches_lines_with_styled_header():
  728. state = ViewState()
  729. state.questions = _questions()
  730. renderables = state.questions_renderables()
  731. assert [item.plain for item in renderables] == state.questions_lines()
  732. header = renderables[0]
  733. assert header.plain == "topic [choice]"
  734. assert [(header.plain[s.start : s.end], str(s.style)) for s in header.spans] == [
  735. ("topic", "bold"),
  736. ("[choice]", "dim"),
  737. ]
  738. assert all(not item.spans for item in renderables[1:4])
  739. def test_questions_renderables_unavailable_is_plain_single_line():
  740. state = ViewState()
  741. state.triage_state = "loading"
  742. state.triage_reason = "warming up"
  743. renderables = state.questions_renderables()
  744. assert [item.plain for item in renderables] == [
  745. "questions: unavailable — warming up"
  746. ]
  747. assert not renderables[0].spans
  748. def test_clear_output_keeps_questions():
  749. state = ViewState()
  750. state.questions = _questions()
  751. state.ingest({"type": "triage", "stage": "signals", "triage": _signals_triage()})
  752. state.clear_output()
  753. assert state.questions == _questions()
  754. assert state.questions_epoch == 0
  755. assert state.questions_lines()[0] == "topic [choice]"