76 lines
1.9 KiB
Rust
76 lines
1.9 KiB
Rust
use crate::user::utils::generate_unique_registered_user_client;
|
|
use client_api::ws::{ConnectState, WSClient, WSClientConfig};
|
|
|
|
#[tokio::test]
|
|
async fn realtime_connect_test() {
|
|
let (c, _user) = generate_unique_registered_user_client().await;
|
|
let ws_client = WSClient::new(WSClientConfig {
|
|
buffer_capacity: 100,
|
|
ping_per_secs: 6,
|
|
retry_connect_per_pings: 5,
|
|
});
|
|
let mut state = ws_client.subscribe_connect_state();
|
|
|
|
loop {
|
|
tokio::select! {
|
|
_ = ws_client.connect(c.ws_url("fake_device_id").unwrap()) => {},
|
|
value = state.recv() => {
|
|
let new_state = value.unwrap();
|
|
if new_state == ConnectState::Connected {
|
|
break;
|
|
}
|
|
},
|
|
}
|
|
}
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn realtime_disconnect_test() {
|
|
let (c, _user) = generate_unique_registered_user_client().await;
|
|
let ws_client = WSClient::new(WSClientConfig {
|
|
buffer_capacity: 100,
|
|
ping_per_secs: 6,
|
|
retry_connect_per_pings: 5,
|
|
});
|
|
ws_client
|
|
.connect(c.ws_url("fake_device_id").unwrap())
|
|
.await
|
|
.unwrap();
|
|
|
|
let mut state = ws_client.subscribe_connect_state();
|
|
loop {
|
|
tokio::select! {
|
|
_ = ws_client.disconnect() => {},
|
|
value = state.recv() => {
|
|
let new_state = value.unwrap();
|
|
if new_state == ConnectState::Disconnected {
|
|
break;
|
|
}
|
|
},
|
|
}
|
|
}
|
|
}
|
|
|
|
// #[tokio::test]
|
|
// async fn max_frame_size() {
|
|
// let (c, _user) = generate_unique_registered_user_client().await;
|
|
// let ws_client = WSClient::new(WSClientConfig {
|
|
// buffer_capacity: 100,
|
|
// ping_per_secs: 6,
|
|
// retry_connect_per_pings: 5,
|
|
// });
|
|
// ws_client
|
|
// .connect(c.ws_url("fake_device_id").unwrap())
|
|
// .await
|
|
// .unwrap();
|
|
//
|
|
// for _ in 0..10 {
|
|
// let sender = ws_client.sender();
|
|
// tokio::spawn(async move {
|
|
// sender.send(Message::Binary(vec![0; 65536])).unwrap();
|
|
// });
|
|
// }
|
|
//
|
|
// tokio::time::sleep(Duration::from_secs(5)).await;
|
|
// }
|