mirror of
https://github.com/AppFlowy-IO/AppFlowy.git
synced 2025-07-14 04:26:02 +00:00
28 lines
677 B
Rust
28 lines
677 B
Rust
|
use std::time::Duration;
|
||
|
|
|
||
|
|
use event_integration::EventIntegrationTest;
|
||
|
|
use flowy_user::{
|
||
|
|
entities::{RepeatedUserWorkspacePB, UserWorkspacePB},
|
||
|
|
protobuf::UserNotification,
|
||
|
|
};
|
||
|
|
|
||
|
|
use crate::util::receive_with_timeout;
|
||
|
|
|
||
|
|
pub async fn get_synced_workspaces(
|
||
|
|
test: &EventIntegrationTest,
|
||
|
|
user_id: i64,
|
||
|
|
) -> Vec<UserWorkspacePB> {
|
||
|
|
let _workspaces = test.get_all_workspaces().await.items;
|
||
|
|
let sub_id = user_id.to_string();
|
||
|
|
let rx = test
|
||
|
|
.notification_sender
|
||
|
|
.subscribe::<RepeatedUserWorkspacePB>(
|
||
|
|
&sub_id,
|
||
|
|
UserNotification::DidUpdateUserWorkspaces as i32,
|
||
|
|
);
|
||
|
|
receive_with_timeout(rx, Duration::from_secs(30))
|
||
|
|
.await
|
||
|
|
.unwrap()
|
||
|
|
.items
|
||
|
|
}
|