|
|
|
@@ -0,0 +1,250 @@
|
|
|
|
|
use std::sync::Arc;
|
|
|
|
|
|
|
|
|
|
use chrono::{Duration, Utc};
|
|
|
|
|
use cloud_object_client::MockObjectClient;
|
|
|
|
|
use settings::manager::SettingsManager;
|
|
|
|
|
use warpui::{App, SingletonEntity};
|
|
|
|
|
|
|
|
|
|
use crate::auth::AuthStateProvider;
|
|
|
|
|
use crate::cloud_object::model::persistence::CloudModel;
|
|
|
|
|
use crate::cloud_object::model::view::CloudViewModel;
|
|
|
|
|
use crate::cloud_object::{Owner, Revision, ServerMetadata, ServerNotebook, ServerPermissions};
|
|
|
|
|
use crate::notebooks::manager::NotebookManager;
|
|
|
|
|
use crate::notebooks::CloudNotebookModel;
|
|
|
|
|
use crate::search::ai_context_menu::notebooks::data_source::NotebookDataSource;
|
|
|
|
|
use crate::search::data_source::Query;
|
|
|
|
|
use crate::search::mixer::SyncDataSource;
|
|
|
|
|
use crate::server::cloud_objects::update_manager::UpdateManager;
|
|
|
|
|
use crate::server::ids::{ServerId, SyncId};
|
|
|
|
|
use crate::server::server_api::team::MockTeamClient;
|
|
|
|
|
use crate::server::server_api::workspace::MockWorkspaceClient;
|
|
|
|
|
use crate::server::server_api::ServerApiProvider;
|
|
|
|
|
use crate::server::sync_queue::SyncQueue;
|
|
|
|
|
use crate::settings::AISettings;
|
|
|
|
|
use crate::system::SystemStats;
|
|
|
|
|
use crate::workspaces::team_tester::TeamTesterStatus;
|
|
|
|
|
use crate::workspaces::user_profiles::UserProfiles;
|
|
|
|
|
use crate::workspaces::user_workspaces::UserWorkspaces;
|
|
|
|
|
use crate::NetworkStatus;
|
|
|
|
|
|
|
|
|
|
fn mock_server_notebook_with_revision(id: i64, title: &str, revision: Revision) -> ServerNotebook {
|
|
|
|
|
ServerNotebook::new(
|
|
|
|
|
SyncId::ServerId(id.into()),
|
|
|
|
|
CloudNotebookModel {
|
|
|
|
|
title: title.to_string(),
|
|
|
|
|
data: format!("{title} content"),
|
|
|
|
|
ai_document_id: None,
|
|
|
|
|
conversation_id: None,
|
|
|
|
|
},
|
|
|
|
|
ServerMetadata {
|
|
|
|
|
uid: ServerId::default(),
|
|
|
|
|
revision,
|
|
|
|
|
metadata_last_updated_ts: Utc::now().into(),
|
|
|
|
|
trashed_ts: None,
|
|
|
|
|
folder_id: None,
|
|
|
|
|
is_welcome_object: false,
|
|
|
|
|
creator_uid: None,
|
|
|
|
|
last_editor_uid: None,
|
|
|
|
|
current_editor_uid: None,
|
|
|
|
|
},
|
|
|
|
|
ServerPermissions {
|
|
|
|
|
space: Owner::mock_current_user(),
|
|
|
|
|
guests: Vec::new(),
|
|
|
|
|
anyone_link_sharing: None,
|
|
|
|
|
permissions_last_updated_ts: Utc::now().into(),
|
|
|
|
|
},
|
|
|
|
|
)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
fn initialize_app(app: &mut App) {
|
|
|
|
|
app.add_singleton_model(|_| NetworkStatus::new());
|
|
|
|
|
app.add_singleton_model(|_| SystemStats::new());
|
|
|
|
|
let mock_team_client = Arc::new(MockTeamClient::new());
|
|
|
|
|
let mock_workspace_client = Arc::new(MockWorkspaceClient::new());
|
|
|
|
|
app.add_singleton_model(|ctx| {
|
|
|
|
|
UserWorkspaces::mock(
|
|
|
|
|
mock_team_client.clone(),
|
|
|
|
|
mock_workspace_client.clone(),
|
|
|
|
|
vec![],
|
|
|
|
|
ctx,
|
|
|
|
|
)
|
|
|
|
|
});
|
|
|
|
|
app.add_singleton_model(TeamTesterStatus::new);
|
|
|
|
|
app.add_singleton_model(SyncQueue::mock);
|
|
|
|
|
app.add_singleton_model(CloudModel::mock);
|
|
|
|
|
app.add_singleton_model(|ctx| UpdateManager::new(None, Arc::new(MockObjectClient::new()), ctx));
|
|
|
|
|
app.add_singleton_model(|_| UserProfiles::new(Vec::new()));
|
|
|
|
|
app.add_singleton_model(CloudViewModel::new);
|
|
|
|
|
app.add_singleton_model(NotebookManager::mock);
|
|
|
|
|
app.add_singleton_model(|_| ServerApiProvider::new_for_test());
|
|
|
|
|
app.add_singleton_model(|_| SettingsManager::default());
|
|
|
|
|
app.add_singleton_model(|_| AuthStateProvider::new_for_test());
|
|
|
|
|
app.update(crate::settings::init_and_register_user_preferences);
|
|
|
|
|
app.update(AISettings::register_and_subscribe_to_events);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
|
fn zero_state_scores_reflect_recency() {
|
|
|
|
|
App::test((), |mut app| async move {
|
|
|
|
|
initialize_app(&mut app);
|
|
|
|
|
|
|
|
|
|
let now = Utc::now();
|
|
|
|
|
CloudModel::handle(&app).update(&mut app, |model, ctx| {
|
|
|
|
|
model.upsert_from_server_notebook(
|
|
|
|
|
mock_server_notebook_with_revision(
|
|
|
|
|
1,
|
|
|
|
|
"oldest",
|
|
|
|
|
(now - Duration::minutes(3)).into(),
|
|
|
|
|
),
|
|
|
|
|
ctx,
|
|
|
|
|
);
|
|
|
|
|
model.upsert_from_server_notebook(
|
|
|
|
|
mock_server_notebook_with_revision(
|
|
|
|
|
2,
|
|
|
|
|
"middle",
|
|
|
|
|
(now - Duration::minutes(2)).into(),
|
|
|
|
|
),
|
|
|
|
|
ctx,
|
|
|
|
|
);
|
|
|
|
|
model.upsert_from_server_notebook(
|
|
|
|
|
mock_server_notebook_with_revision(
|
|
|
|
|
3,
|
|
|
|
|
"newest",
|
|
|
|
|
(now - Duration::minutes(1)).into(),
|
|
|
|
|
),
|
|
|
|
|
ctx,
|
|
|
|
|
);
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
let data_source = NotebookDataSource::new(false);
|
|
|
|
|
let results = app.read(|app| data_source.run_query(&Query::from(""), app).unwrap());
|
|
|
|
|
|
|
|
|
|
assert_eq!(results.len(), 3);
|
|
|
|
|
// run_query sorts descending by score, so first result should be newest
|
|
|
|
|
let scores: Vec<_> = results.iter().map(|r| r.score()).collect();
|
|
|
|
|
assert!(
|
|
|
|
|
scores[0] > scores[1] && scores[1] > scores[2],
|
|
|
|
|
"Expected scores in strictly descending order (newest first), got {scores:?}"
|
|
|
|
|
);
|
|
|
|
|
})
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
|
fn filtered_state_adds_recency_bonus_to_equal_matches() {
|
|
|
|
|
App::test((), |mut app| async move {
|
|
|
|
|
initialize_app(&mut app);
|
|
|
|
|
|
|
|
|
|
let now = Utc::now();
|
|
|
|
|
// All titles contain "plan" so fuzzy scores should be similar
|
|
|
|
|
CloudModel::handle(&app).update(&mut app, |model, ctx| {
|
|
|
|
|
model.upsert_from_server_notebook(
|
|
|
|
|
mock_server_notebook_with_revision(
|
|
|
|
|
1,
|
|
|
|
|
"my first plan",
|
|
|
|
|
(now - Duration::minutes(3)).into(),
|
|
|
|
|
),
|
|
|
|
|
ctx,
|
|
|
|
|
);
|
|
|
|
|
model.upsert_from_server_notebook(
|
|
|
|
|
mock_server_notebook_with_revision(
|
|
|
|
|
2,
|
|
|
|
|
"my second plan",
|
|
|
|
|
(now - Duration::minutes(2)).into(),
|
|
|
|
|
),
|
|
|
|
|
ctx,
|
|
|
|
|
);
|
|
|
|
|
model.upsert_from_server_notebook(
|
|
|
|
|
mock_server_notebook_with_revision(
|
|
|
|
|
3,
|
|
|
|
|
"my third plan",
|
|
|
|
|
(now - Duration::minutes(1)).into(),
|
|
|
|
|
),
|
|
|
|
|
ctx,
|
|
|
|
|
);
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
let data_source = NotebookDataSource::new(false);
|
|
|
|
|
let results = app.read(|app| data_source.run_query(&Query::from("plan"), app).unwrap());
|
|
|
|
|
|
|
|
|
|
assert_eq!(results.len(), 3);
|
|
|
|
|
// All match "plan" similarly; recency bonus should make newer items score higher
|
|
|
|
|
let scores: Vec<_> = results.iter().map(|r| r.score()).collect();
|
|
|
|
|
assert!(
|
|
|
|
|
scores[0] > scores[1] && scores[1] > scores[2],
|
|
|
|
|
"Expected scores in strictly descending order (newest first), got {scores:?}"
|
|
|
|
|
);
|
|
|
|
|
})
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
|
fn test_multibyte_character_truncation() {
|
|
|
|
|
// Test string with multibyte characters (emojis, accented chars)
|
|
|
|
|
let test_content = "This is a test with emojis 🚀 and accented chars like café and naïve that should be truncated properly without panicking. This string is intentionally long to test the 200 character limit and ensure we don't slice in the middle of multibyte characters like 你好世界";
|
|
|
|
|
|
|
|
|
|
let truncated = if test_content.len() > 200 {
|
|
|
|
|
let result = test_content
|
|
|
|
|
.char_indices()
|
|
|
|
|
.take_while(|(i, _)| *i <= 197)
|
|
|
|
|
.last()
|
|
|
|
|
.map(|(i, c)| &test_content[..i + c.len_utf8()])
|
|
|
|
|
.unwrap_or("");
|
|
|
|
|
format!("{result}...")
|
|
|
|
|
} else {
|
|
|
|
|
test_content.to_string()
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
// Should not panic and should produce a valid string
|
|
|
|
|
assert!(!truncated.is_empty());
|
|
|
|
|
assert!(truncated.ends_with("..."));
|
|
|
|
|
// The truncated string should be valid UTF-8
|
|
|
|
|
assert!(std::str::from_utf8(truncated.as_bytes()).is_ok());
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
|
fn test_truncation_with_boundary_at_multibyte_char() {
|
|
|
|
|
// Create a string where byte 197 falls exactly in the middle of a multibyte character
|
|
|
|
|
let mut test_content = "a".repeat(195); // 195 single-byte chars
|
|
|
|
|
test_content.push('🚀'); // 4-byte emoji at positions 195-198
|
|
|
|
|
test_content.push_str("more text after emoji");
|
|
|
|
|
|
|
|
|
|
// This should not panic even though byte 197 is in the middle of the emoji
|
|
|
|
|
let truncated = if test_content.len() > 200 {
|
|
|
|
|
let result = test_content
|
|
|
|
|
.char_indices()
|
|
|
|
|
.take_while(|(i, _)| *i <= 197)
|
|
|
|
|
.last()
|
|
|
|
|
.map(|(i, c)| &test_content[..i + c.len_utf8()])
|
|
|
|
|
.unwrap_or("");
|
|
|
|
|
format!("{result}...")
|
|
|
|
|
} else {
|
|
|
|
|
test_content.to_string()
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
// Should not panic and should produce a valid string
|
|
|
|
|
assert!(!truncated.is_empty());
|
|
|
|
|
// The truncated string should be valid UTF-8
|
|
|
|
|
assert!(std::str::from_utf8(truncated.as_bytes()).is_ok());
|
|
|
|
|
// Should either include the full emoji or stop before it
|
|
|
|
|
assert!(!truncated.contains("🚀") || truncated.contains("🚀..."));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
|
fn test_short_content_not_truncated() {
|
|
|
|
|
let short_content = "This is a short string with emoji 🚀";
|
|
|
|
|
|
|
|
|
|
let result = if short_content.len() > 200 {
|
|
|
|
|
let truncated = short_content
|
|
|
|
|
.char_indices()
|
|
|
|
|
.take_while(|(i, _)| *i <= 197)
|
|
|
|
|
.last()
|
|
|
|
|
.map(|(i, c)| &short_content[..i + c.len_utf8()])
|
|
|
|
|
.unwrap_or("");
|
|
|
|
|
format!("{truncated}...")
|
|
|
|
|
} else {
|
|
|
|
|
short_content.to_string()
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
// Short content should not be truncated
|
|
|
|
|
assert_eq!(result, short_content);
|
|
|
|
|
assert!(!result.ends_with("..."));
|
|
|
|
|
}
|
Ryan Ward