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

use crate::database::database_editor::DatabaseEditorTest;
use database_model::{CalendarLayoutSetting, FieldRevision, LayoutRevision};
use flowy_database::entities::FieldType;
use std::sync::Arc;

pub enum LayoutScript {
  AssertCalendarLayoutSetting { expected: CalendarLayoutSetting },
  GetCalendarEvents,
}

pub struct DatabaseLayoutTest {
  database_test: DatabaseEditorTest,
}

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

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

  pub async fn get_first_date_field(&self) -> Arc<FieldRevision> {
    self
      .database_test
      .get_first_field_rev(FieldType::DateTime)
      .clone()
  }

  pub async fn run_script(&mut self, script: LayoutScript) {
    match script {
      LayoutScript::AssertCalendarLayoutSetting { expected } => {
        let view_id = self.database_test.view_id.clone();
        let layout_ty = LayoutRevision::Calendar;

        let calendar_setting = self
          .database_test
          .editor
          .get_layout_setting(&view_id, layout_ty)
          .await
          .unwrap()
          .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);
      },
      LayoutScript::GetCalendarEvents => {
        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() {
          if index == 0 {
            assert_eq!(event.title, "A");
            assert_eq!(event.timestamp, 1678090778);
          }

          if index == 1 {
            assert_eq!(event.title, "B");
            assert_eq!(event.timestamp, 1677917978);
          }
          if index == 2 {
            assert_eq!(event.title, "C");
            assert_eq!(event.timestamp, 1679213978);
          }
          if index == 4 {
            assert_eq!(event.title, "E");
            assert_eq!(event.timestamp, 1678695578);
          }
        }
      },
    }
  }
}