aboutsummaryrefslogtreecommitdiff
path: root/src/client/shell/tests
diff options
context:
space:
mode:
Diffstat (limited to 'src/client/shell/tests')
-rw-r--r--src/client/shell/tests/close_tab.rs241
-rw-r--r--src/client/shell/tests/graphics.rs1
-rw-r--r--src/client/shell/tests/mod.rs1
3 files changed, 243 insertions, 0 deletions
diff --git a/src/client/shell/tests/close_tab.rs b/src/client/shell/tests/close_tab.rs
new file mode 100644
index 00000000..d5e897a3
--- /dev/null
+++ b/src/client/shell/tests/close_tab.rs
@@ -0,0 +1,241 @@
+use super::*;
+use crate::api::schema::Method;
+use crossterm::event::{MouseButton, MouseEventKind};
+
+fn close_state(confirm: bool, tab_count: usize) -> ClientShellState {
+ let mut projected = snapshot();
+ for number in 2..=tab_count {
+ let mut tab = projected.tabs[0].clone();
+ tab.tab_id = format!("tab_{number}");
+ tab.number = number;
+ tab.focused = false;
+ projected.tabs.push(tab);
+ }
+ let mut state = ClientShellState::new(ClientShellConfig::from_config(&Config::default()));
+ state.config.confirm_close = confirm;
+ state.set_snapshot(Box::new(projected));
+ state.set_pane_surface(surface());
+ state.compose(106, 24).unwrap();
+ state
+}
+
+fn click(state: &mut ClientShellState, rect: Rect) -> ClientShellInput {
+ state.handle_raw_events(vec![crate::raw_input::RawInputEvent::Mouse(MouseEvent {
+ kind: MouseEventKind::Down(MouseButton::Left),
+ column: rect.x,
+ row: rect.y,
+ modifiers: KeyModifiers::empty(),
+ })])
+}
+
+fn request_close(state: &mut ClientShellState, menu: bool) -> ClientShellInput {
+ if menu {
+ state.open_tab_context_menu("tab_1".into(), 30, 1);
+ state.compose(106, 24).unwrap();
+ let close_row = state.hits.context_menu_rows[2].0;
+ click(state, close_row)
+ } else {
+ let mut outcome = ClientShellInput::default();
+ state.record_binding(
+ crate::input::KeybindMatch::Action(crate::input::KeybindAction::CloseTab),
+ &mut outcome,
+ );
+ outcome
+ }
+}
+
+fn assert_no_close(outcome: &ClientShellInput) {
+ assert!(outcome.requests.is_empty());
+ assert!(outcome.actions.iter().all(|action| {
+ !matches!(action, ClientShellAction::Endpoint { request, .. }
+ if matches!(request.method, Method::TabClose(_) | Method::WorkspaceClose(_)))
+ }));
+}
+
+fn assert_tab_close(outcome: &ClientShellInput) {
+ let methods = outcome
+ .actions
+ .iter()
+ .filter_map(|action| match action {
+ ClientShellAction::Endpoint { request, .. } => Some(&request.method),
+ _ => None,
+ })
+ .filter(|method| !matches!(method, Method::TabFocus(_)))
+ .collect::<Vec<_>>();
+ assert!(matches!(methods.as_slice(), [Method::TabClose(target)] if target.tab_id == "tab_1"));
+}
+
+#[test]
+fn last_tab_close_waits_for_keyboard_or_mouse_confirmation() {
+ for menu in [false, true] {
+ let mut state = close_state(true, 1);
+ let requested = request_close(&mut state, menu);
+ assert_no_close(&requested);
+ assert!(requested.repaint);
+ assert!(matches!(
+ state.overlay,
+ Some(ClientShellOverlay::ConfirmClose(_))
+ ));
+ let frame = state.compose(106, 24).unwrap();
+ let text = frame_rows(&frame).join("\n");
+ assert!(text.contains("Close workspace?"));
+ assert!(text.contains("1 pane"));
+ let accepted = if menu {
+ let primary = state.hits.overlay_primary;
+ click(&mut state, primary)
+ } else {
+ state.handle_input_bytes(b"\r")
+ };
+ assert_tab_close(&accepted);
+ assert!(state.overlay.is_none());
+ }
+}
+
+#[test]
+fn last_tab_close_confirmation_can_be_cancelled() {
+ for mouse in [false, true] {
+ let mut state = close_state(true, 1);
+ assert_no_close(&request_close(&mut state, false));
+ assert!(matches!(
+ state.overlay,
+ Some(ClientShellOverlay::ConfirmClose(_))
+ ));
+ state.compose(106, 24).unwrap();
+ let cancelled = if mouse {
+ let cancel = state.hits.overlay_cancel;
+ click(&mut state, cancel)
+ } else {
+ state.handle_input_bytes(b"\x1b")
+ };
+ assert_no_close(&cancelled);
+ assert!(state.overlay.is_none());
+ }
+}
+
+#[test]
+fn tab_close_stays_immediate_with_confirmation_disabled_or_other_tabs() {
+ for (confirm, tabs) in [(false, 1), (false, 2), (true, 2)] {
+ for menu in [false, true] {
+ let mut state = close_state(confirm, tabs);
+ assert_tab_close(&request_close(&mut state, menu));
+ assert!(state.overlay.is_none());
+ }
+ }
+}
+
+#[test]
+fn last_tab_confirmation_preserves_target_across_focus_changes_and_new_tabs() {
+ let mut state = close_state(true, 1);
+ let mut projected = state.snapshot.as_deref().unwrap().clone();
+ let mut other_workspace = projected.workspaces[0].clone();
+ other_workspace.workspace_id = "ws_2".into();
+ other_workspace.active_tab_id = "other_tab".into();
+ other_workspace.focused = false;
+ projected.workspaces.push(other_workspace);
+ let mut other_tab = projected.tabs[0].clone();
+ other_tab.workspace_id = "ws_2".into();
+ other_tab.tab_id = "other_tab".into();
+ other_tab.focused = false;
+ projected.tabs.push(other_tab);
+ state.set_snapshot(Box::new(projected.clone()));
+
+ assert_no_close(&request_close(&mut state, false));
+ assert!(matches!(
+ state.overlay,
+ Some(ClientShellOverlay::ConfirmClose(_))
+ ));
+ let mut new_tab = projected.tabs[0].clone();
+ new_tab.tab_id = "new_tab".into();
+ new_tab.focused = false;
+ projected.tabs.push(new_tab);
+ projected.focused_workspace_id = Some("ws_2".into());
+ projected.focused_tab_id = Some("other_tab".into());
+ for workspace in &mut projected.workspaces {
+ workspace.focused = workspace.workspace_id == "ws_2";
+ }
+ for tab in &mut projected.tabs {
+ tab.focused = tab.tab_id == "other_tab";
+ }
+ state.set_snapshot(Box::new(projected));
+ assert_tab_close(&state.handle_input_bytes(b"\r"));
+}
+
+#[test]
+fn last_tab_confirmation_rejects_missing_moved_or_reconnected_targets() {
+ for change in ["missing", "moved", "reconnected"] {
+ let mut state = close_state(true, 1);
+ assert_no_close(&request_close(&mut state, false));
+ assert!(matches!(
+ state.overlay,
+ Some(ClientShellOverlay::ConfirmClose(_))
+ ));
+ let mut projected = state.snapshot.as_deref().unwrap().clone();
+ match change {
+ "missing" => projected.tabs.clear(),
+ "moved" => projected.tabs[0].workspace_id = "different_workspace".into(),
+ "reconnected" => state.endpoints[0].snapshot_generation = Some(2),
+ _ => unreachable!(),
+ }
+ if change != "reconnected" {
+ state.set_snapshot(Box::new(projected));
+ }
+ assert_no_close(&state.handle_input_bytes(b"\r"));
+ assert!(state.overlay.is_none());
+ }
+}
+
+#[test]
+fn last_tab_close_preserves_parent_group_and_linked_workspace_scope() {
+ for linked in [false, true] {
+ let mut state = close_state(true, 1);
+ let mut projected = state.snapshot.as_deref().unwrap().clone();
+ projected.workspaces[0].worktree = Some(ClientShellWorktree {
+ key: "repo".into(),
+ label: "repo".into(),
+ is_linked_worktree: linked,
+ });
+ let mut sibling = projected.workspaces[0].clone();
+ sibling.workspace_id = "ws_2".into();
+ sibling.active_tab_id = "tab_2".into();
+ sibling.focused = false;
+ sibling.worktree.as_mut().unwrap().is_linked_worktree = !linked;
+ projected.workspaces.push(sibling);
+ let mut tab = projected.tabs[0].clone();
+ tab.tab_id = "tab_2".into();
+ tab.workspace_id = "ws_2".into();
+ tab.focused = false;
+ projected.tabs.push(tab);
+ state.set_snapshot(Box::new(projected));
+
+ let close = request_close(&mut state, false);
+ if linked {
+ assert_no_close(&close);
+ assert!(matches!(
+ state.overlay,
+ Some(ClientShellOverlay::ConfirmClose(_))
+ ));
+ assert_tab_close(&state.handle_input_bytes(b"\r"));
+ } else {
+ assert_tab_close(&close);
+ assert!(state.overlay.is_none());
+ let [ClientShellAction::Endpoint { request, .. }] = close.actions.as_slice() else {
+ panic!("tab close request");
+ };
+ state.handle_endpoint_result(
+ "boot-1",
+ &request.id,
+ Err(ClientShellEndpointError {
+ code: Some("confirmation_required".into()),
+ message: "closing this tab would close a worktree group".into(),
+ }),
+ );
+ assert!(matches!(state.overlay.as_ref(),
+ Some(ClientShellOverlay::ConfirmClose(confirm)) if confirm.title == "Close worktree group?"));
+ let accepted = state.handle_input_bytes(b"\r");
+ assert!(matches!(accepted.actions.as_slice(),
+ [ClientShellAction::Endpoint { request, .. }]
+ if matches!(&request.method, Method::WorkspaceClose(params)
+ if params.workspace_id == "ws_1" && params.close_group)));
+ }
+ }
+}
diff --git a/src/client/shell/tests/graphics.rs b/src/client/shell/tests/graphics.rs
index 98fc62fa..240e031c 100644
--- a/src/client/shell/tests/graphics.rs
+++ b/src/client/shell/tests/graphics.rs
@@ -216,6 +216,7 @@ fn every_dialog_and_menu_occludes_its_panel_not_the_whole_screen() {
}),
ClientShellOverlay::ConfirmClose(ClientConfirmCloseOverlay {
workspace_id: "ws_1".into(),
+ tab_target: None,
title: "close".into(),
detail: "confirm".into(),
}),
diff --git a/src/client/shell/tests/mod.rs b/src/client/shell/tests/mod.rs
index 787d8a75..9f04b527 100644
--- a/src/client/shell/tests/mod.rs
+++ b/src/client/shell/tests/mod.rs
@@ -246,6 +246,7 @@ fn surface_with_popup() -> PaneSurfaceFrame {
mod agents_worktrees_notifications;
mod chrome_context;
+mod close_tab;
mod copy;
mod endpoint_requests;
mod endpoints;
generated by cgit v1.2.3 (git 2.39.1) at 2026-11-06 22:44:13 +0000