first pass of merging in warp (doesn't build)

2026-07-01 16:08:58 -05:00
parent 2f64909469
commit 4770ac06b5
3662 changed files with 414574 additions and 89772 deletions
@@ -0,0 +1,81 @@
+use warpui::integration::{AssertionCallback, TestStep};
+use warpui::{async_assert, App, SingletonEntity, TypedActionView, WindowId};
+
+use crate::ai::blocklist::history_model::BlocklistAIHistoryModel;
+use crate::ai::document::ai_document_model::{AIDocumentModel, AIDocumentVersion};
+use crate::integration_testing::view_getters::{
+    pane_group_view, single_terminal_view_for_tab, workspace_view,
+};
+use crate::workspace::WorkspaceAction;
+
+pub fn create_and_open_ai_document(title: &'static str, markdown: &'static str) -> TestStep {
+    TestStep::new("Create and open AI document").with_action(move |app, window_id, _| {
+        let terminal_view_id = single_terminal_view_for_tab(app, window_id, 0).id();
+        let document_id = app.update(|ctx| {
+            let conversation_id = BlocklistAIHistoryModel::handle(ctx).update(ctx, |model, ctx| {
+                let conversation_id =
+                    model.start_new_conversation(terminal_view_id, false, false, false, ctx);
+                model.set_active_conversation_id(conversation_id, terminal_view_id, ctx);
+                conversation_id
+            });
+
+            AIDocumentModel::handle(ctx).update(ctx, |model, ctx| {
+                model.create_document(title, markdown, conversation_id, None, ctx)
+            })
+        });
+
+        let workspace = workspace_view(app, window_id);
+        workspace.update(app, |workspace, ctx| {
+            workspace.handle_action(
+                &WorkspaceAction::OpenAIDocumentPane {
+                    document_id,
+                    document_version: AIDocumentVersion::default(),
+                },
+                ctx,
+            );
+        });
+
+        let pane_group = pane_group_view(app, window_id, 0);
+        pane_group.update(app, |pane_group, ctx| {
+            let pane_id = pane_group
+                .ai_document_panes()
+                .next()
+                .expect("AI document pane should be open");
+            let pane_configuration = pane_group
+                .pane_by_id(pane_id)
+                .expect("AI document pane should exist")
+                .pane_configuration();
+            pane_configuration.update(ctx, |pane_configuration, ctx| {
+                pane_configuration.refresh_pane_header_overflow_menu_items(ctx);
+            });
+        });
+    })
+}
+
+pub fn ai_document_overflow_button_position_id(app: &mut App, window_id: WindowId) -> String {
+    let pane_group = pane_group_view(app, window_id, 0);
+    pane_group.read(app, |pane_group, _| {
+        let pane_id = pane_group
+            .ai_document_panes()
+            .next()
+            .expect("AI document pane should be open");
+        let pane_configuration_id = pane_group
+            .pane_by_id(pane_id)
+            .expect("AI document pane should exist")
+            .pane_configuration()
+            .id();
+        format!("pane_header_overflow_button:{pane_configuration_id}")
+    })
+}
+
+pub fn assert_ai_document_overflow_button_position_exists() -> AssertionCallback {
+    Box::new(|app, window_id| {
+        let position_id = ai_document_overflow_button_position_id(app, window_id);
+        let presenter = app.presenter(window_id).expect("presenter should exist");
+        let presenter = presenter.borrow();
+        async_assert!(presenter
+            .position_cache()
+            .get_position(position_id)
+            .is_some())
+    })
+}