try fix

apache · Feb 29, 2024 · 6ae543c · 6ae543c
1 parent 729b7bf
commit 6ae543c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/...server/src/test/scala/org/apache/kyuubi/operation/KyuubiOperationPerConnectionSuite.scala b/...server/src/test/scala/org/apache/kyuubi/operation/KyuubiOperationPerConnectionSuite.scala
@@ -335,7 +335,8 @@ class KyuubiOperationPerConnectionSuite extends WithKyuubiServer with HiveJDBCTe
         assert(errorMsg.contains("java.net.SocketException") ||
           errorMsg.contains("org.apache.thrift.transport.TTransportException") ||
           errorMsg.contains("connection does not exist") ||
-          errorMsg.contains(s"Socket for ${SessionHandle(handle)} is closed"))
+          errorMsg.contains(s"Socket for ${SessionHandle(handle)} is closed") ||
+          errorMsg.contains("Error submitting query in background, query rejected"))
         val elapsedTime = System.currentTimeMillis() - startTime
         assert(elapsedTime < 20 * 1000)
         eventually(timeout(3.seconds)) {