use std::ops::Deref; use std::time::Duration; use tokio::sync::mpsc::Receiver; use tokio::time::timeout; use flowy_server_config::supabase_config::SupabaseConfiguration; use flowy_test::event_builder::EventBuilder; use flowy_test::FlowyCoreTest; use flowy_user::entities::{ AuthTypePB, UpdateUserProfilePayloadPB, UserCredentialsPB, UserProfilePB, }; use flowy_user::errors::FlowyError; use flowy_user::event_map::UserCloudServiceProvider; use flowy_user::event_map::UserEvent::*; use flowy_user_deps::entities::AuthType; pub fn get_supabase_config() -> Option { dotenv::from_path(".env.test").ok()?; SupabaseConfiguration::from_env().ok() } pub struct FlowySupabaseTest { inner: FlowyCoreTest, } impl FlowySupabaseTest { pub fn new() -> Option { let _ = get_supabase_config()?; let test = FlowyCoreTest::new(); test.set_auth_type(AuthTypePB::Supabase); test.server_provider.set_auth_type(AuthType::Supabase); Some(Self { inner: test }) } pub async fn check_user_with_uuid(&self, uuid: &str) -> Result<(), FlowyError> { match EventBuilder::new(self.inner.clone()) .event(CheckUser) .payload(UserCredentialsPB::from_uuid(uuid)) .async_send() .await .error() { None => Ok(()), Some(error) => Err(error), } } pub async fn get_user_profile(&self) -> Result { EventBuilder::new(self.inner.clone()) .event(GetUserProfile) .async_send() .await .try_parse::() } pub async fn update_user_profile( &self, payload: UpdateUserProfilePayloadPB, ) -> Option { EventBuilder::new(self.inner.clone()) .event(UpdateUserProfile) .payload(payload) .async_send() .await .error() } } impl Deref for FlowySupabaseTest { type Target = FlowyCoreTest; fn deref(&self) -> &Self::Target { &self.inner } } pub async fn receive_with_timeout( receiver: &mut Receiver, duration: Duration, ) -> Result> { let res = timeout(duration, receiver.recv()) .await? .ok_or(anyhow::anyhow!("recv timeout"))?; Ok(res) }