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

fix: ws connect test with expired token

This commit is contained in:
Fu Zi Xiang 2024-01-22 13:19:15 +08:00
parent 107423562d
commit 3907374cd1
No known key found for this signature in database
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tests/websocket/connect.rs
View File

@ -1,3 +1,5 @@
use std::time::SystemTime;
use crate::user::utils::generate_unique_registered_user_client; use crate::user::utils::generate_unique_registered_user_client;
use client_api::{ConnectState, WSClient, WSClientConfig}; use client_api::{ConnectState, WSClient, WSClientConfig};
@ -23,6 +25,13 @@ async fn realtime_connect_test() {
#[tokio::test] #[tokio::test]
async fn realtime_connect_after_token_exp_test() { async fn realtime_connect_after_token_exp_test() {
let (c, _user) = generate_unique_registered_user_client().await; let (c, _user) = generate_unique_registered_user_client().await;
// Set the token to be expired
c.token().write().as_mut().unwrap().expires_at = SystemTime::now()
.duration_since(SystemTime::UNIX_EPOCH)
.unwrap()
.as_secs() as i64;
let ws_client = WSClient::new(WSClientConfig::default(), c.clone()); let ws_client = WSClient::new(WSClientConfig::default(), c.clone());
let mut state = ws_client.subscribe_connect_state(); let mut state = ws_client.subscribe_connect_state();
let device_id = "fake_device_id"; let device_id = "fake_device_id";