Fix CI

qianheng-aws · qianheng-aws · commit 303b53abfe0e · 2026-01-13T17:58:59.000+08:00
Signed-off-by: Heng Qian &lt;qianheng@amazon.com&gt;
diff --git a/core/src/test/java/org/opensearch/sql/executor/execution/QueryPlanTest.java b/core/src/test/java/org/opensearch/sql/executor/execution/QueryPlanTest.java
@@ -20,7 +20,7 @@
 import org.junit.jupiter.api.extension.ExtendWith;
 import org.mockito.Mock;
 import org.mockito.junit.jupiter.MockitoExtension;
-import org.opensearch.sql.ast.statement.Explain;
+import org.opensearch.sql.ast.statement.ExplainMode;
 import org.opensearch.sql.ast.tree.UnresolvedPlan;
 import org.opensearch.sql.common.response.ResponseListener;
 import org.opensearch.sql.executor.DefaultExecutionEngine;
@@ -45,7 +45,7 @@ class QueryPlanTest {
 
   @Mock private ResponseListener<ExecutionEngine.QueryResponse> queryListener;
 
-  @Mock private Explain.ExplainFormat format;
+  @Mock private ExplainMode mode;
 
   @Test
   public void execute_no_page_size() {
@@ -58,9 +58,9 @@ public void execute_no_page_size() {
   @Test
   public void explain_no_page_size() {
     QueryPlan query = new QueryPlan(queryId, queryType, plan, queryService, queryListener);
-    query.explain(explainListener, format);
+    query.explain(explainListener, mode);
 
-    verify(queryService, times(1)).explain(plan, queryType, explainListener, format);
+    verify(queryService, times(1)).explain(plan, queryType, explainListener, mode);
   }
 
   @Test
@@ -124,6 +124,6 @@ public void onFailure(Exception e) {
                 assertTrue(e instanceof NotImplementedException);
               }
             },
-            format);
+            mode);
   }
 }
