@team-agent/installer 0.5.77 → 0.5.80

package/crates/team-agent/src/tmux_backend/tests.rs

@@ -28,10 +28,10 @@ use crate::layout::worker_env::{
 use crate::model::enums::{PaneLiveness, Provider};
 use crate::transport::{
     command_basename, normalize_capture, tmux_capture_argv, tmux_query_argv, tmux_send_keys_argv,
-    tmux_spawn_argv, tmux_submit_key_name, AttachOutcome, CaptureRange, CaptureSampleOutcome,
-    InjectPayload, InjectStage, InjectVerification, InputSurfaceProbe, Key, PaneField, PaneId,
-    QueryOutcome, SessionName, SetEnvOutcome, SubmitVerification, Target, Transport,
-    TransportError, TurnVerification, WindowName,
+    tmux_spawn_argv, tmux_submit_key_name, AttachOutcome, BusySignalKind, CaptureRange,
+    CaptureSampleOutcome, InjectPayload, InjectStage, InjectVerification, InputSurfaceProbe, Key,
+    PaneField, PaneId, QueryOutcome, SessionName, SetEnvOutcome, SubmitConsumptionReason,
+    SubmitVerification, Target, Transport, TransportError, TurnVerification, WindowName,
 };
 
 type RecordedArgv = Arc<Mutex<Vec<Vec<String>>>>;
@@ -355,6 +355,34 @@ fn backend_with(default: MockResp, queued: Vec<MockResp>) -> (TmuxBackend, Recor
     (TmuxBackend::with_runner(Box::new(runner)), recorded)
 }
 
+#[test]
+fn provider_exit_receipt_accepts_only_shell_exit_status() {
+    for (text, expected) in [
+        ("", None),
+        ("running", None),
+        ("[team-agent] pi exited with 0", None),
+        ("256", None),
+        ("0\n", Some(0)),
+        ("17\n", Some(17)),
+        ("143\n", Some(143)),
+    ] {
+        let (backend, recorded) = backend_with(MockResp::Out(ok(text)), Vec::new());
+        assert_eq!(
+            backend.provider_exit_status(&PaneId::new("%47")).unwrap(),
+            expected
+        );
+        let calls = recorded.lock().unwrap();
+        assert_eq!(calls.len(), 1);
+        assert_eq!(calls[0][4], "%47");
+        assert_eq!(calls[0][5], "#{@team-agent-provider-exit-status}");
+    }
+    let (backend, _) = backend_with(MockResp::Out(fail(1, "pane unavailable")), Vec::new());
+    assert_eq!(
+        backend.provider_exit_status(&PaneId::new("%47")).unwrap(),
+        None
+    );
+}
+
 fn backend_with_stdin(
     default: MockResp,
     queued: Vec<MockResp>,
@@ -1051,6 +1079,15 @@ fn inject_text_runs_buffer_paste_submit_sequence_and_reports_submit() {
         SubmitVerification::EnterSentWithoutPlaceholderCheck
     );
     assert_eq!(report.turn_verification, TurnVerification::NotYetObserved);
+    let diagnostics = report.submit_diagnostics.expect("diagnostics");
+    assert_eq!(
+        diagnostics.consumption_reason,
+        SubmitConsumptionReason::NoMarker
+    );
+    assert_eq!(diagnostics.busy_signal_kind, None);
+    assert_eq!(diagnostics.current_marker_in_bottom_15, None);
+    assert_eq!(diagnostics.paste_identity_in_composer, None);
+    assert_eq!(diagnostics.consumption_from_capture_result, None);
 }
 
 #[test]
@@ -1409,10 +1446,18 @@ fn e46_post_submit_matched_token_without_scroll_is_unverified() {
         "token still in composer and no busy must report 没开跑, not unknown/started"
     );
     let diagnostics = report.submit_diagnostics.expect("diagnostics");
+    assert_eq!(diagnostics.busy_signal_kind, None);
+    assert_eq!(diagnostics.current_marker_in_bottom_15, Some(true));
+    assert_eq!(diagnostics.paste_identity_in_composer, None);
+    assert_eq!(diagnostics.consumption_from_capture_result, Some(false));
     assert!(
         diagnostics.attempts_detail.iter().any(|obs| obs.matched),
         "the unverified verdict must still record that paste landed"
     );
+    assert_eq!(
+        diagnostics.consumption_reason,
+        SubmitConsumptionReason::Unverified
+    );
 }
 
 #[test]
@@ -1452,6 +1497,56 @@ fn e46_unconsumed_token_with_live_busy_state_is_treated_as_processing() {
         "busy-state capture should be recorded in attempts_detail: {:?}",
         diagnostics.attempts_detail
     );
