From 3907374cd12c0865639fee83b5e40b3c4a567ea0 Mon Sep 17 00:00:00 2001 From: Fu Zi Xiang Date: Mon, 22 Jan 2024 13:19:15 +0800 Subject: [PATCH] fix: ws connect test with expired token --- tests/websocket/connect.rs | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/tests/websocket/connect.rs b/tests/websocket/connect.rs index 03779da7..c51f626a 100644 --- a/tests/websocket/connect.rs +++ b/tests/websocket/connect.rs @@ -1,3 +1,5 @@ +use std::time::SystemTime; + use crate::user::utils::generate_unique_registered_user_client; use client_api::{ConnectState, WSClient, WSClientConfig}; @@ -23,6 +25,13 @@ async fn realtime_connect_test() { #[tokio::test] async fn realtime_connect_after_token_exp_test() { 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 mut state = ws_client.subscribe_connect_state(); let device_id = "fake_device_id";