From 292ea91c8dc5c33dec58d3b005b44d0fd2f1a8de Mon Sep 17 00:00:00 2001 From: sychen Date: Mon, 19 Feb 2024 21:47:08 +0800 Subject: [PATCH] DriverSubmissionTest --- .../kyuubi/server/api/v1/BatchesResourceSuite.scala | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/kyuubi-server/src/test/scala/org/apache/kyuubi/server/api/v1/BatchesResourceSuite.scala b/kyuubi-server/src/test/scala/org/apache/kyuubi/server/api/v1/BatchesResourceSuite.scala index 65c58c6cb0b..fd4cca329f9 100644 --- a/kyuubi-server/src/test/scala/org/apache/kyuubi/server/api/v1/BatchesResourceSuite.scala +++ b/kyuubi-server/src/test/scala/org/apache/kyuubi/server/api/v1/BatchesResourceSuite.scala @@ -751,9 +751,15 @@ abstract class BatchesResourceSuiteBase extends KyuubiFunSuite getBatchJobSubmissionStateCounter(OperationState.ERROR) val batchId = UUID.randomUUID().toString - val requestObj = newSparkBatchRequest(Map( - "spark.master" -> "local", - KYUUBI_BATCH_ID_KEY -> batchId)) + val requestObj = newBatchRequest( + sparkBatchTestBatchType, + sparkBatchTestResource.get, + "org.apache.spark.examples.DriverSubmissionTest", + "DriverSubmissionTest-" + batchId, + Map( + "spark.master" -> "local", + KYUUBI_BATCH_ID_KEY -> batchId), + Seq("120")) eventually(timeout(10.seconds)) { val response = webTarget.path("api/v1/batches")