Make TUI status and resume guards reflect the real test surface

Turn a stubbed status path into a real heartbeat lookup, convert controller-cost wiring from private-state mutation into a wiring assertion, and remove two skips that were only masking ordinary edge cases.

Constraint: Resume replay remains partially unimplemented; the remaining skip markers are still attached to real work, not test laziness.
Rejected: Leave the status handler returning unknown | that keeps a user-visible command from proving anything useful.
Confidence: high
Scope-risk: narrow
Directive: Keep future tests focused on public commands or explicit wiring, not direct state injection unless the private state itself is the contract.
Tested: git diff --check; python3 scripts/validate_docs_consistency.py; python3 -m pytest tests/test_tui.py tests/test_cli_chat.py -q
Not-tested: the remaining resume replay skips and broader suite-wide skip inventory were not fully remediated in this pass

Author: johnteee

teaagent/tui/_commands.py

@@ -174,11 +174,11 @@ def _handle_tui_command(tui: 'TeaAgentTUI', raw_command: str) -> bool:
             tui.output_fn('error: status requires a run id')
             return True
         run_id = args[0]
-        from teaagent.run_store import RunStore
-
         store = RunStore(tui.root)  # type: ignore[assignment]
-        # Simplified status check
-        tui._print_json({'run_id': run_id, 'status': 'unknown'})
+        try:
+            tui._print_json(store.heartbeat_for_run(run_id))
+        except FileNotFoundError:
+            tui.output_fn(f"error: run '{run_id}' not found")
         return True
 
     if action == 'plan':
@@ -246,6 +246,10 @@ def _handle_tui_command(tui: 'TeaAgentTUI', raw_command: str) -> bool:
             tui.output_fn('error: resume requires a run id')
             return True
         run_id = args[0]
+        store = RunStore(tui.root)
+        if not store.run_path(run_id).is_file():
+            tui.output_fn(f"error: run '{run_id}' not found")
+            return True
         _safe_run_agent_task(tui, '', resumed_from=run_id)
         return True
 

tests/test_tui.py

@@ -785,9 +785,6 @@ def test_tui_resume_requires_run_id(self) -> None:
         self.assertTrue(tui.handle_command('resume'))
         self.assertIn('requires a run id', output[0])
 
-    @pytest.mark.skip(
-        reason='Resume unknown run ID handling not working - requires investigation'
-    )
     def test_tui_resume_unknown_run_id(self) -> None:
         with tempfile.TemporaryDirectory() as tmp:
             output = []
@@ -905,7 +902,6 @@ def test_tui_print_header_output(self) -> None:
 
         self.assertEqual(output[0], 'TeaAgent TUI 0.1.0')
 
-    @pytest.mark.skip(reason='TUI status command not working - requires investigation')
     def test_tui_status_with_valid_run(self) -> None:
         with tempfile.TemporaryDirectory() as tmp:
             output = []
@@ -955,15 +951,9 @@ def test_tui_uses_chat_session_controller_for_cost_tracking(self) -> None:
             # Verify chat controller is created
             controller = tui._get_chat_controller()
             self.assertIsNotNone(controller)
+            self.assertIs(controller.session_state, tui._session_state)
             self.assertEqual(controller.session_state.session_cost_cents, 0.0)
-
-            # Simulate cost accumulation through controller (as would happen in real execution)
-            controller.session_state.session_cost_cents += 50.0
-            tui._session_cost_cents = controller.session_state.session_cost_cents
-
-            # Verify cost was tracked through controller
-            self.assertEqual(tui._session_cost_cents, 50.0)
-            self.assertEqual(controller.session_state.session_cost_cents, 50.0)
+            self.assertIs(tui._get_chat_controller(), controller)
 
     def test_tui_cost_command_shows_controller_cost(self) -> None:
         """TUI /cost should reflect controller state and keep currency formatting."""