Files
2026-11-01 15:56:42 +07:00

130 lines
4.5 KiB
Rust

use std::{cell::RefCell, rc::Rc};
use gpui::{
Context, IntoElement, MouseButton, ParentElement as _, Render, Styled as _, TestAppContext,
VisualTestContext, Window, div, point, px,
};
use gpui_native_core::context_menu::ContextMenu;
use gpui_native_core::menu::MenuItem;
use gpui_native_theme::Theme;
type Actions = Rc<RefCell<Vec<&'static str>>>;
struct Harness {
actions: Actions,
}
impl Render for Harness {
fn render(&mut self, _window: &mut Window, _cx: &mut Context<Self>) -> impl IntoElement {
let actions = self.actions.clone();
div().size_full().p_4().child(
ContextMenu::new("ctx")
.target(|_, _| div().h(px(40.)).w(px(200.)).child("Row").into_any_element())
.items({
let actions = actions.clone();
move |_, _| {
vec![
MenuItem::new("Rename").handler({
let actions = actions.clone();
move |_, _| actions.borrow_mut().push("rename")
}),
MenuItem::new("Delete").handler({
let actions = actions.clone();
move |_, _| actions.borrow_mut().push("delete")
}),
]
}
}),
)
}
}
/// The target row rides the `p_4` band: `16..216` horizontally, `16..56`
/// vertically, so right-clicks land near `(100, 30)`.
fn context_window() -> (TestAppContext, VisualTestContext, Actions) {
let mut cx = TestAppContext::single();
cx.update(Theme::init);
let actions = Rc::new(RefCell::new(Vec::new()));
let harness_actions = actions.clone();
let window = cx.add_window(move |_, _| Harness {
actions: harness_actions,
});
let mut visual = VisualTestContext::from_window(window.into(), &cx);
visual.update(|window, cx| {
window.draw(cx).clear(cx);
});
(cx, visual, actions)
}
fn right_click(visual: &mut VisualTestContext, position: gpui::Point<gpui::Pixels>) {
visual.simulate_mouse_down(position, MouseButton::Right, gpui::Modifiers::default());
visual.simulate_mouse_up(position, MouseButton::Right, gpui::Modifiers::default());
}
fn click(visual: &mut VisualTestContext, position: gpui::Point<gpui::Pixels>) {
visual.simulate_mouse_down(position, MouseButton::Left, gpui::Modifiers::default());
visual.simulate_mouse_up(position, MouseButton::Left, gpui::Modifiers::default());
}
fn draw(visual: &mut VisualTestContext) {
visual.update(|window, cx| {
window.draw(cx).clear(cx);
});
}
#[test]
fn right_click_opens_the_menu_and_items_activate() {
let (_cx, mut visual, actions) = context_window();
right_click(&mut visual, point(px(100.), px(30.)));
draw(&mut visual);
// The panel claims focus, so the first item is keyboard-highlighted.
visual.simulate_keystrokes("enter");
draw(&mut visual);
assert_eq!(*actions.borrow(), vec!["rename"]);
// The menu closed with the choice, so Enter reports nothing more.
visual.simulate_keystrokes("enter");
draw(&mut visual);
assert_eq!(*actions.borrow(), vec!["rename"]);
// Right-click again reopens; Down reaches the second item.
right_click(&mut visual, point(px(100.), px(30.)));
draw(&mut visual);
visual.simulate_keystrokes("down");
visual.simulate_keystrokes("enter");
draw(&mut visual);
assert_eq!(*actions.borrow(), vec!["rename", "delete"]);
}
#[test]
fn a_left_click_on_the_target_does_not_open_the_menu() {
let (_cx, mut visual, actions) = context_window();
click(&mut visual, point(px(100.), px(30.)));
draw(&mut visual);
visual.simulate_keystrokes("enter");
draw(&mut visual);
assert_eq!(*actions.borrow(), Vec::<&str>::new());
}
#[test]
fn a_click_outside_closes_the_menu() {
let (_cx, mut visual, actions) = context_window();
right_click(&mut visual, point(px(100.), px(30.)));
draw(&mut visual);
click(&mut visual, point(px(450.), px(300.)));
draw(&mut visual);
visual.simulate_keystrokes("enter");
draw(&mut visual);
assert_eq!(*actions.borrow(), Vec::<&str>::new());
// A right-click outside the target never opens it either.
right_click(&mut visual, point(px(450.), px(300.)));
draw(&mut visual);
visual.simulate_keystrokes("enter");
draw(&mut visual);
assert_eq!(*actions.borrow(), Vec::<&str>::new());
}