fix: honor space row gaps across machines (#3740)

* fix: honor space row gaps across machines

refs #3738

* test: cover scrolling with machine space gaps

refs #3738

---------

Co-authored-by: akbash-bot <300245827+akbash-bot@users.noreply.github.com>

1 files changed, 109 insertions, 0 deletions

diff --git a/src/client/shell/tests/endpoints.rs b/src/client/shell/tests/endpoints.rs
index a3a80979..2e5e87ac 100644
--- a/src/client/shell/tests/endpoints.rs
+++ b/src/client/shell/tests/endpoints.rs
@@ -438,6 +438,115 @@ fn saved_machine_preserves_endpoint_scoped_worktree_collapses() {
 }
 
 #[test]
+fn expanded_machine_sidebar_applies_space_row_gap_within_each_machine() {
+    let (mut state, remote_id) = state_with_remote();
+    state.config.spaces.row_gap = 1;
+
+    let add_second_workspace = |snapshot: &mut ClientShellSnapshot| {
+        let mut workspace = snapshot.workspaces[0].clone();
+        workspace.workspace_id = "ws_2".into();
+        workspace.active_tab_id = "tab_2".into();
+        workspace.number = 2;
+        workspace.label = "second-workspace".into();
+        workspace.focused = false;
+        snapshot.workspaces.push(workspace);
+    };
+    let mut local = snapshot();
+    add_second_workspace(&mut local);
+    state.set_snapshot(Box::new(local));
+    let mut remote = snapshot();
+    remote.boot_id = "remote-boot".into();
+    remote.workspaces[0].worktree = Some(ClientShellWorktree {
+        key: "repo".into(),
+        label: "repo".into(),
+        is_linked_worktree: false,
+    });
+    add_second_workspace(&mut remote);
+    remote.workspaces[1].worktree = Some(ClientShellWorktree {
+        key: "repo".into(),
+        label: "repo".into(),
+        is_linked_worktree: true,
+    });
+    let mut third = remote.workspaces[1].clone();
+    third.workspace_id = "ws_3".into();
+    third.number = 3;
+    third.label = "third-workspace".into();
+    third.worktree = None;
+    remote.workspaces.push(third);
+    state.set_endpoint_snapshot(&remote_id, Box::new(remote));
+
+    state.compose(100, 40).expect("combined endpoint frame");
+    let local_workspaces = state
+        .hits
+        .workspaces
+        .iter()
+        .filter(|hit| hit.endpoint_id.is_local())
+        .collect::<Vec<_>>();
+    assert_eq!(local_workspaces.len(), 2);
+    assert_eq!(
+        local_workspaces[1].rect.y,
+        local_workspaces[0].rect.bottom() + 1
+    );
+
+    let local_machine = state
+        .hits
+        .machines
+        .iter()
+        .find(|hit| hit.endpoint_id.is_local())
+        .expect("local machine");
+    let remote_machine = state
+        .hits
+        .machines
+        .iter()
+        .find(|hit| hit.endpoint_id == remote_id)
+        .expect("remote machine");
+    assert_eq!(local_workspaces[0].rect.y, local_machine.rect.bottom());
+    assert_eq!(remote_machine.rect.y, local_workspaces[1].rect.bottom());
+
+    let remote_workspaces = state
+        .hits
+        .workspaces
+        .iter()
+        .filter(|hit| hit.endpoint_id == remote_id)
+        .collect::<Vec<_>>();
+    assert_eq!(remote_workspaces.len(), 3);
+    assert_eq!(remote_workspaces[0].rect.y, remote_machine.rect.bottom());
+    assert_eq!(
+        remote_workspaces[1].rect.y,
+        remote_workspaces[0].rect.bottom()
+    );
+    assert_eq!(
+        remote_workspaces[2].rect.y,
+        remote_workspaces[1].rect.bottom() + 1
+    );
+
+    state.workspace_scroll = usize::MAX;
+    state.compose(100, 18).expect("scrolled endpoint frame");
+    let metrics = state
+        .hits
+        .workspace_scroll_metrics
+        .expect("workspace scroll metrics");
+    assert!(metrics.max_offset_from_bottom > 0);
+    assert_eq!(metrics.offset_from_bottom, 0);
+    assert_eq!(state.workspace_scroll, metrics.max_offset_from_bottom);
+    let visible_remote = state
+        .hits
+        .workspaces
+        .iter()
+        .filter(|hit| hit.endpoint_id == remote_id)
+        .collect::<Vec<_>>();
+    assert_eq!(visible_remote.len(), 3);
+    let gap_y = visible_remote[1].rect.bottom();
+    assert_eq!(visible_remote[2].rect.y, gap_y + 1);
+    assert!(visible_remote[2].rect.bottom() <= state.hits.workspace_body.bottom());
+    assert!(state
+        .hits
+        .workspaces
+        .iter()
+        .all(|hit| gap_y < hit.rect.top() || gap_y >= hit.rect.bottom()));
+}
+
+#[test]
 fn active_workspace_is_the_only_highlight_when_machine_is_expanded() {
     let (mut state, endpoint_id) = state_with_remote();
     assert!(state.activate_endpoint_projection(&endpoint_id));