invalidCount

apache · Nov 18, 2023 · 084a1e1 · 084a1e1
1 parent a881cf5
commit 084a1e1
Showing 1 changed file with 3 additions and 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
@@ -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)
     }
   }