use warpui::integration::{AssertionCallback, AssertionWithDataCallback};use warpui::{AppContext, SingletonEntity, async_assert, async_assert_eq}; use crate::ai::facts::CloudAIFactModel;use crate::ai::facts::view::AIFactPage;use crate::cloud_object::model::generic_string_model::GenericStringObjectId;use crate::cloud_object::model::persistence::CloudModel;use crate::integration_testing::view_getters::workspace_view;use crate::server::ids::SyncId; /// Assert that a specific AI fact exists with the given contentpub fn assert_rule_exists(    expected_id_key: impl Into<String>,    expected_content: impl Into<String>,) -> AssertionWithDataCallback {    let expected_id_key = expected_id_key.into();    let expected_content = expected_content.into();    Box::new(move |app, _window_id, data| {        let sync_id: &SyncId = data.get(&expected_id_key).expect("No saved AI fact ID");        CloudModel::handle(app).read(app, |cloud_model, _| {            if let Some(ai_fact) =                cloud_model.get_object_of_type::<GenericStringObjectId, CloudAIFactModel>(sync_id)            {                let content = match &ai_fact.model().string_model {                    crate::ai::facts::AIFact::Memory(memory) => &memory.content,                };                async_assert_eq!(content, &expected_content, "AI fact content should match")            } else {                async_assert!(false, "AI fact should exist")            }        })    })} /// Assert that the total number of AI facts matches the expected countpub fn assert_rule_count(expected_count: usize) -> AssertionCallback {    Box::new(move |app, _| {        CloudModel::handle(app).read(app, |cloud_model, ctx| {            let count = rule_count(cloud_model, ctx);            async_assert_eq!(count, expected_count, "Rule count should match")        })    })} /// Helper function to count AI facts in the cloud modelpub fn rule_count(cloud_model: &CloudModel, _ctx: &AppContext) -> usize {    cloud_model        .get_all_objects_of_type::<GenericStringObjectId, CloudAIFactModel>()        .count()} pub fn assert_rule_pane_open(key: impl Into<String>) -> AssertionWithDataCallback {    let key = key.into();    Box::new(move |app, window_id, data| {        workspace_view(app, window_id).read(app, |workspace, _ctx| {            let sync_id: &SyncId = data.get(&key).expect("No saved AI fact ID");            workspace.ai_fact_view().read(app, |ai_fact_view, _ctx| {                let current_page = ai_fact_view.current_page();                async_assert_eq!(                    current_page,                    AIFactPage::RuleEditor {                        sync_id: Some(*sync_id)                    },                    "Rule pane should be open"                )            })        })    })}