From 084a1e1ea9c06595068494089f63fbf3e8abcf3c Mon Sep 17 00:00:00 2001
From: sychen <sychen@ctrip.com>
Date: Sat, 18 Nov 2023 23:55:54 +0800
Subject: [PATCH] invalidCount

---
 .../org/apache/kyuubi/server/rest/client/BatchCliSuite.scala  | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/kyuubi-server/src/test/scala/org/apache/kyuubi/server/rest/client/BatchCliSuite.scala b/kyuubi-server/src/test/scala/org/apache/kyuubi/server/rest/client/BatchCliSuite.scala
index df93963e27b..599daa51377 100644
--- a/kyuubi-server/src/test/scala/org/apache/kyuubi/server/rest/client/BatchCliSuite.scala
+++ b/kyuubi-server/src/test/scala/org/apache/kyuubi/server/rest/client/BatchCliSuite.scala
@@ -153,11 +153,13 @@ class BatchCliSuite extends RestClientTestHelper with TestPrematureExit with Bat
       "--password",
       ldapUserPasswd)
     eventually(timeout(60.seconds), interval(100.milliseconds)) {
+      invalidCount += 1
       result = testPrematureExitForControlCli(logArgs, "")
       val rows = result.split("\n")
       assert(rows.length >= 2)
       // org.apache.spark.examples.DriverSubmissionTest output
       assert(result.contains("Alive for"))
+      invalidCount -= 1
     }
 
     val deleteArgs = Array(
@@ -172,7 +174,7 @@ class BatchCliSuite extends RestClientTestHelper with TestPrematureExit with Bat
 
     eventually(timeout(3.seconds), interval(200.milliseconds)) {
       assert(MetricsSystem.counterValue(
-        MetricsConstants.REST_CONN_TOTAL).getOrElse(0L) - totalConnections - invalidCount === 5)
+        MetricsConstants.REST_CONN_TOTAL).getOrElse(0L) - totalConnections - invalidCount >= 5)
       assert(MetricsSystem.counterValue(MetricsConstants.REST_CONN_OPEN).getOrElse(0L) === 0)
     }
   }