feat: manage multiple ssh machines from one client (#3670)

1 files changed, 25 insertions, 0 deletions

diff --git a/src/client/tests/mod.rs b/src/client/tests/mod.rs
index 5d789444..cca73232 100644
--- a/src/client/tests/mod.rs
+++ b/src/client/tests/mod.rs
@@ -156,6 +156,26 @@ fn host_cursor_policy_native_and_drawn_override_auto_detection() {
     ));
 }
 
+#[test]
+fn image_bridge_follows_the_selected_remote_endpoint() {
+    let remote = crate::client::endpoint::ClientEndpointId::Ssh(
+        crate::client::endpoint::ProfileId::parse("0123456789abcdef0123456789abcdef").unwrap(),
+    );
+
+    assert!(endpoint_accepts_local_images(false, &remote, true));
+    assert!(endpoint_accepts_local_images(
+        true,
+        &crate::client::endpoint::ClientEndpointId::Local,
+        true,
+    ));
+    assert!(!endpoint_accepts_local_images(
+        false,
+        &crate::client::endpoint::ClientEndpointId::Local,
+        true,
+    ));
+    assert!(!endpoint_accepts_local_images(false, &remote, false));
+}
+
 #[cfg(unix)]
 #[test]
 fn clipboard_image_paste_bridge_triggers_on_configured_key_and_empty_paste() {
@@ -181,6 +201,11 @@ fn clipboard_image_paste_bridge_triggers_on_configured_key_and_empty_paste() {
         Some(ctrl_v)
     ));
     assert!(!should_bridge_clipboard_image_paste(
+        &[0x16],
+        false,
+        Some(ctrl_v)
+    ));
+    assert!(!should_bridge_clipboard_image_paste(
         b"\x1b[200~text\x1b[201~",
         true,
         Some(ctrl_v)