From fd2da010498426e35c484970cdd15d1a6100e877 Mon Sep 17 00:00:00 2001
From: Tomoyuki Morita <moritato@amazon.com>
Date: Tue, 30 Apr 2024 16:54:01 -0700
Subject: [PATCH] Fix test failure in SparkQueryDispatcherTest due to merge

Signed-off-by: Tomoyuki Morita <moritato@amazon.com>
---
 .../sql/spark/dispatcher/SparkQueryDispatcherTest.java           | 1 +
 1 file changed, 1 insertion(+)

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
index ec87a86717..f668765160 100644
--- 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);