diff --git a/kyuubi-server/src/test/scala/org/apache/kyuubi/server/rest/client/BatchRestApiSuite.scala b/kyuubi-server/src/test/scala/org/apache/kyuubi/server/rest/client/BatchRestApiSuite.scala index ef53fbe3e..d04826a9d 100644 --- a/kyuubi-server/src/test/scala/org/apache/kyuubi/server/rest/client/BatchRestApiSuite.scala +++ b/kyuubi-server/src/test/scala/org/apache/kyuubi/server/rest/client/BatchRestApiSuite.scala @@ -68,8 +68,10 @@ class BatchRestApiSuite extends RestClientTestHelper with BatchTestHelper { assert(batch.getBatchType === "SPARK") // get batch log - val log = batchRestApi.getBatchLocalLog(batch.getId(), 0, 1) - assert(log.getRowCount == 1) + eventually(timeout(1.minutes)) { + val log = batchRestApi.getBatchLocalLog(batch.getId(), 0, 1) + assert(log.getRowCount == 1) + } // delete batch val closeResp = batchRestApi.deleteBatch(batch.getId(), null)