AppFlowy/frontend/rust-lib/flowy-test/src/lib.rs

pub mod doc_script;
pub mod event_builder;
pub mod helper;

use crate::helper::*;
use backend_service::configuration::{get_client_server_configuration, ClientServerConfiguration};
use flowy_net::services::ws_conn::FlowyRawWebSocket;
use flowy_sdk::{FlowySDK, FlowySDKConfig};
use flowy_user::entities::UserProfile;
use lib_infra::uuid_string;
use std::sync::Arc;

pub mod prelude {
    pub use crate::{event_builder::*, helper::*, *};
    pub use lib_dispatch::prelude::*;
}

#[derive(Clone)]
pub struct FlowySDKTest {
    pub inner: FlowySDK,
    pub ws: Option<Arc<dyn FlowyRawWebSocket>>,
}

impl std::ops::Deref for FlowySDKTest {
    type Target = FlowySDK;

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

impl std::default::Default for FlowySDKTest {
    fn default() -> Self {
        let server_config = get_client_server_configuration().unwrap();
        let sdk = Self::new(server_config, None);
        std::mem::forget(sdk.dispatcher());
        sdk
    }
}

impl FlowySDKTest {
    pub fn new(server_config: ClientServerConfiguration, ws: Option<Arc<dyn FlowyRawWebSocket>>) -> Self {
        let config = FlowySDKConfig::new(&root_dir(), server_config, &uuid_string()).log_filter("debug");
        let sdk = FlowySDK::new(config);
        std::mem::forget(sdk.dispatcher());
        Self { inner: sdk, ws }
    }

    pub async fn sign_up(&self) -> SignUpContext {
        let context = async_sign_up(self.inner.dispatcher()).await;
        context
    }

    pub async fn init_user(&self) -> UserProfile {
        let context = async_sign_up(self.inner.dispatcher()).await;
        init_user_setting(self.inner.dispatcher()).await;
        context.user_profile
    }
}