@team-agent/installer 0.5.41 → 0.5.43

package/crates/team-agent/src/transport/tests/mod.rs

@@ -1,203 +1,194 @@
-    #![allow(clippy::unwrap_used, clippy::expect_used, clippy::panic)]
-    use super::*;
-    use std::collections::BTreeMap;
-    use std::path::Path;
+#![allow(clippy::unwrap_used, clippy::expect_used, clippy::panic)]
+use super::*;
+use std::collections::BTreeMap;
+use std::path::Path;
 
-    // —— 每个 daemon 可达方法 `unimplemented!()` 的 stub:驱动 trait 即 RED ——
-    struct StubBackend(BackendKind);
+// —— 每个 daemon 可达方法 `unimplemented!()` 的 stub:驱动 trait 即 RED ——
+struct StubBackend(BackendKind);
 
-    impl Transport for StubBackend {
-        fn kind(&self) -> BackendKind {
-            self.0
-        }
-        fn spawn_first(
-            &self,
-            session: &SessionName,
-            window: &WindowName,
-            argv: &[String],
-            cwd: &Path,
-            env: &BTreeMap<String, String>,
-        ) -> Result<SpawnResult, TransportError> {
-            let _ = (argv, cwd, env);
-            Ok(SpawnResult {
-                pane_id: PaneId::new(format!("%{}-{}", session.as_str(), window.as_str())),
-                session: session.clone(),
-                window: window.clone(),
-                child_pid: Some(1000),
-            })
-        }
-        fn spawn_into(
-            &self,
-            session: &SessionName,
-            window: &WindowName,
-            argv: &[String],
-            cwd: &Path,
-            env: &BTreeMap<String, String>,
-        ) -> Result<SpawnResult, TransportError> {
-            let _ = (argv, cwd, env);
-            Ok(SpawnResult {
-                pane_id: PaneId::new(format!("%{}-{}", session.as_str(), window.as_str())),
-                session: session.clone(),
-                window: window.clone(),
-                child_pid: Some(1001),
-            })
-        }
-        fn inject(
-            &self,
-            target: &Target,
-            payload: &InjectPayload,
-            submit: Key,
-            bracketed: bool,
-        ) -> Result<InjectReport, TransportError> {
-            let _ = (target, bracketed);
-            let (inject_verification, turn_verification) = match payload {
-                InjectPayload::Empty => (
-                    InjectVerification::EmptyTextSendKeys,
-                    TurnVerification::NotRequired,
-                ),
-                InjectPayload::Text(text) | InjectPayload::TextSkipConsumptionPoll(text)
-                    if text.contains("[team-agent-token:") =>
+impl Transport for StubBackend {
+    fn kind(&self) -> BackendKind {
+        self.0
+    }
+    fn spawn_first(
+        &self,
+        session: &SessionName,
+        window: &WindowName,
+        argv: &[String],
+        cwd: &Path,
+        env: &BTreeMap<String, String>,
+    ) -> Result<SpawnResult, TransportError> {
+        let _ = (argv, cwd, env);
+        Ok(SpawnResult {
+            pane_id: PaneId::new(format!("%{}-{}", session.as_str(), window.as_str())),
+            session: session.clone(),
+            window: window.clone(),
+            child_pid: Some(1000),
+        })
+    }
+    fn spawn_into(
+        &self,
+        session: &SessionName,
+        window: &WindowName,
+        argv: &[String],
+        cwd: &Path,
+        env: &BTreeMap<String, String>,
+    ) -> Result<SpawnResult, TransportError> {
+        let _ = (argv, cwd, env);
+        Ok(SpawnResult {
+            pane_id: PaneId::new(format!("%{}-{}", session.as_str(), window.as_str())),
+            session: session.clone(),
+            window: window.clone(),
+            child_pid: Some(1001),
+        })
+    }
+    fn inject(
+        &self,
+        target: &Target,
+        payload: &InjectPayload,
+        submit: Key,
+        bracketed: bool,
+    ) -> Result<InjectReport, TransportError> {
+        let _ = (target, bracketed);
+        let (inject_verification, turn_verification) = match payload {
+            InjectPayload::Empty => (
+                InjectVerification::EmptyTextSendKeys,
+                TurnVerification::NotRequired,
+            ),
+            InjectPayload::Text(text) | InjectPayload::TextSkipConsumptionPoll(text)
+                if text.contains("[team-agent-token:") =>
+            {
                 (
                     InjectVerification::CaptureContainsToken,
                     TurnVerification::NotYetObserved,
-                ),
-                InjectPayload::Text(_) | InjectPayload::TextSkipConsumptionPoll(_) => (
-                    InjectVerification::NoToken,
-                    TurnVerification::NotYetObserved,
-                ),
-            };
-            let submit_verification = match submit {
-                Key::Enter => SubmitVerification::EnterSentWithoutPlaceholderCheck,
-                key => SubmitVerification::KeySentAfterVisibleToken { key },
-            };
-            Ok(InjectReport {
-                stage_reached: InjectStage::Submit,
-                inject_verification,
-                submit_verification,
-                turn_verification,
-                attempts: 1,
-                submit_diagnostics: None,
-            })
-        }
-        fn send_keys(&self, _target: &Target, _keys: &[Key]) -> Result<(), TransportError> {
-            Ok(())
-        }
-        fn capture(
-            &self,
-            target: &Target,
-            range: CaptureRange,
-        ) -> Result<CapturedText, TransportError> {
-            let _ = target;
-            Ok(CapturedText {
-                text: normalize_capture("line one  \nbusy\u{a0}marker   \n  \n"),
-                range,
-            })
-        }
-        fn query(
-            &self,
-            _target: &Target,
-            field: PaneField,
-        ) -> Result<Option<String>, TransportError> {
-            if self.0 != BackendKind::Tmux && field == PaneField::PaneMode {
-                return Ok(None);
-            }
-            let value = match field {
-                PaneField::PaneId => "%7",
-                PaneField::PaneMode => "0",
-                PaneField::PaneWidth => "120",
-                PaneField::PaneCurrentCommand => "sh",
-                PaneField::PaneCurrentPath => "/tmp/ws",
-                PaneField::SessionName => "team-sess",
-                PaneField::PaneTty => "/dev/ttys000",
-            };
-            Ok(Some(value.to_string()))
-        }
-        fn liveness(&self, pane: &PaneId) -> Result<PaneLiveness, TransportError> {
-            if self.0 == BackendKind::ConPty && pane.as_str() == "foreign-leader" {
-                return Ok(PaneLiveness::Unknown);
+                )
             }
-            Ok(PaneLiveness::Live)
-        }
-        fn list_targets(&self) -> Result<Vec<PaneInfo>, TransportError> {
-            Ok(vec![
-                PaneInfo {
-                    pane_id: PaneId::new("%team-sess-win-1"),
-                    session: SessionName::new("team-sess"),
-                    window_index: Some(0),
-                    window_name: Some(WindowName::new("win-1")),
-                    pane_index: Some(0),
-                    tty: Some("/dev/ttys001".to_string()),
-                    current_command: Some("sh".to_string()),
-                    current_path: Some(PathBuf::from("/tmp/ws")),
-                    active: true,
-                    pane_pid: Some(1000),
-                    leader_env: BTreeMap::new(),
-                },
-                PaneInfo {
-                    pane_id: PaneId::new("%team-sess-worker-2"),
-                    session: SessionName::new("team-sess"),
-                    window_index: Some(1),
-                    window_name: Some(WindowName::new("worker-2")),
-                    pane_index: Some(0),
-                    tty: Some("/dev/ttys002".to_string()),
-                    current_command: Some("sh".to_string()),
-                    current_path: Some(PathBuf::from("/tmp/ws")),
-                    active: true,
-                    pane_pid: Some(1001),
-                    leader_env: BTreeMap::new(),
-                },
-            ])
-        }
-        fn has_session(&self, _session: &SessionName) -> Result<bool, TransportError> {
-            Ok(true)
-        }
-        fn list_windows(
-            &self,
-            _session: &SessionName,
-        ) -> Result<Vec<WindowName>, TransportError> {
-            Ok(vec![WindowName::new("win-1"), WindowName::new("worker-2")])
-        }
-        fn set_session_env(
-            &self,
-            _session: &SessionName,
-            _key: &str,
-            _value: &str,
-        ) -> Result<SetEnvOutcome, TransportError> {
-            match self.0 {
-                BackendKind::Tmux => Ok(SetEnvOutcome::Applied),
-                BackendKind::WezTerm | BackendKind::ConPty => Ok(SetEnvOutcome::InternalizedAtSpawn),
-            }
-        }
-        fn kill_session(&self, _session: &SessionName) -> Result<(), TransportError> {
-            Ok(())
+            InjectPayload::Text(_) | InjectPayload::TextSkipConsumptionPoll(_) => (
+                InjectVerification::NoToken,
+                TurnVerification::NotYetObserved,
+            ),
+        };
+        let submit_verification = match submit {
+            Key::Enter => SubmitVerification::EnterSentWithoutPlaceholderCheck,
+            key => SubmitVerification::KeySentAfterVisibleToken { key },
+        };
+        Ok(InjectReport {
+            stage_reached: InjectStage::Submit,
+            inject_verification,
+            submit_verification,
+            turn_verification,
+            attempts: 1,
+            submit_diagnostics: None,
+        })
+    }
+    fn send_keys(&self, _target: &Target, _keys: &[Key]) -> Result<(), TransportError> {
+        Ok(())
+    }
+    fn capture(
+        &self,
+        target: &Target,
+        range: CaptureRange,
+    ) -> Result<CapturedText, TransportError> {
+        let _ = target;
+        Ok(CapturedText {
+            text: normalize_capture("line one  \nbusy\u{a0}marker   \n  \n"),
+            range,
+        })
+    }
+    fn query(&self, _target: &Target, field: PaneField) -> Result<Option<String>, TransportError> {
+        if self.0 != BackendKind::Tmux && field == PaneField::PaneMode {
+            return Ok(None);
         }
-        fn kill_window(&self, _target: &Target) -> Result<(), TransportError> {
-            Ok(())
+        let value = match field {
+            PaneField::PaneId => "%7",
+            PaneField::PaneMode => "0",
+            PaneField::PaneWidth => "120",
+            PaneField::PaneCurrentCommand => "sh",
+            PaneField::PaneCurrentPath => "/tmp/ws",
+            PaneField::SessionName => "team-sess",
+            PaneField::PaneTty => "/dev/ttys000",
+        };
+        Ok(Some(value.to_string()))
+    }
+    fn liveness(&self, pane: &PaneId) -> Result<PaneLiveness, TransportError> {
+        if self.0 == BackendKind::ConPty && pane.as_str() == "foreign-leader" {
+            return Ok(PaneLiveness::Unknown);
         }
-        fn attach_session(
-            &self,
-            _session: &SessionName,
-        ) -> Result<AttachOutcome, TransportError> {
-            match self.0 {
-                BackendKind::Tmux => Ok(AttachOutcome::Attached),
-                BackendKind::WezTerm => Ok(AttachOutcome::GuiAttachIsImplicit),
-                BackendKind::ConPty => Ok(AttachOutcome::Unsupported {
-                    reason: "conpty has no attach concept".to_string(),
-                }),
-            }
+        Ok(PaneLiveness::Live)
+    }
+    fn list_targets(&self) -> Result<Vec<PaneInfo>, TransportError> {
+        Ok(vec![
+            PaneInfo {
+                pane_id: PaneId::new("%team-sess-win-1"),
+                session: SessionName::new("team-sess"),
+                window_index: Some(0),
+                window_name: Some(WindowName::new("win-1")),
+                pane_index: Some(0),
+                tty: Some("/dev/ttys001".to_string()),
+                current_command: Some("sh".to_string()),
+                current_path: Some(PathBuf::from("/tmp/ws")),
+                active: true,
+                pane_pid: Some(1000),
+                leader_env: BTreeMap::new(),
+            },
+            PaneInfo {
+                pane_id: PaneId::new("%team-sess-worker-2"),
+                session: SessionName::new("team-sess"),
+                window_index: Some(1),
+                window_name: Some(WindowName::new("worker-2")),
+                pane_index: Some(0),
+                tty: Some("/dev/ttys002".to_string()),
+                current_command: Some("sh".to_string()),
+                current_path: Some(PathBuf::from("/tmp/ws")),
+                active: true,
+                pane_pid: Some(1001),
+                leader_env: BTreeMap::new(),
+            },
+        ])
+    }
+    fn has_session(&self, _session: &SessionName) -> Result<bool, TransportError> {
+        Ok(true)
+    }
+    fn list_windows(&self, _session: &SessionName) -> Result<Vec<WindowName>, TransportError> {
+        Ok(vec![WindowName::new("win-1"), WindowName::new("worker-2")])
+    }
+    fn set_session_env(
+        &self,
+        _session: &SessionName,
+        _key: &str,
+        _value: &str,
+    ) -> Result<SetEnvOutcome, TransportError> {
+        match self.0 {
+            BackendKind::Tmux => Ok(SetEnvOutcome::Applied),
+            BackendKind::WezTerm | BackendKind::ConPty => Ok(SetEnvOutcome::InternalizedAtSpawn),
         }
     }
-
-    fn tmux() -> StubBackend {
-        StubBackend(BackendKind::Tmux)
+    fn kill_session(&self, _session: &SessionName) -> Result<(), TransportError> {
+        Ok(())
     }
-    fn wezterm() -> StubBackend {
-        StubBackend(BackendKind::WezTerm)
+    fn kill_window(&self, _target: &Target) -> Result<(), TransportError> {
+        Ok(())
     }
-    fn conpty() -> StubBackend {
-        StubBackend(BackendKind::ConPty)
+    fn attach_session(&self, _session: &SessionName) -> Result<AttachOutcome, TransportError> {
+        match self.0 {
+            BackendKind::Tmux => Ok(AttachOutcome::Attached),
+            BackendKind::WezTerm => Ok(AttachOutcome::GuiAttachIsImplicit),
+            BackendKind::ConPty => Ok(AttachOutcome::Unsupported {
+                reason: "conpty has no attach concept".to_string(),
+            }),
+        }
     }
+}
 
+fn tmux() -> StubBackend {
+    StubBackend(BackendKind::Tmux)
+}
+fn wezterm() -> StubBackend {
+    StubBackend(BackendKind::WezTerm)
+}
+fn conpty() -> StubBackend {
+    StubBackend(BackendKind::ConPty)
+}
 
-mod wire;
 mod behavior;
+mod wire;