chk
/
AppFlowy-Cloud
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: handle receiver close for connection test (#462)

This commit is contained in:
Khor Shu Heng 2024-04-10 09:30:08 +08:00 committed by GitHub
parent 517b7b6760
commit c7ebdbeda7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 14 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
tests/websocket/conn_test.rs
View File

@ -12,9 +12,13 @@ async fn realtime_connect_test() {
let connect_info = c.ws_connect_info().await.unwrap(); let connect_info = c.ws_connect_info().await.unwrap();
tokio::spawn(async move { ws_client.connect(&c.ws_url(), connect_info).await }); tokio::spawn(async move { ws_client.connect(&c.ws_url(), connect_info).await });
let connect_future = async { let connect_future = async {
while let Ok(state) = state.recv().await { loop {
if state == ConnectState::Connected { match state.recv().await {
break; Ok(ConnectState::Connected) => {
break;
},
Ok(_) => {},
Err(err) => panic!("Receiver Error: {:?}", err),
} }
} }
}; };
@ -42,9 +46,13 @@ async fn realtime_connect_after_token_exp_test() {
tokio::spawn(async move { ws_client.connect(&c.ws_url(), connect_info).await }); tokio::spawn(async move { ws_client.connect(&c.ws_url(), connect_info).await });
let connect_future = async { let connect_future = async {
while let Ok(state) = state.recv().await { loop {
if state == ConnectState::Connected { match state.recv().await {
break; Ok(ConnectState::Connected) => {
break;
},
Ok(_) => {},
Err(err) => panic!("Receiver Error: {:?}", err),
} }
} }
}; };