+    assert_eq!(
+        diagnostics.consumption_reason,
+        SubmitConsumptionReason::FallbackBusy
+    );
+    assert_eq!(diagnostics.before_busy_signal_kind, None);
+    assert_eq!(diagnostics.busy_signal_kind, Some(BusySignalKind::Working));
+    assert_eq!(diagnostics.busy_line_from_bottom, Some(0));
+    assert_eq!(diagnostics.current_marker_in_bottom_15, Some(true));
+    assert_eq!(diagnostics.paste_identity_in_composer, None);
+    assert_eq!(diagnostics.consumption_from_capture_result, Some(false));
+}
+
+#[test]
+fn e46_capture_error_clears_previous_consumption_observation() {
+    let token_text = "Team Agent message from leader:\n\nhi\n\n[team-agent-token:msg_capture_error]";
+    let busy_tail = format!("{token_text}\n● Working (1s · esc to interrupt)\n");
+    let visible = ok(token_text);
+    let queued = vec![
+        MockResp::Out(ok("")),         // set-buffer
+        MockResp::Out(ok("")),         // paste-buffer
+        MockResp::Out(ok("")),         // delete-buffer
+        MockResp::Out(visible.clone()), // pasted-content prompt: continue
+        MockResp::Out(visible),         // pre-submit token visibility
+        MockResp::Out(ok("")),         // Escape
+        MockResp::Out(ok("")),         // Enter
+        MockResp::Out(ok(&busy_tail)),  // first post-Enter capture: busy
+        // Default capture error forces the next poll/fallback/resend reads to fail.
+    ];
+    let (be, _rec) = backend_with(MockResp::Io(std::io::ErrorKind::Other), queued);
+    let report = be
+        .inject(
+            &Target::Pane(PaneId::new("%7")),
+            &InjectPayload::Text(token_text.to_string()),
+            Key::Enter,
+            true,
+        )
+        .expect("inject runs through capture failure");
+    let diagnostics = report.submit_diagnostics.expect("diagnostics");
+    assert_eq!(
+        report.submit_verification,
+        SubmitVerification::SubmitConsumptionUnverified
+    );
+    assert_eq!(diagnostics.consumption_reason, SubmitConsumptionReason::Unverified);
+    assert!(diagnostics.capture_err_count > 0);
+    assert_eq!(diagnostics.before_busy_signal_kind, None);
+    assert_eq!(diagnostics.busy_signal_kind, None);
+    assert_eq!(diagnostics.busy_line_from_bottom, None);
+    assert_eq!(diagnostics.current_marker_in_bottom_15, None);
+    assert_eq!(diagnostics.paste_identity_in_composer, None);
+    assert_eq!(diagnostics.consumption_from_capture_result, None);
 }
 
 /// NeverSeen + 无占位符：空 pane 不得再把「token 不在底部」写成 consumed。
@@ -1521,6 +1616,14 @@ fn e46_inject_text_with_token_consumed_after_enter_keeps_enter_sent_without_plac
              hold (provider_submit_verification_red.rs:113-159). Got {:?}",
         report.submit_verification
     );
+    let diagnostics = report.submit_diagnostics.expect("diagnostics");
+    assert_eq!(
+        diagnostics.consumption_reason,
+        SubmitConsumptionReason::ConsumptionFromCapture
+    );
+    assert_eq!(diagnostics.busy_signal_kind, None);
+    assert_eq!(diagnostics.current_marker_in_bottom_15, Some(false));
+    assert_eq!(diagnostics.consumption_from_capture_result, Some(true));
 }
 
 /// **E46 RED-3 (resend-to-cap, no double-submit)**: when the FIRST Enter
@@ -3289,6 +3392,15 @@ fn cursor_single_enter_busy_transcript_placeholder_does_not_retry() {
         "busy after first Enter is consumption for cursor; got {:?}",
         report.submit_verification
     );
+    let diagnostics = report.submit_diagnostics.expect("diagnostics");
+    assert_eq!(
+        diagnostics.consumption_reason,
+        SubmitConsumptionReason::CursorPollingBusy
+    );
+    assert_eq!(diagnostics.busy_signal_kind, Some(BusySignalKind::Working));
+    assert!(diagnostics.busy_line_from_bottom.is_some());
+    assert!(diagnostics.current_marker_in_bottom_15.is_some());
+    assert!(diagnostics.consumption_from_capture_result.is_some());
     let calls = rec.lock().unwrap().clone();
     assert_eq!(
         count_submit_enters(&calls),
@@ -3570,6 +3682,39 @@ fn grok_fold_worker_text_payload_polls_consumption() {
     );
 }
 
+#[test]
+fn provider_busy_signal_match_preserves_category_priority_and_tail_boundary() {
+    let cases = [
+        ("working", BusySignalKind::Working),
+        ("thinking", BusySignalKind::Thinking),
+        ("processing", BusySignalKind::Processing),
+        ("esc to interrupt", BusySignalKind::EscToInterrupt),
+        ("⠋", BusySignalKind::SpinnerGlyph),
+        ("working ⠋", BusySignalKind::Working),
+    ];
+    for (text, expected) in cases {
+        let matched = super::provider_busy_signal_match(text).expect("busy match");
+        assert_eq!(matched.kind, expected, "text={text:?}");
+        assert_eq!(matched.line_from_bottom, 0, "text={text:?}");
+        assert!(super::provider_busy_signal_in_tail(text));
+    }
+
+    let inside_bottom_15 = std::iter::once("working".to_string())
+        .chain((0..14).map(|index| format!("tail-{index}")))
+        .collect::<Vec<_>>()
+        .join("\n");
+    let matched = super::provider_busy_signal_match(&inside_bottom_15).expect("15th line");
+    assert_eq!(matched.kind, BusySignalKind::Working);
+    assert_eq!(matched.line_from_bottom, 14);
+
+    let outside_bottom_15 = std::iter::once("working".to_string())
+        .chain((0..15).map(|index| format!("tail-{index}")))
+        .collect::<Vec<_>>()
+        .join("\n");
+    assert!(super::provider_busy_signal_match(&outside_bottom_15).is_none());
+    assert!(!super::provider_busy_signal_in_tail(&outside_bottom_15));
+}
+
 const G4_TOKEN: &str = "[team-agent-token:msg_g4]";
 
 #[test]