From ae2237c4ae168faed46dcf375ff23b26ad7da72b Mon Sep 17 00:00:00 2001
From: Yang Xiufeng <yangxiufeng.c@gmail.com>
Date: Sat, 14 Sep 2024 01:21:27 +0800
Subject: [PATCH] fix.

---
 src/query/service/tests/it/auth.rs                     |  8 ++++----
 .../tests/it/servers/http/http_query_handlers.rs       | 10 ++++++++++
 2 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/src/query/service/tests/it/auth.rs b/src/query/service/tests/it/auth.rs
index ac14a84f453fb..4fcebcc5d1ab6 100644
--- a/src/query/service/tests/it/auth.rs
+++ b/src/query/service/tests/it/auth.rs
@@ -131,7 +131,7 @@ async fn test_auth_mgr_with_jwt_multi_sources() -> Result<()> {
             .auth(&mut session, &Credential::Jwt {
                 token: token2,
                 client_ip: None,
-            })
+            }, true)
             .await;
         assert!(res.is_ok());
 
@@ -269,7 +269,7 @@ async fn test_auth_mgr_with_jwt() -> Result<()> {
             .auth(&mut session, &Credential::Jwt {
                 token,
                 client_ip: None,
-            })
+            }, true)
             .await;
         assert!(res.is_err());
         assert!(
@@ -291,7 +291,7 @@ async fn test_auth_mgr_with_jwt() -> Result<()> {
             .auth(&mut session, &Credential::Jwt {
                 token,
                 client_ip: None,
-            })
+            }, true)
             .await;
         assert!(res.is_err());
         assert!(
@@ -313,7 +313,7 @@ async fn test_auth_mgr_with_jwt() -> Result<()> {
             .auth(&mut session, &Credential::Jwt {
                 token,
                 client_ip: None,
-            })
+            }, true)
             .await?;
         let user_info = session.get_current_user()?;
         assert_eq!(user_info.grants.roles().len(), 0);
diff --git a/src/query/service/tests/it/servers/http/http_query_handlers.rs b/src/query/service/tests/it/servers/http/http_query_handlers.rs
index 6fa2fd8f5e5d9..23d611edfbc44 100644
--- a/src/query/service/tests/it/servers/http/http_query_handlers.rs
+++ b/src/query/service/tests/it/servers/http/http_query_handlers.rs
@@ -1391,6 +1391,8 @@ async fn test_affect() -> Result<()> {
                     ("timezone".to_string(), "Asia/Shanghai".to_string()),
                 ])),
                 txn_state: Some(TxnState::AutoCommit),
+                need_sticky: false,
+                need_refresh: false,
                 last_server_info: None,
                 last_query_ids: vec![],
                 internal: None,
@@ -1414,6 +1416,8 @@ async fn test_affect() -> Result<()> {
                     "6".to_string(),
                 )])),
                 txn_state: Some(TxnState::AutoCommit),
+                need_sticky: false,
+                need_refresh: false,
                 last_server_info: None,
                 last_query_ids: vec![],
                 internal: None,
@@ -1432,6 +1436,8 @@ async fn test_affect() -> Result<()> {
                     "6".to_string(),
                 )])),
                 txn_state: Some(TxnState::AutoCommit),
+                need_sticky: false,
+                need_refresh: false,
                 last_server_info: None,
                 last_query_ids: vec![],
                 internal: None,
@@ -1452,6 +1458,8 @@ async fn test_affect() -> Result<()> {
                     "6".to_string(),
                 )])),
                 txn_state: Some(TxnState::AutoCommit),
+                need_sticky: false,
+                need_refresh: false,
                 last_server_info: None,
                 last_query_ids: vec![],
                 internal: None,
@@ -1474,6 +1482,8 @@ async fn test_affect() -> Result<()> {
                     "Asia/Shanghai".to_string(),
                 )])),
                 txn_state: Some(TxnState::AutoCommit),
+                need_sticky: false,
+                need_refresh: false,
                 last_server_info: None,
                 last_query_ids: vec![],
                 internal: None,