AppFlowy/frontend/rust-lib/flowy-database/tests/grid/group_test/script.rs

use crate::grid::database_editor::DatabaseEditorTest;
use flowy_database::entities::{
    CreateRowParams, FieldType, GroupPB, LayoutTypePB, MoveGroupParams, MoveGroupRowParams, RowPB,
};
use flowy_database::services::cell::{delete_select_option_cell, insert_select_option_cell, insert_url_cell};
use flowy_database::services::field::{
    edit_single_select_type_option, SelectOptionPB, SelectTypeOptionSharedAction, SingleSelectTypeOptionPB,
};
use grid_model::{FieldRevision, RowChangeset};
use std::sync::Arc;

pub enum GroupScript {
    AssertGroupRowCount {
        group_index: usize,
        row_count: usize,
    },
    AssertGroupCount(usize),
    AssertGroup {
        group_index: usize,
        expected_group: GroupPB,
    },
    AssertRow {
        group_index: usize,
        row_index: usize,
        row: RowPB,
    },
    MoveRow {
        from_group_index: usize,
        from_row_index: usize,
        to_group_index: usize,
        to_row_index: usize,
    },
    CreateRow {
        group_index: usize,
    },
    DeleteRow {
        group_index: usize,
        row_index: usize,
    },
    UpdateGroupedCell {
        from_group_index: usize,
        row_index: usize,
        to_group_index: usize,
    },
    UpdateGroupedCellWithData {
        from_group_index: usize,
        row_index: usize,
        cell_data: String,
    },
    MoveGroup {
        from_group_index: usize,
        to_group_index: usize,
    },
    UpdateSingleSelectSelectOption {
        inserted_options: Vec<SelectOptionPB>,
    },
    GroupByField {
        field_id: String,
    },
}

pub struct DatabaseGroupTest {
    inner: DatabaseEditorTest,
}

impl DatabaseGroupTest {
    pub async fn new() -> Self {
        let editor_test = DatabaseEditorTest::new_board().await;
        Self { inner: editor_test }
    }

    pub async fn run_scripts(&mut self, scripts: Vec<GroupScript>) {
        for script in scripts {
            self.run_script(script).await;
        }
    }

