mirror of
https://github.com/AppFlowy-IO/AppFlowy-Cloud.git
synced 2025-04-24 22:07:07 -04:00
102 lines
3.9 KiB
Rust
102 lines
3.9 KiB
Rust
use crate::collab::util::empty_document_editor;
|
|
use client_api_test::TestClient;
|
|
use collab_entity::CollabType;
|
|
use database_entity::dto::CreateCollabParams;
|
|
use uuid::Uuid;
|
|
|
|
#[tokio::test]
|
|
async fn query_collab_embedding_after_create_test() {
|
|
let object_id = Uuid::new_v4();
|
|
let mut editor = empty_document_editor(&object_id);
|
|
let contents = vec![
|
|
"AppFlowy is an open-source project.",
|
|
"It is an alternative to tools like Notion.",
|
|
"AppFlowy provides full control of your data.",
|
|
"The project is built using Flutter for the frontend.",
|
|
"Rust powers AppFlowy's backend for safety and performance.",
|
|
"AppFlowy supports both personal and collaborative workflows.",
|
|
"It is customizable and self-hostable.",
|
|
"Users can create documents, databases, and workflows with AppFlowy.",
|
|
"The community contributes actively to AppFlowy's development.",
|
|
"AppFlowy aims to be fast, reliable, and feature-rich.",
|
|
];
|
|
editor.insert_paragraphs(contents.into_iter().map(|s| s.to_string()).collect());
|
|
|
|
let test_client = TestClient::new_user().await;
|
|
let workspace_id = test_client.workspace_id().await;
|
|
let params = CreateCollabParams {
|
|
workspace_id,
|
|
object_id,
|
|
encoded_collab_v1: editor.encode_collab().encode_to_bytes().unwrap(),
|
|
collab_type: CollabType::Document,
|
|
};
|
|
test_client.api_client.create_collab(params).await.unwrap();
|
|
test_client
|
|
.wait_until_get_embedding(&workspace_id, &object_id)
|
|
.await;
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn document_full_sync_then_search_test() {
|
|
let object_id = Uuid::new_v4();
|
|
let mut local_document = empty_document_editor(&object_id);
|
|
let test_client = TestClient::new_user().await;
|
|
let workspace_id = test_client.workspace_id().await;
|
|
let doc_state = local_document.encode_collab().encode_to_bytes().unwrap();
|
|
let params = CreateCollabParams {
|
|
workspace_id,
|
|
object_id,
|
|
encoded_collab_v1: doc_state,
|
|
collab_type: CollabType::Document,
|
|
};
|
|
test_client.api_client.create_collab(params).await.unwrap();
|
|
test_client
|
|
.insert_view_to_general_space(
|
|
&workspace_id,
|
|
&object_id.to_string(),
|
|
"AppFlowy",
|
|
collab_folder::ViewLayout::Document,
|
|
)
|
|
.await;
|
|
|
|
let contents = vec![
|
|
"AppFlowy is an open-source project.",
|
|
"It is an alternative to tools like Notion.",
|
|
"AppFlowy provides full control of your data.",
|
|
"The project is built using Flutter for the frontend.",
|
|
"Rust powers AppFlowy's backend for safety and performance.",
|
|
"AppFlowy supports both personal and collaborative workflows.",
|
|
"It is customizable and self-hostable.",
|
|
"Users can create documents, databases, and workflows with AppFlowy.",
|
|
"The community contributes actively to AppFlowy's development.",
|
|
"AppFlowy aims to be fast, reliable, and feature-rich.",
|
|
];
|
|
local_document.insert_paragraphs(contents.into_iter().map(|s| s.to_string()).collect());
|
|
let encode_collab = local_document.encode_collab();
|
|
|
|
// After full sync, two document should be the same
|
|
test_client
|
|
.api_client
|
|
.collab_full_sync(
|
|
&workspace_id,
|
|
&object_id,
|
|
CollabType::Document,
|
|
encode_collab.doc_state.to_vec(),
|
|
encode_collab.state_vector.to_vec(),
|
|
)
|
|
.await
|
|
.unwrap();
|
|
|
|
let remote_document = test_client
|
|
.create_document_collab(workspace_id, object_id)
|
|
.await;
|
|
let remote_plain_text = remote_document.to_plain_text(false, false).unwrap();
|
|
let local_plain_text = local_document.document.to_plain_text(false, false).unwrap();
|
|
assert_eq!(local_plain_text, remote_plain_text);
|
|
|
|
let search_result = test_client
|
|
.wait_unit_get_search_result(&workspace_id, "workflows", 1)
|
|
.await;
|
|
assert_eq!(search_result.len(), 1);
|
|
assert_eq!(search_result[0].preview, Some("AppFlowy is an open-source project. It is an alternative to tools like Notion. AppFlowy provides full control of your data. The project is built using Flutter for the frontend. Rust powers AppFlowy's ".to_string()));
|
|
}
|