From a5af690aa51aceddf686353bce372ada2707e4a1 Mon Sep 17 00:00:00 2001
From: wangjunbo <wangjunbo@qiyi.com>
Date: Wed, 3 Apr 2024 10:17:04 +0800
Subject: [PATCH] fix unit test

---
 .github/workflows/master.yml                                     | 1 +
 .../kyuubi/operation/KyuubiOperationPerConnectionSuite.scala     | 1 +
 2 files changed, 2 insertions(+)

diff --git a/.github/workflows/master.yml b/.github/workflows/master.yml
index 907378a313c..5caa94529f0 100644
--- a/.github/workflows/master.yml
+++ b/.github/workflows/master.yml
@@ -128,6 +128,7 @@ jobs:
             **/target/unit-tests.log
             **/kyuubi-spark-sql-engine.log*
             **/kyuubi-spark-batch-submit.log*
+            !**/target/tmp/
 
   scala-test:
     name: Scala Test
diff --git a/kyuubi-server/src/test/scala/org/apache/kyuubi/operation/KyuubiOperationPerConnectionSuite.scala b/kyuubi-server/src/test/scala/org/apache/kyuubi/operation/KyuubiOperationPerConnectionSuite.scala
index 851830a4b83..711e637cf11 100644
--- a/kyuubi-server/src/test/scala/org/apache/kyuubi/operation/KyuubiOperationPerConnectionSuite.scala
+++ b/kyuubi-server/src/test/scala/org/apache/kyuubi/operation/KyuubiOperationPerConnectionSuite.scala
@@ -365,6 +365,7 @@ class KyuubiOperationPerConnectionSuite extends WithKyuubiServer with HiveJDBCTe
         exitReq.setRunAsync(true)
         client.ExecuteStatement(exitReq)
 
+        Thread.sleep(5000)
         val executeStmtReq = new TExecuteStatementReq()
         executeStmtReq.setSessionHandle(handle)
         executeStmtReq.setRunAsync(true)