AppFlowy/frontend/rust-lib/flowy-database2/tests/database/block_test/row_test.rs

use collab_database::fields::date_type_option::DateCellData;
use flowy_database2::entities::FieldType;
use lib_infra::util::timestamp;
use std::time::Duration;

use crate::database::block_test::script::DatabaseRowTest;

#[tokio::test]
async fn created_at_field_test() {
  let mut test = DatabaseRowTest::new().await;

  // Get initial row count
  let row_count = test.rows.len();

  // Create a new row and assert the row count has increased by 1
  test.create_empty_row().await;
  test.assert_row_count(row_count + 1).await;

  // Get created time of the new row.
  let row = test.get_rows().await.last().cloned().unwrap();
  let created_at_field = test.get_first_field(FieldType::CreatedTime).await;
  let cell = test
    .editor
    .get_cell(&created_at_field.id, &row.id)
    .await
    .unwrap();
  let created_at_timestamp = DateCellData::from(&cell).timestamp.unwrap();

  assert!(created_at_timestamp > 0);
  assert!(created_at_timestamp <= timestamp());
}

#[tokio::test]
async fn update_at_field_test() {
  let mut test = DatabaseRowTest::new().await;

  // Get the first row and the current LastEditedTime field
  let row = test.get_rows().await.remove(0);
  let last_edit_field = test.get_first_field(FieldType::LastEditedTime).await;
  let cell = test
    .editor
    .get_cell(&last_edit_field.id, &row.id)
    .await
    .unwrap();
  let old_updated_at = DateCellData::from(&cell).timestamp.unwrap();

  // Wait for 1 second before updating the row
  tokio::time::sleep(Duration::from_millis(1000)).await;

  // Update the text cell of the first row
  test.update_text_cell(row.id.clone(), "test").await;

  // Get the updated time of the row
  let row = test.get_rows().await.remove(0);
  let last_edit_field = test.get_first_field(FieldType::LastEditedTime).await;
  let cell = test
    .editor
    .get_cell(&last_edit_field.id, &row.id)
    .await
    .unwrap();
  let new_updated_at = DateCellData::from(&cell).timestamp.unwrap();

  assert!(old_updated_at < new_updated_at);
}