Fix broken UT

Signed-off-by: Chen Dai <daichen@amazon.com>
dai-chen · Dec 1, 2023 · 3582e4f · 3582e4f
1 parent 54aef87
commit 3582e4f
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/spark/src/test/java/org/opensearch/sql/spark/dispatcher/IndexDMLHandlerTest.java b/spark/src/test/java/org/opensearch/sql/spark/dispatcher/IndexDMLHandlerTest.java
@@ -6,16 +6,19 @@
 package org.opensearch.sql.spark.dispatcher;
 
 import static org.junit.jupiter.api.Assertions.*;
+import static org.opensearch.sql.spark.data.constants.SparkConstants.ERROR_FIELD;
+import static org.opensearch.sql.spark.data.constants.SparkConstants.STATUS_FIELD;
 
+import org.json.JSONObject;
 import org.junit.jupiter.api.Test;
 
 class IndexDMLHandlerTest {
   @Test
   public void getResponseFromExecutor() {
-    assertThrows(
-        IllegalStateException.class,
-        () ->
-            new IndexDMLHandler(null, null, null, null, null, null, null)
-                .getResponseFromExecutor(null));
+    JSONObject result =
+        new IndexDMLHandler(null, null, null, null, null, null, null).getResponseFromExecutor(null);
+
+    assertEquals("running", result.getString(STATUS_FIELD));
+    assertEquals("", result.getString(ERROR_FIELD));
   }
 }