AppFlowy/rust-lib/flowy-sdk/tests/sdk/helper.rs

pub use flowy_sdk::*;
use flowy_sys::prelude::*;
use std::{
    fmt::{Debug, Display},
    fs,
    hash::Hash,
    sync::Once,
};

static INIT: Once = Once::new();
pub fn init_sdk() {
    let root_dir = root_dir();

    INIT.call_once(|| {
        FlowySDK::init_log(&root_dir);
    });
    FlowySDK::init(&root_dir);
}

fn root_dir() -> String {
    let mut path = fs::canonicalize(".").unwrap();
    path.push("tests/temp/flowy/");
    let path_str = path.to_str().unwrap().to_string();
    if !std::path::Path::new(&path).exists() {
        std::fs::create_dir_all(path).unwrap();
    }
    path_str
}

pub struct EventTester {
    request: DispatchRequest,
}

impl EventTester {
    pub fn new<E, P>(event: E, payload: P) -> Self
    where
        E: Eq + Hash + Debug + Clone + Display,
        P: std::convert::Into<Payload>,
    {
        init_sdk();
        Self {
            request: DispatchRequest::new(event, payload.into()),
        }
    }

    // #[allow(dead_code)]
    // pub fn bytes_payload<T>(mut self, payload: T) -> Self
    // where
    //     T: serde::Serialize,
    // {
    //     let bytes: Vec<u8> = bincode::serialize(&payload).unwrap();
    //     self.request = self.request.payload(Payload::Bytes(bytes));
    //     self
    // }
    //
    // #[allow(dead_code)]
    // pub fn protobuf_payload<T>(mut self, payload: T) -> Self
    // where
    //     T: ::protobuf::Message,
    // {
    //     let bytes: Vec<u8> = payload.write_to_bytes().unwrap();
    //     self.request = self.request.payload(Payload::Bytes(bytes));
    //     self
    // }

    #[allow(dead_code)]
    pub async fn async_send(self) -> EventResponse {
        let resp = async_send(self.request).await;
        dbg!(&resp);
        resp
    }

    #[allow(dead_code)]
    pub fn sync_send(self) -> EventResponse {
        let resp = sync_send(self.request);
        dbg!(&resp);
        resp
    }
}