Fix test failure in SparkQueryDispatcherTest due to merge (opensearch…

…-project#2645) Signed-off-by: Tomoyuki Morita <[email protected]>
vmmusings · May 1, 2024 · 0f4e1c0 · 0f4e1c0
1 parent d32cf94
commit 0f4e1c0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/spark/src/test/java/org/opensearch/sql/spark/dispatcher/SparkQueryDispatcherTest.java b/spark/src/test/java/org/opensearch/sql/spark/dispatcher/SparkQueryDispatcherTest.java
@@ -172,6 +172,7 @@ void testDispatchSelectQuery() {
 
   @Test
   void testDispatchSelectQueryWithLakeFormation() {
+    when(emrServerlessClientFactory.getClient()).thenReturn(emrServerlessClient);
     HashMap<String, String> tags = new HashMap<>();
     tags.put(DATASOURCE_TAG_KEY, "my_glue");
     tags.put(CLUSTER_NAME_TAG_KEY, TEST_CLUSTER_NAME);