AppFlowy/frontend/rust-lib/flowy-database2/tests/database/layout_test/test.rs

use collab_database::views::DatabaseLayout;
use flowy_database2::services::setting::BoardLayoutSetting;
use flowy_database2::services::setting::CalendarLayoutSetting;

use crate::database::layout_test::script::DatabaseLayoutTest;
use crate::database::layout_test::script::LayoutScript::*;

#[tokio::test]
async fn board_layout_setting_test() {
  let mut test = DatabaseLayoutTest::new_board().await;
  let default_board_setting = BoardLayoutSetting::new();
  let new_board_setting = BoardLayoutSetting {
    hide_ungrouped_column: true,
    ..default_board_setting
  };
  let scripts = vec![
    AssertBoardLayoutSetting {
      expected: default_board_setting,
    },
    UpdateBoardLayoutSetting {
      new_setting: new_board_setting.clone(),
    },
    AssertBoardLayoutSetting {
      expected: new_board_setting,
    },
  ];
  test.run_scripts(scripts).await;
}

#[tokio::test]
async fn calendar_initial_layout_setting_test() {
  let mut test = DatabaseLayoutTest::new_calendar().await;
  let date_field = test.get_first_date_field().await;
  let default_calendar_setting = CalendarLayoutSetting::new(date_field.id.clone());
  let scripts = vec![AssertCalendarLayoutSetting {
    expected: default_calendar_setting,
  }];
  test.run_scripts(scripts).await;
}

#[tokio::test]
async fn calendar_get_events_test() {
  let mut test = DatabaseLayoutTest::new_calendar().await;
  let scripts = vec![AssertDefaultAllCalendarEvents];
  test.run_scripts(scripts).await;
}

#[tokio::test]
async fn grid_to_calendar_layout_test() {
  let mut test = DatabaseLayoutTest::new_no_date_grid().await;
  let scripts = vec![
    UpdateDatabaseLayout {
      layout: DatabaseLayout::Calendar,
    },
    AssertAllCalendarEventsCount { expected: 3 },
  ];
  test.run_scripts(scripts).await;
}