cadcraft/crates/ui-egui/tests/status_bar_toggles.rs

//! The drafting toggles on the status bar are transparent, as F7-F12 are: clicking Grid or Snap
//! while drawing a line changes the setting and leaves LINE running (#1108).

use cadcraft_engine::doc::{Drawing, EntityKind, Space};
use cadcraft_engine::geom::Vec2;
use cadcraft_engine::{Input, Session};
use cadcraft_ui_egui::{CadApp, Services};
use egui::vec2;
use egui_kittest::Harness;
use egui_kittest::kittest::Queryable;

fn harness() -> Harness<'static, CadApp> {
    let mut session = Session::empty();
    session.open_drawing(Drawing::new_imperial(), "Status bar toggles", None);
    let app = CadApp::new(session, Services::default());
    let mut h = Harness::builder().with_size(vec2(1600.0, 1000.0)).build_ui_state(
        |ui, app: &mut CadApp| {
            app.logic(ui.ctx());
            app.ui(ui);
        },
        app,
    );
    h.run_steps(3);
    h
}

fn running(h: &Harness<'_, CadApp>) -> Option<String> {
    h.state().session.running.as_ref().map(|r| r.id.clone())
}

#[test]
fn clicking_grid_and_snap_on_the_status_bar_keeps_line_running() {
    let mut h = harness();
    let (grid, snap) = (h.state().session.settings.gridmode, h.state().session.settings.snapmode);
    {
        let s = &mut h.state_mut().session;
        s.start("line").unwrap();
        s.input(Input::Point(Vec2::new(0.0, 0.0))).unwrap();
    }
    h.run_steps(2);
    h.get_by_label("Grid (F7)").click();
    h.run_steps(2);
    h.get_by_label("Snap Mode (F9)").click();
    h.run_steps(2);
    assert_eq!(h.state().session.settings.gridmode, !grid);
    assert_eq!(h.state().session.settings.snapmode, !snap);
    assert_eq!(running(&h).as_deref(), Some("line"));
    h.state_mut().session.input(Input::Point(Vec2::new(10.0, 0.0))).unwrap();
    let lines: Vec<_> = h
        .state()
        .session
        .state()
        .unwrap()
        .doc
        .space(&Space::Model)
        .unwrap()
        .iter()
        .filter_map(|e| match &e.kind {
            EntityKind::Line(l) => Some((l.a.xy(), l.b.xy())),
            _ => None,
        })
        .collect();
    assert_eq!(lines, vec![(Vec2::new(0.0, 0.0), Vec2::new(10.0, 0.0))]);
}