    pub async fn run_script(&mut self, script: GroupScript) {
        match script {
            GroupScript::AssertGroupRowCount { group_index, row_count } => {
                assert_eq!(row_count, self.group_at_index(group_index).await.rows.len());
            }
            GroupScript::AssertGroupCount(count) => {
                let groups = self.editor.load_groups().await.unwrap();
                assert_eq!(count, groups.len());
            }
            GroupScript::MoveRow {
                from_group_index,
                from_row_index,
                to_group_index,
                to_row_index,
            } => {
                let groups: Vec<GroupPB> = self.editor.load_groups().await.unwrap().items;
                let from_row = groups.get(from_group_index).unwrap().rows.get(from_row_index).unwrap();
                let to_group = groups.get(to_group_index).unwrap();
                let to_row = to_group.rows.get(to_row_index).unwrap();
                let params = MoveGroupRowParams {
                    view_id: self.inner.view_id.clone(),
                    from_row_id: from_row.id.clone(),
                    to_group_id: to_group.group_id.clone(),
                    to_row_id: Some(to_row.id.clone()),
                };

                self.editor.move_group_row(params).await.unwrap();
            }
            GroupScript::AssertRow {
                group_index,
                row_index,
                row,
            } => {
                //
                let group = self.group_at_index(group_index).await;
                let compare_row = group.rows.get(row_index).unwrap().clone();
                assert_eq!(row.id, compare_row.id);
            }
            GroupScript::CreateRow { group_index } => {
                let group = self.group_at_index(group_index).await;
                let params = CreateRowParams {
                    database_id: self.editor.database_id.clone(),
                    start_row_id: None,
                    group_id: Some(group.group_id.clone()),
                    layout: LayoutTypePB::Board,
                };
                let _ = self.editor.create_row(params).await.unwrap();
            }
            GroupScript::DeleteRow { group_index, row_index } => {
                let row = self.row_at_index(group_index, row_index).await;
                self.editor.delete_row(&row.id).await.unwrap();
            }
            GroupScript::UpdateGroupedCell {
                from_group_index,
                row_index,
                to_group_index,
            } => {
                let from_group = self.group_at_index(from_group_index).await;
                let to_group = self.group_at_index(to_group_index).await;
                let field_id = from_group.field_id;
                let field_rev = self.editor.get_field_rev(&field_id).await.unwrap();
                let field_type: FieldType = field_rev.ty.into();

                let cell_rev = if to_group.is_default {
                    match field_type {
                        FieldType::SingleSelect => {
                            delete_select_option_cell(vec![to_group.group_id.clone()], &field_rev)
                        }
                        FieldType::MultiSelect => {
                            delete_select_option_cell(vec![to_group.group_id.clone()], &field_rev)
                        }
                        _ => {
                            panic!("Unsupported group field type");
                        }
                    }
                } else {
                    match field_type {
                        FieldType::SingleSelect => {
                            insert_select_option_cell(vec![to_group.group_id.clone()], &field_rev)
                        }
                        FieldType::MultiSelect => {
                            insert_select_option_cell(vec![to_group.group_id.clone()], &field_rev)
                        }
                        FieldType::URL => insert_url_cell(to_group.group_id.clone(), &field_rev),
                        _ => {
                            panic!("Unsupported group field type");
                        }
                    }
                };

                let row_id = self.row_at_index(from_group_index, row_index).await.id;
                let mut row_changeset = RowChangeset::new(row_id);
                row_changeset.cell_by_field_id.insert(field_id, cell_rev);
                self.editor.update_row(row_changeset).await.unwrap();
            }
            GroupScript::UpdateGroupedCellWithData {
                from_group_index,
                row_index,
                cell_data,
            } => {
                let from_group = self.group_at_index(from_group_index).await;
                let field_id = from_group.field_id;
                let field_rev = self.editor.get_field_rev(&field_id).await.unwrap();
                let field_type: FieldType = field_rev.ty.into();
                let cell_rev = match field_type {
                    FieldType::URL => insert_url_cell(cell_data, &field_rev),
                    _ => {
                        panic!("Unsupported group field type");
                    }
                };

                let row_id = self.row_at_index(from_group_index, row_index).await.id;
                let mut row_changeset = RowChangeset::new(row_id);
                row_changeset.cell_by_field_id.insert(field_id, cell_rev);
                self.editor.update_row(row_changeset).await.unwrap();
            }
            GroupScript::MoveGroup {
                from_group_index,
                to_group_index,
            } => {
                let from_group = self.group_at_index(from_group_index).await;
                let to_group = self.group_at_index(to_group_index).await;
                let params = MoveGroupParams {
                    view_id: self.editor.database_id.clone(),
                    from_group_id: from_group.group_id,
                    to_group_id: to_group.group_id,
                };
                self.editor.move_group(params).await.unwrap();
                //
            }
            GroupScript::AssertGroup {
                group_index,
                expected_group: group_pb,
            } => {
                let group = self.group_at_index(group_index).await;
                assert_eq!(group.group_id, group_pb.group_id);
                assert_eq!(group.desc, group_pb.desc);
            }
            GroupScript::UpdateSingleSelectSelectOption { inserted_options } => {
                self.edit_single_select_type_option(|type_option| {
                    for inserted_option in inserted_options {
                        type_option.insert_option(inserted_option);
                    }
                })
                .await;
            }
            GroupScript::GroupByField { field_id } => {
                self.editor.group_by_field(&field_id).await.unwrap();
            }
        }
    }

    pub async fn group_at_index(&self, index: usize) -> GroupPB {
        let groups = self.editor.load_groups().await.unwrap().items;
        groups.get(index).unwrap().clone()
    }

    pub async fn row_at_index(&self, group_index: usize, row_index: usize) -> RowPB {
        let groups = self.group_at_index(group_index).await;
        groups.rows.get(row_index).unwrap().clone()
    }

    #[allow(dead_code)]
    pub async fn get_multi_select_field(&self) -> Arc<FieldRevision> {
        let field = self
            .inner
            .field_revs
            .iter()
            .find(|field_rev| {
                let field_type: FieldType = field_rev.ty.into();
                field_type.is_multi_select()
            })
            .unwrap()
            .clone();
        field
    }

    pub async fn get_single_select_field(&self) -> Arc<FieldRevision> {
        self.inner
            .field_revs
            .iter()
            .find(|field_rev| {
                let field_type: FieldType = field_rev.ty.into();
                field_type.is_single_select()
            })
            .unwrap()
            .clone()
    }

    pub async fn edit_single_select_type_option(&self, action: impl FnOnce(&mut SingleSelectTypeOptionPB)) {
        let single_select = self.get_single_select_field().await;
        edit_single_select_type_option(&single_select.id, self.editor.clone(), action)
            .await
            .unwrap();
    }

    pub async fn get_url_field(&self) -> Arc<FieldRevision> {
        self.inner
            .field_revs
            .iter()
            .find(|field_rev| {
                let field_type: FieldType = field_rev.ty.into();
                field_type.is_url()
            })
            .unwrap()
            .clone()
    }
}

impl std::ops::Deref for DatabaseGroupTest {
    type Target = DatabaseEditorTest;

    fn deref(&self) -> &Self::Target {
        &self.inner
    }
}

impl std::ops::DerefMut for DatabaseGroupTest {
    fn deref_mut(&mut self) -> &mut Self::Target {
        &mut self.inner
    }
}