AppFlowy-Cloud/tests/workspace/page_view.rs

812 lines
23 KiB
Rust

use std::{collections::HashSet, time::Duration};
use client_api::entity::{QueryCollab, QueryCollabParams};
use client_api_test::{
generate_unique_registered_user, generate_unique_registered_user_client, TestClient,
};
use collab::core::origin::CollabClient;
use collab_entity::CollabType;
use collab_folder::{CollabOrigin, Folder};
use serde_json::{json, Value};
use shared_entity::dto::workspace_dto::{
CreatePageParams, CreateSpaceParams, IconType, MovePageParams, PublishPageParams,
SpacePermission, UpdatePageParams, UpdateSpaceParams, ViewIcon, ViewLayout,
};
use tokio::time::sleep;
use uuid::Uuid;
async fn get_latest_folder(test_client: &TestClient, workspace_id: &str) -> Folder {
// Wait for websocket updates
sleep(Duration::from_secs(1)).await;
let lock = test_client
.collabs
.get(workspace_id)
.unwrap()
.collab
.read()
.await;
let collab_type = CollabType::Folder;
let encoded_collab = lock
.encode_collab_v1(|collab| collab_type.validate_require_data(collab))
.unwrap();
let uid = test_client.uid().await;
Folder::from_collab_doc_state(
uid,
CollabOrigin::Client(CollabClient::new(uid, test_client.device_id.clone())),
encoded_collab.into(),
workspace_id,
vec![],
)
.unwrap()
}
#[tokio::test]
async fn get_page_view() {
let (c, _user) = generate_unique_registered_user_client().await;
let workspaces = c.get_workspaces().await.unwrap();
assert_eq!(workspaces.len(), 1);
let workspace_id = workspaces[0].workspace_id;
let folder_view = c
.get_workspace_folder(&workspace_id.to_string(), Some(2), None)
.await
.unwrap();
let general_space = &folder_view
.children
.into_iter()
.find(|v| v.name == "General")
.unwrap();
let todo = general_space
.children
.iter()
.find(|v| v.name == "To-dos")
.unwrap();
let todo_list_view_id = &todo.view_id;
let resp = c
.get_workspace_page_view(workspace_id, todo_list_view_id)
.await
.unwrap();
assert_eq!(resp.data.row_data.len(), 5);
let getting_started = general_space
.children
.iter()
.find(|v| v.name == "Getting started")
.unwrap();
let getting_started_view_id = &getting_started.view_id;
let resp = c
.get_workspace_page_view(workspace_id, getting_started_view_id)
.await
.unwrap();
assert_eq!(resp.data.row_data.len(), 0);
}
#[tokio::test]
async fn create_new_page_with_database() {
let (c, _user) = generate_unique_registered_user_client().await;
let workspaces = c.get_workspaces().await.unwrap();
assert_eq!(workspaces.len(), 1);
let workspace_id = workspaces[0].workspace_id;
let folder_view = c
.get_workspace_folder(&workspace_id.to_string(), Some(2), None)
.await
.unwrap();
let general_space = &folder_view
.children
.into_iter()
.find(|v| v.name == "General")
.unwrap();
let calendar_page = c
.create_workspace_page_view(
workspace_id,
&CreatePageParams {
parent_view_id: general_space.view_id.clone(),
layout: ViewLayout::Calendar,
name: Some("New calendar".to_string()),
},
)
.await
.unwrap();
let grid_page = c
.create_workspace_page_view(
workspace_id,
&CreatePageParams {
parent_view_id: general_space.view_id.clone(),
layout: ViewLayout::Grid,
name: Some("New grid".to_string()),
},
)
.await
.unwrap();
let board_page = c
.create_workspace_page_view(
workspace_id,
&CreatePageParams {
parent_view_id: general_space.view_id.clone(),
layout: ViewLayout::Grid,
name: Some("New board".to_string()),
},
)
.await
.unwrap();
sleep(Duration::from_secs(1)).await;
let folder_view = c
.get_workspace_folder(&workspace_id.to_string(), Some(2), None)
.await
.unwrap();
let general_space = &folder_view
.children
.into_iter()
.find(|v| v.name == "General")
.unwrap();
let views_under_general_space: HashSet<String> = general_space
.children
.iter()
.map(|v| v.view_id.clone())
.collect();
for view_id in &[
calendar_page.view_id.clone(),
grid_page.view_id.clone(),
board_page.view_id.clone(),
] {
assert!(views_under_general_space.contains(view_id));
c.get_workspace_page_view(workspace_id, view_id)
.await
.unwrap();
}
}
#[tokio::test]
async fn create_new_document_page() {
let (c, _user) = generate_unique_registered_user_client().await;
let workspaces = c.get_workspaces().await.unwrap();
assert_eq!(workspaces.len(), 1);
let workspace_id = workspaces[0].workspace_id;
let folder_view = c
.get_workspace_folder(&workspace_id.to_string(), Some(2), None)
.await
.unwrap();
let general_space = &folder_view
.children
.into_iter()
.find(|v| v.name == "General")
.unwrap();
let page = c
.create_workspace_page_view(
workspace_id,
&CreatePageParams {
parent_view_id: general_space.view_id.clone(),
layout: ViewLayout::Document,
name: Some("New document".to_string()),
},
)
.await
.unwrap();
sleep(Duration::from_secs(1)).await;
let folder_view = c
.get_workspace_folder(&workspace_id.to_string(), Some(2), None)
.await
.unwrap();
let general_space = &folder_view
.children
.into_iter()
.find(|v| v.name == "General")
.unwrap();
let view = general_space
.children
.iter()
.find(|v| v.view_id == page.view_id)
.unwrap();
assert_eq!(view.name, "New document");
c.get_collab(QueryCollabParams {
workspace_id: workspace_id.to_string(),
inner: QueryCollab {
object_id: page.view_id,
collab_type: CollabType::Document,
},
})
.await
.unwrap();
}
#[tokio::test]
async fn move_page_to_another_space() {
let registered_user = generate_unique_registered_user().await;
let mut app_client = TestClient::user_with_new_device(registered_user.clone()).await;
let web_client = TestClient::user_with_new_device(registered_user.clone()).await;
let workspace_id = app_client.workspace_id().await;
let workspace_uuid = Uuid::parse_str(&workspace_id).unwrap();
app_client.open_workspace_collab(&workspace_id).await;
app_client
.wait_object_sync_complete(&workspace_id)
.await
.unwrap();
let folder_view = web_client
.api_client
.get_workspace_folder(&workspace_id, Some(2), None)
.await
.unwrap();
let general_space = folder_view
.children
.iter()
.find(|v| v.name == "General")
.unwrap()
.clone();
let todo_view_id = general_space
.children
.iter()
.find(|v| v.name == "To-dos")
.map(|v| v.view_id.clone())
.unwrap();
let shared_space = &folder_view
.children
.iter()
.find(|v| v.name == "Shared")
.unwrap()
.clone();
web_client
.api_client
.move_workspace_page_view(
workspace_uuid,
&todo_view_id,
&MovePageParams {
new_parent_view_id: shared_space.view_id.clone(),
prev_view_id: None,
},
)
.await
.unwrap();
let folder = get_latest_folder(&app_client, &workspace_id).await;
let first_children_id = folder.get_view(&shared_space.view_id).unwrap().children[0]
.id
.clone();
assert_eq!(first_children_id, todo_view_id);
}
#[tokio::test]
async fn move_page_to_trash_then_restore() {
let registered_user = generate_unique_registered_user().await;
let mut app_client = TestClient::user_with_new_device(registered_user.clone()).await;
let web_client = TestClient::user_with_new_device(registered_user.clone()).await;
let workspace_id = app_client.workspace_id().await;
let folder_view = web_client
.api_client
.get_workspace_folder(&workspace_id, Some(2), None)
.await
.unwrap();
let general_space = &folder_view
.children
.into_iter()
.find(|v| v.name == "General")
.unwrap();
let view_ids_to_be_deleted = [
general_space.children[0].view_id.clone(),
general_space.children[1].view_id.clone(),
];
app_client.open_workspace_collab(&workspace_id).await;
app_client
.wait_object_sync_complete(&workspace_id)
.await
.unwrap();
for view_id in view_ids_to_be_deleted.iter() {
web_client
.api_client
.move_workspace_page_view_to_trash(Uuid::parse_str(&workspace_id).unwrap(), view_id)
.await
.unwrap();
}
let folder = get_latest_folder(&app_client, &workspace_id).await;
let views_in_trash_for_app = folder
.get_my_trash_sections()
.iter()
.map(|v| v.id.clone())
.collect::<HashSet<String>>();
for view_id in view_ids_to_be_deleted.iter() {
assert!(views_in_trash_for_app.contains(view_id));
}
let views_in_trash_for_web = web_client
.api_client
.get_workspace_trash(&workspace_id)
.await
.unwrap()
.views
.iter()
.map(|v| v.view.view_id.clone())
.collect::<HashSet<String>>();
for view_id in view_ids_to_be_deleted.iter() {
assert!(views_in_trash_for_web.contains(view_id));
}
web_client
.api_client
.restore_workspace_page_view_from_trash(
Uuid::parse_str(&workspace_id).unwrap(),
&view_ids_to_be_deleted[0],
)
.await
.unwrap();
let folder = get_latest_folder(&app_client, &workspace_id).await;
assert!(!folder
.get_my_trash_sections()
.iter()
.any(|v| v.id == view_ids_to_be_deleted[0]));
let view_found = web_client
.api_client
.get_workspace_trash(&workspace_id)
.await
.unwrap()
.views
.iter()
.any(|v| v.view.view_id == view_ids_to_be_deleted[0]);
assert!(!view_found);
web_client
.api_client
.restore_all_workspace_page_views_from_trash(Uuid::parse_str(&workspace_id).unwrap())
.await
.unwrap();
let folder = get_latest_folder(&app_client, &workspace_id).await;
assert!(!folder
.get_my_trash_sections()
.iter()
.any(|v| v.id == view_ids_to_be_deleted[1]));
let view_found = web_client
.api_client
.get_workspace_trash(&workspace_id)
.await
.unwrap()
.views
.iter()
.any(|v| v.view.view_id == view_ids_to_be_deleted[1]);
assert!(!view_found);
}
#[tokio::test]
async fn move_page_with_child_to_trash_then_restore() {
let registered_user = generate_unique_registered_user().await;
let mut app_client = TestClient::user_with_new_device(registered_user.clone()).await;
let web_client = TestClient::user_with_new_device(registered_user.clone()).await;
let workspace_id = app_client.workspace_id().await;
let folder_view = web_client
.api_client
.get_workspace_folder(&workspace_id, Some(2), None)
.await
.unwrap();
let general_space = &folder_view
.children
.into_iter()
.find(|v| v.name == "General")
.unwrap();
app_client.open_workspace_collab(&workspace_id).await;
app_client
.wait_object_sync_complete(&workspace_id)
.await
.unwrap();
web_client
.api_client
.move_workspace_page_view_to_trash(
Uuid::parse_str(&workspace_id).unwrap(),
&general_space.view_id,
)
.await
.unwrap();
let folder = get_latest_folder(&app_client, &workspace_id).await;
let views_in_trash_for_app = folder
.get_my_trash_sections()
.iter()
.map(|v| v.id.clone())
.collect::<HashSet<String>>();
assert!(views_in_trash_for_app.contains(&general_space.view_id));
for view in general_space.children.iter() {
assert!(!views_in_trash_for_app.contains(&view.view_id));
}
let views_in_trash_for_web = web_client
.api_client
.get_workspace_trash(&workspace_id)
.await
.unwrap()
.views
.iter()
.map(|v| v.view.view_id.clone())
.collect::<HashSet<String>>();
assert!(views_in_trash_for_web.contains(&general_space.view_id));
web_client
.api_client
.restore_workspace_page_view_from_trash(
Uuid::parse_str(&workspace_id).unwrap(),
&general_space.view_id,
)
.await
.unwrap();
let folder = get_latest_folder(&app_client, &workspace_id).await;
assert!(!folder
.get_my_trash_sections()
.iter()
.any(|v| v.id == general_space.view_id));
let view_found = web_client
.api_client
.get_workspace_trash(&workspace_id)
.await
.unwrap()
.views
.iter()
.any(|v| v.view.view_id == general_space.view_id);
assert!(!view_found);
}
#[tokio::test]
async fn move_page_with_child_to_trash_then_delete_permanently() {
let registered_user = generate_unique_registered_user().await;
let mut app_client = TestClient::user_with_new_device(registered_user.clone()).await;
let web_client = TestClient::user_with_new_device(registered_user.clone()).await;
let workspace_id = app_client.workspace_id().await;
let folder_view = web_client
.api_client
.get_workspace_folder(&workspace_id, Some(2), None)
.await
.unwrap();
let general_space = &folder_view
.children
.into_iter()
.find(|v| v.name == "General")
.unwrap();
app_client.open_workspace_collab(&workspace_id).await;
app_client
.wait_object_sync_complete(&workspace_id)
.await
.unwrap();
web_client
.api_client
.move_workspace_page_view_to_trash(
Uuid::parse_str(&workspace_id).unwrap(),
&general_space.view_id,
)
.await
.unwrap();
let folder = get_latest_folder(&app_client, &workspace_id).await;
let views_in_trash_for_app = folder
.get_my_trash_sections()
.iter()
.map(|v| v.id.clone())
.collect::<HashSet<String>>();
assert!(views_in_trash_for_app.contains(&general_space.view_id));
for view in general_space.children.iter() {
assert!(!views_in_trash_for_app.contains(&view.view_id));
}
let views_in_trash_for_web = web_client
.api_client
.get_workspace_trash(&workspace_id)
.await
.unwrap()
.views
.iter()
.map(|v| v.view.view_id.clone())
.collect::<HashSet<String>>();
assert!(views_in_trash_for_web.contains(&general_space.view_id));
web_client
.api_client
.delete_workspace_page_view_from_trash(
Uuid::parse_str(&workspace_id).unwrap(),
&general_space.view_id,
)
.await
.unwrap();
let folder = get_latest_folder(&app_client, &workspace_id).await;
assert!(folder.get_view(&general_space.view_id).is_none());
assert!(!folder
.get_my_trash_sections()
.iter()
.any(|v| v.id == general_space.view_id));
let view_found = web_client
.api_client
.get_workspace_trash(&workspace_id)
.await
.unwrap()
.views
.iter()
.any(|v| v.view.view_id == general_space.view_id);
assert!(!view_found);
}
#[tokio::test]
async fn move_page_with_child_to_trash_then_delete_all_permanently() {
let registered_user = generate_unique_registered_user().await;
let mut app_client = TestClient::user_with_new_device(registered_user.clone()).await;
let web_client = TestClient::user_with_new_device(registered_user.clone()).await;
let workspace_id = app_client.workspace_id().await;
let folder_view = web_client
.api_client
.get_workspace_folder(&workspace_id, Some(2), None)
.await
.unwrap();
let general_space = &folder_view
.children
.into_iter()
.find(|v| v.name == "General")
.unwrap();
app_client.open_workspace_collab(&workspace_id).await;
app_client
.wait_object_sync_complete(&workspace_id)
.await
.unwrap();
web_client
.api_client
.move_workspace_page_view_to_trash(
Uuid::parse_str(&workspace_id).unwrap(),
&general_space.view_id,
)
.await
.unwrap();
let folder = get_latest_folder(&app_client, &workspace_id).await;
let views_in_trash_for_app = folder
.get_my_trash_sections()
.iter()
.map(|v| v.id.clone())
.collect::<HashSet<String>>();
assert!(views_in_trash_for_app.contains(&general_space.view_id));
for view in general_space.children.iter() {
assert!(!views_in_trash_for_app.contains(&view.view_id));
}
let views_in_trash_for_web = web_client
.api_client
.get_workspace_trash(&workspace_id)
.await
.unwrap()
.views
.iter()
.map(|v| v.view.view_id.clone())
.collect::<HashSet<String>>();
assert!(views_in_trash_for_web.contains(&general_space.view_id));
web_client
.api_client
.delete_all_workspace_page_views_from_trash(Uuid::parse_str(&workspace_id).unwrap())
.await
.unwrap();
let folder = get_latest_folder(&app_client, &workspace_id).await;
assert!(folder.get_view(&general_space.view_id).is_none());
assert!(!folder
.get_my_trash_sections()
.iter()
.any(|v| v.id == general_space.view_id));
let view_found = web_client
.api_client
.get_workspace_trash(&workspace_id)
.await
.unwrap()
.views
.iter()
.any(|v| v.view.view_id == general_space.view_id);
assert!(!view_found);
}
#[tokio::test]
async fn update_page() {
let registered_user = generate_unique_registered_user().await;
let mut app_client = TestClient::user_with_new_device(registered_user.clone()).await;
let web_client = TestClient::user_with_new_device(registered_user.clone()).await;
let workspace_id = app_client.workspace_id().await;
app_client.open_workspace_collab(&workspace_id).await;
app_client
.wait_object_sync_complete(&workspace_id)
.await
.unwrap();
let folder_view = web_client
.api_client
.get_workspace_folder(&workspace_id.to_string(), Some(2), None)
.await
.unwrap();
let general_space = &folder_view
.children
.into_iter()
.find(|v| v.name == "General")
.unwrap();
let view_id_to_be_updated = general_space.children[0].view_id.clone();
web_client
.api_client
.update_workspace_page_view(
Uuid::parse_str(&workspace_id).unwrap(),
&view_id_to_be_updated,
&UpdatePageParams {
name: "New Name".to_string(),
icon: Some(ViewIcon {
ty: IconType::Emoji,
value: "🚀".to_string(),
}),
extra: Some(json!({"is_pinned": true})),
},
)
.await
.unwrap();
let folder = get_latest_folder(&app_client, &workspace_id).await;
let updated_view = folder.get_view(&view_id_to_be_updated).unwrap();
assert_eq!(updated_view.name, "New Name");
assert_eq!(
updated_view.icon,
Some(collab_folder::ViewIcon {
ty: collab_folder::IconType::Emoji,
value: "🚀".to_string(),
})
);
assert_eq!(
updated_view.extra,
Some(json!({"is_pinned": true}).to_string())
);
}
#[tokio::test]
async fn create_space() {
let registered_user = generate_unique_registered_user().await;
let mut app_client = TestClient::user_with_new_device(registered_user.clone()).await;
let web_client = TestClient::user_with_new_device(registered_user.clone()).await;
let workspace_id = app_client.workspace_id().await;
app_client.open_workspace_collab(&workspace_id).await;
app_client
.wait_object_sync_complete(&workspace_id)
.await
.unwrap();
let workspace_uuid = Uuid::parse_str(&workspace_id).unwrap();
let public_space = web_client
.api_client
.create_space(
workspace_uuid,
&CreateSpaceParams {
space_permission: SpacePermission::PublicToAll,
name: "Public Space".to_string(),
space_icon: "space_icon_1".to_string(),
space_icon_color: "0xFFA34AFD".to_string(),
},
)
.await
.unwrap();
web_client
.api_client
.create_space(
workspace_uuid,
&CreateSpaceParams {
space_permission: SpacePermission::Private,
name: "Private Space".to_string(),
space_icon: "space_icon_2".to_string(),
space_icon_color: "0xFFA34AFD".to_string(),
},
)
.await
.unwrap();
let folder = get_latest_folder(&app_client, &workspace_id).await;
let view = folder.get_view(&public_space.view_id).unwrap();
let space_info: Value = serde_json::from_str(view.extra.as_ref().unwrap()).unwrap();
assert!(space_info["is_space"].as_bool().unwrap());
assert_eq!(
space_info["space_permission"].as_u64().unwrap() as u8,
SpacePermission::PublicToAll as u8
);
assert_eq!(space_info["space_icon"].as_str().unwrap(), "space_icon_1");
assert_eq!(
space_info["space_icon_color"].as_str().unwrap(),
"0xFFA34AFD"
);
let folder_view = web_client
.api_client
.get_workspace_folder(&workspace_id, Some(2), Some(workspace_id.to_string()))
.await
.unwrap();
folder_view
.children
.iter()
.find(|v| v.name == "Public Space")
.unwrap();
let private_space = folder_view
.children
.iter()
.find(|v| v.name == "Private Space")
.unwrap();
assert!(private_space.is_private);
web_client
.api_client
.update_space(
workspace_uuid,
&private_space.view_id,
&UpdateSpaceParams {
space_permission: SpacePermission::PublicToAll,
name: "Renamed Space".to_string(),
space_icon: "space_icon_3".to_string(),
space_icon_color: "#000000".to_string(),
},
)
.await
.unwrap();
let folder = get_latest_folder(&app_client, &workspace_id).await;
let view = folder.get_view(&private_space.view_id).unwrap();
let space_info: Value = serde_json::from_str(view.extra.as_ref().unwrap()).unwrap();
assert!(space_info["is_space"].as_bool().unwrap());
assert_eq!(
space_info["space_permission"].as_u64().unwrap() as u8,
SpacePermission::PublicToAll as u8
);
assert_eq!(space_info["space_icon"].as_str().unwrap(), "space_icon_3");
assert_eq!(space_info["space_icon_color"].as_str().unwrap(), "#000000");
}
#[tokio::test]
async fn publish_page() {
let registered_user = generate_unique_registered_user().await;
let web_client = TestClient::user_with_new_device(registered_user.clone()).await;
let workspace_id = web_client.workspace_id().await;
let folder_view = web_client
.api_client
.get_workspace_folder(&workspace_id, Some(2), None)
.await
.unwrap();
let general_space = &folder_view
.children
.into_iter()
.find(|v| v.name == "General")
.unwrap();
let database_page_id = general_space
.children
.iter()
.find(|v| v.name == "To-dos")
.unwrap()
.view_id
.clone();
let document_page_id = general_space
.children
.iter()
.find(|v| v.name == "Getting started")
.unwrap()
.view_id
.clone();
let page_to_be_published = vec![database_page_id, document_page_id];
let workspace_uuid = Uuid::parse_str(&workspace_id).unwrap();
for view_id in &page_to_be_published {
web_client
.api_client
.publish_page(
workspace_uuid,
view_id,
&PublishPageParams {
publish_name: None,
visible_database_view_ids: None,
},
)
.await
.unwrap();
}
let publish_namespace = web_client
.api_client
.get_workspace_publish_namespace(&workspace_id)
.await
.unwrap();
let published_view = web_client
.api_client
.get_published_outline(&publish_namespace)
.await
.unwrap();
let published_view_ids: HashSet<String> = published_view
.children
.iter()
.find(|v| v.name == "General")
.unwrap()
.children
.iter()
.map(|v| v.view_id.clone())
.collect();
for view_id in &page_to_be_published {
assert!(published_view_ids.contains(view_id));
}
for view_id in &page_to_be_published {
web_client
.api_client
.unpublish_page(workspace_uuid, view_id)
.await
.unwrap();
}
let published_view = web_client
.api_client
.get_published_outline(&publish_namespace)
.await
.unwrap();
assert_eq!(published_view.children.len(), 0);
}