AppFlowy/frontend/rust-lib/flowy-test/tests/document/af_cloud_test/util.rs

use std::ops::Deref;

use crate::util::{generate_test_email, AFCloudTest};

pub struct AFCloudDocumentTest {
  inner: AFCloudTest,
}

impl AFCloudDocumentTest {
  pub async fn new() -> Option<Self> {
    let inner = AFCloudTest::new()?;
    let email = generate_test_email();
    let _ = inner.af_cloud_sign_in_with_email(&email).await.unwrap();
    Some(Self { inner })
  }

  pub async fn create_document(&self) -> String {
    let current_workspace = self.inner.get_current_workspace().await;
    let view = self
      .inner
      .create_document(
        &current_workspace.workspace.id,
        "my document".to_string(),
        vec![],
      )
      .await;
    view.id
  }
}

impl Deref for AFCloudDocumentTest {
  type Target = AFCloudTest;

  fn deref(&self) -> &Self::Target {
    &self.inner
  }
}