spotlessApply

Signed-off-by: Sean Kao <[email protected]>
opensearch-project · Jun 10, 2024 · 4e05e17 · 4e05e17
1 parent c343d71
commit 4e05e17
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 19 deletions.
diff --git a/spark/src/main/java/org/opensearch/sql/spark/dispatcher/BatchQueryHandler.java b/spark/src/main/java/org/opensearch/sql/spark/dispatcher/BatchQueryHandler.java
@@ -25,7 +25,6 @@
 import org.opensearch.sql.spark.dispatcher.model.DispatchQueryResponse;
 import org.opensearch.sql.spark.dispatcher.model.JobType;
 import org.opensearch.sql.spark.leasemanager.LeaseManager;
-import org.opensearch.sql.spark.leasemanager.model.LeaseRequest;
 import org.opensearch.sql.spark.response.JobExecutionResponseReader;
 
 /**

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
@@ -806,12 +806,13 @@ void testDispatchAlterToManualRefreshIndexQuery() {
     DataSourceMetadata dataSourceMetadata = constructMyGlueDataSourceMetadata();
     when(dataSourceService.verifyDataSourceAccessAndGetRawMetadata("my_glue"))
         .thenReturn(dataSourceMetadata);
-    when(queryHandlerFactory.getIndexDMLHandler()).thenReturn(
-        new IndexDMLHandler(
-            jobExecutionResponseReader,
-            flintIndexMetadataService,
-            indexDMLResultStorageService,
-            flintIndexOpFactory));
+    when(queryHandlerFactory.getIndexDMLHandler())
+        .thenReturn(
+            new IndexDMLHandler(
+                jobExecutionResponseReader,
+                flintIndexMetadataService,
+                indexDMLResultStorageService,
+                flintIndexOpFactory));
 
     sparkQueryDispatcher.dispatch(getBaseDispatchQueryRequest(query), asyncQueryRequestContext);
     verify(queryHandlerFactory, times(1)).getIndexDMLHandler();
@@ -828,12 +829,13 @@ void testDispatchDropIndexQuery() {
     DataSourceMetadata dataSourceMetadata = constructMyGlueDataSourceMetadata();
     when(dataSourceService.verifyDataSourceAccessAndGetRawMetadata("my_glue"))
         .thenReturn(dataSourceMetadata);
-    when(queryHandlerFactory.getIndexDMLHandler()).thenReturn(
-        new IndexDMLHandler(
-            jobExecutionResponseReader,
-            flintIndexMetadataService,
-            indexDMLResultStorageService,
-            flintIndexOpFactory));
+    when(queryHandlerFactory.getIndexDMLHandler())
+        .thenReturn(
+            new IndexDMLHandler(
+                jobExecutionResponseReader,
+                flintIndexMetadataService,
+                indexDMLResultStorageService,
+                flintIndexOpFactory));
 
     sparkQueryDispatcher.dispatch(getBaseDispatchQueryRequest(query), asyncQueryRequestContext);
     verify(queryHandlerFactory, times(1)).getIndexDMLHandler();
@@ -850,12 +852,13 @@ void testDispatchVacuumIndexQuery() {
     DataSourceMetadata dataSourceMetadata = constructMyGlueDataSourceMetadata();
     when(dataSourceService.verifyDataSourceAccessAndGetRawMetadata("my_glue"))
         .thenReturn(dataSourceMetadata);
-    when(queryHandlerFactory.getIndexDMLHandler()).thenReturn(
-        new IndexDMLHandler(
-            jobExecutionResponseReader,
-            flintIndexMetadataService,
-            indexDMLResultStorageService,
-            flintIndexOpFactory));
+    when(queryHandlerFactory.getIndexDMLHandler())
+        .thenReturn(
+            new IndexDMLHandler(
+                jobExecutionResponseReader,
+                flintIndexMetadataService,
+                indexDMLResultStorageService,
+                flintIndexOpFactory));
 
     sparkQueryDispatcher.dispatch(getBaseDispatchQueryRequest(query), asyncQueryRequestContext);
     verify(queryHandlerFactory, times(1)).getIndexDMLHandler();