use std::convert::TryFrom; use bytes::Bytes; use event_integration_test::event_builder::EventBuilder; use event_integration_test::EventIntegrationTest; use flowy_database2::entities::{ CellChangesetPB, CellIdPB, CheckboxCellDataPB, ChecklistCellDataChangesetPB, DatabaseLayoutPB, DatabaseSettingChangesetPB, DatabaseViewIdPB, DateCellChangesetPB, FieldType, OrderObjectPositionPB, RelationCellChangesetPB, SelectOptionCellDataPB, UpdateRowMetaChangesetPB, }; use lib_infra::util::timestamp; #[tokio::test] async fn get_database_id_event_test() { let test = EventIntegrationTest::new_anon().await; let current_workspace = test.get_current_workspace().await; let grid_view = test .create_grid(¤t_workspace.id, "my grid view".to_owned(), vec![]) .await; // The view id can be used to get the database id. let database_id = EventBuilder::new(test.clone()) .event(flowy_database2::event_map::DatabaseEvent::GetDatabaseId) .payload(DatabaseViewIdPB { value: grid_view.id.clone(), }) .async_send() .await .parse::() .value; assert_ne!(database_id, grid_view.id); } #[tokio::test] async fn get_database_event_test() { let test = EventIntegrationTest::new_anon().await; let current_workspace = test.get_current_workspace().await; let grid_view = test .create_grid(¤t_workspace.id, "my grid view".to_owned(), vec![]) .await; let database = test.get_database(&grid_view.id).await; assert_eq!(database.fields.len(), 3); assert_eq!(database.rows.len(), 3); assert_eq!(database.layout_type, DatabaseLayoutPB::Grid); } #[tokio::test] async fn get_field_event_test() { let test = EventIntegrationTest::new_anon().await; let current_workspace = test.get_current_workspace().await; let grid_view = test .create_grid(¤t_workspace.id, "my grid view".to_owned(), vec![]) .await; let fields = test.get_all_database_fields(&grid_view.id).await.items; assert_eq!(fields[0].field_type, FieldType::RichText); assert_eq!(fields[1].field_type, FieldType::SingleSelect); assert_eq!(fields[2].field_type, FieldType::Checkbox); assert_eq!(fields.len(), 3); } #[tokio::test] async fn create_field_event_test() { let test = EventIntegrationTest::new_anon().await; let current_workspace = test.get_current_workspace().await; let grid_view = test .create_grid(¤t_workspace.id, "my grid view".to_owned(), vec![]) .await; test.create_field(&grid_view.id, FieldType::Checkbox).await; let fields = test.get_all_database_fields(&grid_view.id).await.items; assert_eq!(fields.len(), 4); assert_eq!(fields[3].field_type, FieldType::Checkbox); } #[tokio::test] async fn delete_field_event_test() { let test = EventIntegrationTest::new_anon().await; let current_workspace = test.get_current_workspace().await; let grid_view = test .create_grid(¤t_workspace.id, "my grid view".to_owned(), vec![]) .await; let fields = test.get_all_database_fields(&grid_view.id).await.items; assert_eq!(fields[0].field_type, FieldType::RichText); assert_eq!(fields[1].field_type, FieldType::SingleSelect); assert_eq!(fields[2].field_type, FieldType::Checkbox); let error = test.delete_field(&grid_view.id, &fields[1].id).await; assert!(error.is_none()); let fields = test.get_all_database_fields(&grid_view.id).await.items; assert_eq!(fields.len(), 2); } // The primary field is not allowed to be deleted. #[tokio::test] async fn delete_primary_field_event_test() { let test = EventIntegrationTest::new_anon().await; let current_workspace = test.get_current_workspace().await; let grid_view = test .create_grid(¤t_workspace.id, "my grid view".to_owned(), vec![]) .await; let fields = test.get_all_database_fields(&grid_view.id).await.items; // the primary field is not allowed to be deleted. assert!(fields[0].is_primary); let error = test.delete_field(&grid_view.id, &fields[0].id).await; assert!(error.is_some()); } #[tokio::test] async fn update_field_type_event_test() { let test = EventIntegrationTest::new_anon().await; let current_workspace = test.get_current_workspace().await; let grid_view = test .create_grid(¤t_workspace.id, "my grid view".to_owned(), vec![]) .await; let fields = test.get_all_database_fields(&grid_view.id).await.items; let error = test .update_field_type(&grid_view.id, &fields[1].id, FieldType::Checklist) .await; assert!(error.is_none()); let fields = test.get_all_database_fields(&grid_view.id).await.items; assert_eq!(fields[1].field_type, FieldType::Checklist); } #[tokio::test] async fn update_primary_field_type_event_test() { let test = EventIntegrationTest::new_anon().await; let current_workspace = test.get_current_workspace().await; let grid_view = test .create_grid(¤t_workspace.id, "my grid view".to_owned(), vec![]) .await; let fields = test.get_all_database_fields(&grid_view.id).await.items; // the primary field is not allowed to be deleted. assert!(fields[0].is_primary); // the primary field is not allowed to be updated. let error = test .update_field_type(&grid_view.id, &fields[0].id, FieldType::Checklist) .await; assert!(error.is_some()); } #[tokio::test] async fn duplicate_field_event_test() { let test = EventIntegrationTest::new_anon().await; let current_workspace = test.get_current_workspace().await; let grid_view = test .create_grid(¤t_workspace.id, "my grid view".to_owned(), vec![]) .await; let fields = test.get_all_database_fields(&grid_view.id).await.items; // the primary field is not allowed to be updated. let error = test.duplicate_field(&grid_view.id, &fields[1].id).await; assert!(error.is_none()); let fields = test.get_all_database_fields(&grid_view.id).await.items; assert_eq!(fields.len(), 4); } // The primary field is not allowed to be duplicated. So this test should return an error. #[tokio::test] async fn duplicate_primary_field_test() { let test = EventIntegrationTest::new_anon().await; let current_workspace = test.get_current_workspace().await; let grid_view = test .create_grid(¤t_workspace.id, "my grid view".to_owned(), vec![]) .await; let fields = test.get_all_database_fields(&grid_view.id).await.items; // the primary field is not allowed to be duplicated. let error = test.duplicate_field(&grid_view.id, &fields[0].id).await; assert!(error.is_some()); } #[tokio::test] async fn get_primary_field_event_test() { let test = EventIntegrationTest::new_anon().await; let current_workspace = test.get_current_workspace().await; let grid_view = test .create_grid(¤t_workspace.id, "my grid view".to_owned(), vec![]) .await; // By default the primary field type is RichText. let field = test.get_primary_field(&grid_view.id).await; assert_eq!(field.field_type, FieldType::RichText); } #[tokio::test] async fn create_row_event_test() { let test = EventIntegrationTest::new_anon().await; let current_workspace = test.get_current_workspace().await; let grid_view = test .create_grid(¤t_workspace.id, "my grid view".to_owned(), vec![]) .await; let _ = test .create_row(&grid_view.id, OrderObjectPositionPB::default(), None) .await; let database = test.get_database(&grid_view.id).await; assert_eq!(database.rows.len(), 4); } #[tokio::test] async fn delete_row_event_test() { let test = EventIntegrationTest::new_anon().await; let current_workspace = test.get_current_workspace().await; let grid_view = test .create_grid(¤t_workspace.id, "my grid view".to_owned(), vec![]) .await; // delete the row let database = test.get_database(&grid_view.id).await; let remove_row_id = database.rows[0].id.clone(); assert_eq!(database.rows.len(), 3); let error = test.delete_row(&grid_view.id, &remove_row_id).await; assert!(error.is_none()); let database = test.get_database(&grid_view.id).await; assert_eq!(database.rows.len(), 2); // get the row again and check if it is deleted. let optional_row = test.get_row(&grid_view.id, &remove_row_id).await; assert!(optional_row.row.is_none()); } #[tokio::test] async fn get_row_event_test() { let test = EventIntegrationTest::new_anon().await; let current_workspace = test.get_current_workspace().await; let grid_view = test .create_grid(¤t_workspace.id, "my grid view".to_owned(), vec![]) .await; let database = test.get_database(&grid_view.id).await; let row = test.get_row(&grid_view.id, &database.rows[0].id).await.row; assert!(row.is_some()); let row = test.get_row_meta(&grid_view.id, &database.rows[0].id).await; assert!(!row.document_id.is_empty()); } #[tokio::test] async fn update_row_meta_event_with_url_test() { let test = EventIntegrationTest::new_anon().await; let current_workspace = test.get_current_workspace().await; let grid_view = test .create_grid(¤t_workspace.id, "my grid view".to_owned(), vec![]) .await; let database = test.get_database(&grid_view.id).await; // By default the row icon is None. let row = test.get_row_meta(&grid_view.id, &database.rows[0].id).await; assert_eq!(row.icon, None); // Insert icon url to the row. let changeset = UpdateRowMetaChangesetPB { id: database.rows[0].id.clone(), view_id: grid_view.id.clone(), icon_url: Some("icon_url".to_owned()), cover_url: None, is_document_empty: None, }; let error = test.update_row_meta(changeset).await; assert!(error.is_none()); // Check if the icon is updated. let row = test.get_row_meta(&grid_view.id, &database.rows[0].id).await; assert_eq!(row.icon, Some("icon_url".to_owned())); } #[tokio::test] async fn update_row_meta_event_with_cover_test() { let test = EventIntegrationTest::new_anon().await; let current_workspace = test.get_current_workspace().await; let grid_view = test .create_grid(¤t_workspace.id, "my grid view".to_owned(), vec![]) .await; let database = test.get_database(&grid_view.id).await; // By default the row icon is None. let row = test.get_row_meta(&grid_view.id, &database.rows[0].id).await; assert_eq!(row.cover, None); // Insert cover to the row. let changeset = UpdateRowMetaChangesetPB { id: database.rows[0].id.clone(), view_id: grid_view.id.clone(), cover_url: Some("cover url".to_owned()), icon_url: None, is_document_empty: None, }; let error = test.update_row_meta(changeset).await; assert!(error.is_none()); // Check if the icon is updated. let row = test.get_row_meta(&grid_view.id, &database.rows[0].id).await; assert_eq!(row.cover, Some("cover url".to_owned())); } #[tokio::test] async fn delete_row_event_with_invalid_row_id_test() { let test = EventIntegrationTest::new_anon().await; let current_workspace = test.get_current_workspace().await; let grid_view = test .create_grid(¤t_workspace.id, "my grid view".to_owned(), vec![]) .await; // delete the row with empty row_id. It should do nothing let error = test.delete_row(&grid_view.id, "").await; assert!(error.is_none()); let database = test.get_database(&grid_view.id).await; assert_eq!(database.rows.len(), 3); } #[tokio::test] async fn duplicate_row_event_test() { let test = EventIntegrationTest::new_anon().await; let current_workspace = test.get_current_workspace().await; let grid_view = test .create_grid(¤t_workspace.id, "my grid view".to_owned(), vec![]) .await; let database = test.get_database(&grid_view.id).await; let error = test .duplicate_row(&grid_view.id, &database.rows[0].id) .await; assert!(error.is_none()); let database = test.get_database(&grid_view.id).await; assert_eq!(database.rows.len(), 4); } #[tokio::test] async fn duplicate_row_event_with_invalid_row_id_test() { let test = EventIntegrationTest::new_anon().await; let current_workspace = test.get_current_workspace().await; let grid_view = test .create_grid(¤t_workspace.id, "my grid view".to_owned(), vec![]) .await; let database = test.get_database(&grid_view.id).await; assert_eq!(database.rows.len(), 3); let error = test.duplicate_row(&grid_view.id, "").await; assert!(error.is_some()); let database = test.get_database(&grid_view.id).await; assert_eq!(database.rows.len(), 3); } #[tokio::test] async fn move_row_event_test() { let test = EventIntegrationTest::new_anon().await; let current_workspace = test.get_current_workspace().await; let grid_view = test .create_grid(¤t_workspace.id, "my grid view".to_owned(), vec![]) .await; let database = test.get_database(&grid_view.id).await; let row_1 = database.rows[0].id.clone(); let row_2 = database.rows[1].id.clone(); let row_3 = database.rows[2].id.clone(); let error = test.move_row(&grid_view.id, &row_1, &row_3).await; assert!(error.is_none()); let database = test.get_database(&grid_view.id).await; assert_eq!(database.rows[0].id, row_2); assert_eq!(database.rows[1].id, row_3); assert_eq!(database.rows[2].id, row_1); } #[tokio::test] async fn move_row_event_test2() { let test = EventIntegrationTest::new_anon().await; let current_workspace = test.get_current_workspace().await; let grid_view = test .create_grid(¤t_workspace.id, "my grid view".to_owned(), vec![]) .await; let database = test.get_database(&grid_view.id).await; let row_1 = database.rows[0].id.clone(); let row_2 = database.rows[1].id.clone(); let row_3 = database.rows[2].id.clone(); let error = test.move_row(&grid_view.id, &row_2, &row_1).await; assert!(error.is_none()); let database = test.get_database(&grid_view.id).await; assert_eq!(database.rows[0].id, row_2); assert_eq!(database.rows[1].id, row_1); assert_eq!(database.rows[2].id, row_3); } #[tokio::test] async fn move_row_event_with_invalid_row_id_test() { let test = EventIntegrationTest::new_anon().await; let current_workspace = test.get_current_workspace().await; let grid_view = test .create_grid(¤t_workspace.id, "my grid view".to_owned(), vec![]) .await; let database = test.get_database(&grid_view.id).await; let row_1 = database.rows[0].id.clone(); let row_2 = database.rows[1].id.clone(); let row_3 = database.rows[2].id.clone(); for i in 0..2 { if i == 0 { let error = test.move_row(&grid_view.id, &row_1, "").await; assert!(error.is_some()); } else { let error = test.move_row(&grid_view.id, "", &row_1).await; assert!(error.is_some()); } let database = test.get_database(&grid_view.id).await; assert_eq!(database.rows[0].id, row_1); assert_eq!(database.rows[1].id, row_2); assert_eq!(database.rows[2].id, row_3); } } #[tokio::test] async fn update_text_cell_event_test() { let test = EventIntegrationTest::new_anon().await; let current_workspace = test.get_current_workspace().await; let grid_view = test .create_grid(¤t_workspace.id, "my grid view".to_owned(), vec![]) .await; let database = test.get_database(&grid_view.id).await; let fields = test.get_all_database_fields(&grid_view.id).await.items; let row_id = database.rows[0].id.clone(); let field_id = fields[0].id.clone(); assert_eq!(fields[0].field_type, FieldType::RichText); // Update the first cell of the first row. let error = test .update_cell(CellChangesetPB { view_id: grid_view.id.clone(), row_id: row_id.clone(), field_id: field_id.clone(), cell_changeset: "hello world".to_string(), }) .await; assert!(error.is_none()); let cell = test.get_cell(&grid_view.id, &row_id, &field_id).await; let s = String::from_utf8(cell.data).unwrap(); assert_eq!(s, "hello world"); } #[tokio::test] async fn update_checkbox_cell_event_test() { let test = EventIntegrationTest::new_anon().await; let current_workspace = test.get_current_workspace().await; let grid_view = test .create_grid(¤t_workspace.id, "my grid view".to_owned(), vec![]) .await; let database = test.get_database(&grid_view.id).await; let fields = test.get_all_database_fields(&grid_view.id).await.items; let row_id = database.rows[0].id.clone(); let field_id = fields[2].id.clone(); assert_eq!(fields[2].field_type, FieldType::Checkbox); for input in &["yes", "true", "1"] { let error = test .update_cell(CellChangesetPB { view_id: grid_view.id.clone(), row_id: row_id.clone(), field_id: field_id.clone(), cell_changeset: input.to_string(), }) .await; assert!(error.is_none()); let cell = test.get_cell(&grid_view.id, &row_id, &field_id).await; let output = CheckboxCellDataPB::try_from(Bytes::from(cell.data)).unwrap(); assert!(output.is_checked); } } #[tokio::test] async fn update_single_select_cell_event_test() { let test = EventIntegrationTest::new_anon().await; let current_workspace = test.get_current_workspace().await; let grid_view = test .create_grid(¤t_workspace.id, "my grid view".to_owned(), vec![]) .await; let database = test.get_database(&grid_view.id).await; let fields = test.get_all_database_fields(&grid_view.id).await.items; let row_id = database.rows[0].id.clone(); let field_id = fields[1].id.clone(); assert_eq!(fields[1].field_type, FieldType::SingleSelect); // Insert a new option. This should update the cell with the new option. let error = test .insert_option(&grid_view.id, &field_id, &row_id, "task 1") .await; assert!(error.is_none()); // Check that the cell data is updated. let cell = test.get_cell(&grid_view.id, &row_id, &field_id).await; let select_option_cell = SelectOptionCellDataPB::try_from(Bytes::from(cell.data)).unwrap(); assert_eq!(select_option_cell.select_options.len(), 1); } #[tokio::test] async fn update_date_cell_event_test() { let test = EventIntegrationTest::new_anon().await; let current_workspace = test.get_current_workspace().await; let grid_view = test .create_grid(¤t_workspace.id, "my grid view".to_owned(), vec![]) .await; let database = test.get_database(&grid_view.id).await; // Create a date field let date_field = test.create_field(&grid_view.id, FieldType::DateTime).await; let cell_path = CellIdPB { view_id: grid_view.id.clone(), field_id: date_field.id.clone(), row_id: database.rows[0].id.clone(), }; // Insert data into the date cell of the first row. let timestamp = 1686300557; let error = test .update_date_cell(DateCellChangesetPB { cell_id: cell_path, date: Some(timestamp), ..Default::default() }) .await; assert!(error.is_none()); // Check that the cell data is updated. let cell = test .get_date_cell(&grid_view.id, &database.rows[0].id, &date_field.id) .await; assert_eq!(cell.date, "Jun 09, 2023"); assert_eq!(cell.timestamp, timestamp); } #[tokio::test] async fn update_date_cell_event_with_empty_time_str_test() { let test = EventIntegrationTest::new_anon().await; let current_workspace = test.get_current_workspace().await; let grid_view = test .create_grid(¤t_workspace.id, "my grid view".to_owned(), vec![]) .await; let database = test.get_database(&grid_view.id).await; let row_id = database.rows[0].id.clone(); // Create a date field let date_field = test.create_field(&grid_view.id, FieldType::DateTime).await; let cell_path = CellIdPB { view_id: grid_view.id.clone(), field_id: date_field.id.clone(), row_id: row_id.clone(), }; // Insert empty timestamp string let error = test .update_date_cell(DateCellChangesetPB { cell_id: cell_path, date: None, ..Default::default() }) .await; assert!(error.is_none()); // Check that the cell data is updated. let cell = test .get_date_cell(&grid_view.id, &row_id, &date_field.id) .await; assert_eq!(cell.date, ""); assert_eq!(cell.timestamp, 0); } #[tokio::test] async fn create_checklist_field_test() { let test = EventIntegrationTest::new_anon().await; let current_workspace = test.get_current_workspace().await; let grid_view = test .create_grid(¤t_workspace.id, "my grid view".to_owned(), vec![]) .await; // create checklist field let checklist_field = test.create_field(&grid_view.id, FieldType::Checklist).await; let database = test.get_database(&grid_view.id).await; // Get the checklist cell let cell = test .get_checklist_cell(&grid_view.id, &checklist_field.id, &database.rows[0].id) .await; assert_eq!(cell.options.len(), 0); assert_eq!(cell.selected_options.len(), 0); assert_eq!(cell.percentage, 0.0); } #[tokio::test] async fn update_checklist_cell_test() { let test = EventIntegrationTest::new_anon().await; let current_workspace = test.get_current_workspace().await; let grid_view = test .create_grid(¤t_workspace.id, "my grid view".to_owned(), vec![]) .await; // create checklist field let checklist_field = test.create_field(&grid_view.id, FieldType::Checklist).await; let database = test.get_database(&grid_view.id).await; // update the checklist cell let changeset = ChecklistCellDataChangesetPB { view_id: grid_view.id.clone(), row_id: database.rows[0].id.clone(), field_id: checklist_field.id.clone(), insert_options: vec![ "task 1".to_string(), "task 2".to_string(), "task 3".to_string(), ], selected_option_ids: vec![], delete_option_ids: vec![], update_options: vec![], }; test.update_checklist_cell(changeset).await; // get the cell let cell = test .get_checklist_cell(&grid_view.id, &checklist_field.id, &database.rows[0].id) .await; assert_eq!(cell.options.len(), 3); assert_eq!(cell.selected_options.len(), 0); // select some options let changeset = ChecklistCellDataChangesetPB { view_id: grid_view.id.clone(), row_id: database.rows[0].id.clone(), field_id: checklist_field.id.clone(), selected_option_ids: vec![cell.options[0].id.clone(), cell.options[1].id.clone()], ..Default::default() }; test.update_checklist_cell(changeset).await; // get the cell let cell = test .get_checklist_cell(&grid_view.id, &checklist_field.id, &database.rows[0].id) .await; assert_eq!(cell.options.len(), 3); assert_eq!(cell.selected_options.len(), 2); assert_eq!(cell.percentage, 0.67); } // Update the database layout type from grid to board #[tokio::test] async fn update_database_layout_event_test() { let test = EventIntegrationTest::new_anon().await; let current_workspace = test.get_current_workspace().await; let grid_view = test .create_grid(¤t_workspace.id, "my grid view".to_owned(), vec![]) .await; let error = test .update_setting(DatabaseSettingChangesetPB { view_id: grid_view.id.clone(), layout_type: Some(DatabaseLayoutPB::Board), ..Default::default() }) .await; assert!(error.is_none()); let database = test.get_database(&grid_view.id).await; assert_eq!(database.layout_type, DatabaseLayoutPB::Board); } // Update the database layout type from grid to board. Set the checkbox field as the grouping field #[tokio::test] async fn update_database_layout_event_test2() { let test = EventIntegrationTest::new_anon().await; let current_workspace = test.get_current_workspace().await; let grid_view = test .create_grid(¤t_workspace.id, "my grid view".to_owned(), vec![]) .await; let fields = test.get_all_database_fields(&grid_view.id).await.items; let checkbox_field = fields .iter() .find(|field| field.field_type == FieldType::Checkbox) .unwrap(); test .set_group_by_field(&grid_view.id, &checkbox_field.id, vec![]) .await; let error = test .update_setting(DatabaseSettingChangesetPB { view_id: grid_view.id.clone(), layout_type: Some(DatabaseLayoutPB::Board), ..Default::default() }) .await; assert!(error.is_none()); // Empty to group id let groups = test.get_groups(&grid_view.id).await; assert_eq!(groups.len(), 2); } // Create a checkbox field in the default board and then set it as the grouping field. #[tokio::test] async fn set_group_by_checkbox_field_test() { let test = EventIntegrationTest::new_anon().await; let current_workspace = test.get_current_workspace().await; let board_view = test .create_board(¤t_workspace.id, "my board view".to_owned(), vec![]) .await; let checkbox_field = test.create_field(&board_view.id, FieldType::Checkbox).await; test .set_group_by_field(&board_view.id, &checkbox_field.id, vec![]) .await; let groups = test.get_groups(&board_view.id).await; assert_eq!(groups.len(), 2); } #[tokio::test] async fn get_all_calendar_event_test() { let test = EventIntegrationTest::new_anon().await; let current_workspace = test.get_current_workspace().await; let calendar_view = test .create_calendar(¤t_workspace.id, "my calendar view".to_owned(), vec![]) .await; // By default, there should be no events let events = test.get_all_calendar_events(&calendar_view.id).await; assert!(events.is_empty()); } #[tokio::test] async fn create_calendar_event_test() { let test = EventIntegrationTest::new_anon().await; let current_workspace = test.get_current_workspace().await; let calendar_view = test .create_calendar(¤t_workspace.id, "my calendar view".to_owned(), vec![]) .await; let fields = test.get_all_database_fields(&calendar_view.id).await.items; let date_field = fields .iter() .find(|field| field.field_type == FieldType::DateTime) .unwrap(); // create a new row let row = test .create_row(&calendar_view.id, OrderObjectPositionPB::default(), None) .await; // Insert data into the date cell of the first row. let error = test .update_date_cell(DateCellChangesetPB { cell_id: CellIdPB { view_id: calendar_view.id.clone(), field_id: date_field.id.clone(), row_id: row.id, }, date: Some(timestamp()), ..Default::default() }) .await; assert!(error.is_none()); let events = test.get_all_calendar_events(&calendar_view.id).await; assert_eq!(events.len(), 1); } #[tokio::test] async fn update_relation_cell_test() { let test = EventIntegrationTest::new_anon().await; let current_workspace = test.get_current_workspace().await; let grid_view = test .create_grid(¤t_workspace.id, "my grid view".to_owned(), vec![]) .await; let relation_field = test.create_field(&grid_view.id, FieldType::Relation).await; let database = test.get_database(&grid_view.id).await; // update the relation cell let changeset = RelationCellChangesetPB { view_id: grid_view.id.clone(), cell_id: CellIdPB { view_id: grid_view.id.clone(), field_id: relation_field.id.clone(), row_id: database.rows[0].id.clone(), }, inserted_row_ids: vec![ "row1rowid".to_string(), "row2rowid".to_string(), "row3rowid".to_string(), ], ..Default::default() }; test.update_relation_cell(changeset).await; // get the cell let cell = test .get_relation_cell(&grid_view.id, &relation_field.id, &database.rows[0].id) .await; assert_eq!(cell.row_ids.len(), 3); // update the relation cell let changeset = RelationCellChangesetPB { view_id: grid_view.id.clone(), cell_id: CellIdPB { view_id: grid_view.id.clone(), field_id: relation_field.id.clone(), row_id: database.rows[0].id.clone(), }, removed_row_ids: vec![ "row1rowid".to_string(), "row3rowid".to_string(), "row4rowid".to_string(), ], ..Default::default() }; test.update_relation_cell(changeset).await; // get the cell let cell = test .get_relation_cell(&grid_view.id, &relation_field.id, &database.rows[0].id) .await; assert_eq!(cell.row_ids.len(), 1); } #[tokio::test] async fn get_detailed_relation_cell_data() { let test = EventIntegrationTest::new_anon().await; let current_workspace = test.get_current_workspace().await; let origin_grid_view = test .create_grid(¤t_workspace.id, "origin".to_owned(), vec![]) .await; let relation_grid_view = test .create_grid(¤t_workspace.id, "relation grid".to_owned(), vec![]) .await; let relation_field = test .create_field(&relation_grid_view.id, FieldType::Relation) .await; let origin_database = test.get_database(&origin_grid_view.id).await; let origin_fields = test.get_all_database_fields(&origin_grid_view.id).await; let linked_row = origin_database.rows[0].clone(); test .update_cell(CellChangesetPB { view_id: origin_grid_view.id.clone(), row_id: linked_row.id.clone(), field_id: origin_fields.items[0].id.clone(), cell_changeset: "hello world".to_string(), }) .await; let new_database = test.get_database(&relation_grid_view.id).await; // update the relation cell let changeset = RelationCellChangesetPB { view_id: relation_grid_view.id.clone(), cell_id: CellIdPB { view_id: relation_grid_view.id.clone(), field_id: relation_field.id.clone(), row_id: new_database.rows[0].id.clone(), }, inserted_row_ids: vec![linked_row.id.clone()], ..Default::default() }; test.update_relation_cell(changeset).await; // get the cell let cell = test .get_relation_cell( &relation_grid_view.id, &relation_field.id, &new_database.rows[0].id, ) .await; // using the row ids, get the row data let rows = test .get_related_row_data(origin_database.id.clone(), cell.row_ids) .await; assert_eq!(rows.len(), 1); assert_eq!(rows[0].name, "hello world"); }