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

use collab_database::fields::Field;
use collab_database::views::DatabaseLayout;

use flowy_database2::entities::{FieldType, LayoutSettingChangeset, LayoutSettingParams};
use flowy_database2::services::setting::{BoardLayoutSetting, CalendarLayoutSetting};

use crate::database::database_editor::DatabaseEditorTest;

pub struct DatabaseLayoutTest {
  database_test: DatabaseEditorTest,
}

impl DatabaseLayoutTest {
  pub async fn new_no_date_grid() -> Self {
    let database_test = DatabaseEditorTest::new_no_date_grid().await;
    Self { database_test }
  }

  pub async fn new_board() -> Self {
    let database_test = DatabaseEditorTest::new_board().await;
    Self { database_test }
  }

  pub async fn new_calendar() -> Self {
    let database_test = DatabaseEditorTest::new_calendar().await;
    Self { database_test }
  }

  pub async fn get_first_date_field(&self) -> Field {
    self
      .database_test
      .get_first_field(FieldType::DateTime)
      .await
  }

  async fn get_layout_setting(
    &self,
    view_id: &str,
    layout_ty: DatabaseLayout,
  ) -> LayoutSettingParams {
    self
      .database_test
      .editor
      .get_layout_setting(view_id, layout_ty)
      .await
      .unwrap()
  }

  pub async fn update_database_layout(&mut self, layout: DatabaseLayout) {
    self
      .database_test
      .editor
      .update_view_layout(&self.database_test.view_id, layout)
      .await
      .unwrap();
  }

  pub async fn assert_all_calendar_events_count(&self, expected: usize) {
    let events = self
      .database_test
      .editor
      .get_all_calendar_events(&self.database_test.view_id)
      .await;
    assert_eq!(events.len(), expected);
  }

  pub async fn assert_board_layout_setting(&self, expected: BoardLayoutSetting) {
    let view_id = self.database_test.view_id.clone();
    let layout_ty = DatabaseLayout::Board;

    let layout_settings = self.get_layout_setting(&view_id, layout_ty).await;

    assert!(layout_settings.calendar.is_none());
    assert_eq!(
      layout_settings.board.unwrap().hide_ungrouped_column,
      expected.hide_ungrouped_column
    );
  }

  pub async fn assert_calendar_layout_setting(&self, expected: CalendarLayoutSetting) {
    let view_id = self.database_test.view_id.clone();
    let layout_ty = DatabaseLayout::Calendar;

    let layout_settings = self.get_layout_setting(&view_id, layout_ty).await;

    assert!(layout_settings.board.is_none());

    let calendar_setting = layout_settings.calendar.unwrap();
    assert_eq!(calendar_setting.layout_ty, expected.layout_ty);
    assert_eq!(
      calendar_setting.first_day_of_week,
      expected.first_day_of_week
    );
    assert_eq!(calendar_setting.show_weekends, expected.show_weekends);
  }

  pub async fn update_board_layout_setting(&mut self, new_setting: BoardLayoutSetting) {
    let changeset = LayoutSettingChangeset {
      view_id: self.database_test.view_id.clone(),
      layout_type: DatabaseLayout::Board,
      board: Some(new_setting),
      calendar: None,
    };
    self
      .database_test
      .editor
      .set_layout_setting(&self.database_test.view_id, changeset)
      .await
      .unwrap();
  }

  pub async fn assert_default_all_calendar_events(&self) {
    let events = self
      .database_test
      .editor
      .get_all_calendar_events(&self.database_test.view_id)
      .await;
    assert_eq!(events.len(), 5);

    for (index, event) in events.into_iter().enumerate() {
      match index {
        0 => {
          assert_eq!(event.title, "A");
          assert_eq!(event.timestamp, Some(1678090778));
        },
        1 => {
          assert_eq!(event.title, "B");
          assert_eq!(event.timestamp, Some(1677917978));
        },
        2 => {
          assert_eq!(event.title, "C");
          assert_eq!(event.timestamp, Some(1679213978));
        },
        4 => {
          assert_eq!(event.title, "E");
          assert_eq!(event.timestamp, Some(1678695578));
        },
        _ => {},
      }
    }
  }
}