Function insert_collab_sql_test

tests/sql_test/workspace_test.rs:13–68

(pool: PgPool)

#[sqlx::test(migrations = false)]
async fn insert_collab_sql_test(pool: PgPool) {
  setup_db(&pool).await.unwrap();

  let user_uuid = uuid::Uuid::new_v4();
  let name = user_uuid.to_string();
  let email = format!("{}@appflowy.io", name);
  let user = create_test_user(&pool, user_uuid, &email, &name)
    .await
    .unwrap();

  let mut object_ids = vec![];

  let data_sizes = vec![
    5120,    // 5 KB
    10240,   // 10 KB
    102400,  // 100 KB
    512000,  // 500 KB
    5120000, // 5 MB
  ];
  let start_time = std::time::Instant::now();
  let updated_at = Utc::now();
  for &data_size in &data_sizes {
    let encoded_collab_v1 = generate_random_bytes(data_size);
    let object_id = uuid::Uuid::new_v4();
    object_ids.push(object_id);
    let mut txn = pool.begin().await.unwrap();
    let params = CollabParams {
      object_id,
      collab_type: CollabType::Unknown,
      encoded_collab_v1: encoded_collab_v1.into(),
      updated_at: Some(updated_at),
    };
    insert_into_af_collab(&mut txn, &user.uid, &user.workspace_id, &params)
      .await
      .unwrap();
    txn.commit().await.unwrap();
  }
  let duration = start_time.elapsed();
  println!("Insert time: {:?}", duration);

  for object_id in object_ids {
    let meta = select_collab_meta_from_af_collab(&pool, &object_id, &CollabType::Unknown)
      .await
      .unwrap()
      .unwrap();

    assert_eq!(meta.oid, object_id.to_string());
    assert_eq!(meta.workspace_id, user.workspace_id);
    assert_eq!(
      meta.updated_at.timestamp_millis(),
      updated_at.timestamp_millis()
    );
    assert!(meta.created_at.is_some());
    assert!(meta.deleted_at.is_none());
  }
}
#[sqlx::test(migrations = false)]
async fn insert_bulk_collab_sql_test(pool: PgPool) {