Add more UT on config class

Signed-off-by: Chen Dai <[email protected]>

spark/src/test/java/org/opensearch/sql/spark/config/SparkExecutionEngineConfigTest.java

@@ -0,0 +1,49 @@
+/*
+ * Copyright OpenSearch Contributors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+package org.opensearch.sql.spark.config;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertNull;
+
+import org.junit.jupiter.api.Test;
+
+public class SparkExecutionEngineConfigTest {
+
+  @Test
+  public void testToSparkExecutionEngineConfigWithoutAllFields() {
+    String json =
+        "{"
+            + "\"applicationId\": \"app-1\","
+            + "\"executionRoleARN\": \"role-1\","
+            + "\"region\": \"us-west-1\""
+            + "}";
+    SparkExecutionEngineConfig config =
+        SparkExecutionEngineConfig.toSparkExecutionEngineConfig(json);
+
+    assertEquals("app-1", config.getApplicationId());
+    assertEquals("role-1", config.getExecutionRoleARN());
+    assertEquals("us-west-1", config.getRegion());
+    assertNull(config.getSparkSubmitParameters());
+  }
+
+  @Test
+  public void testToSparkExecutionEngineConfigWithAllFields() {
+    String json =
+        "{"
+            + "\"applicationId\": \"app-1\","
+            + "\"executionRoleARN\": \"role-1\","
+            + "\"region\": \"us-west-1\","
+            + "\"sparkSubmitParameters\": \"--conf A=1\""
+            + "}";
+    SparkExecutionEngineConfig config =
+        SparkExecutionEngineConfig.toSparkExecutionEngineConfig(json);
+
+    assertEquals("app-1", config.getApplicationId());
+    assertEquals("role-1", config.getExecutionRoleARN());
+    assertEquals("us-west-1", config.getRegion());
+    assertEquals("--conf A=1", config.getSparkSubmitParameters());
+  }
+}

spark/src/test/java/org/opensearch/sql/spark/dispatcher/SparkQueryDispatcherTest.java

@@ -574,8 +574,6 @@ void testGetQueryResponseWithSuccess() {
             dataSourceUserAuthorizationHelper,
             jobExecutionResponseReader,
             flintIndexMetadataReader);
-    when(emrServerlessClient.getJobRunResult(EMRS_APPLICATION_ID, EMR_JOB_ID))
-        .thenReturn(new GetJobRunResult().withJobRun(new JobRun().withState(JobRunState.SUCCESS)));
     JSONObject queryResult = new JSONObject();
     Map<String, Object> resultMap = new HashMap<>();
     resultMap.put(STATUS_FIELD, "SUCCESS");