From 0f4e1c0e25cb0a22bea93f187bda05fc3032c105 Mon Sep 17 00:00:00 2001
From: Tomoyuki MORITA <moritato@amazon.com>
Date: Wed, 1 May 2024 09:09:20 -0700
Subject: [PATCH] Fix test failure in SparkQueryDispatcherTest due to merge
 (#2645)

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);