Remove unused code and test

Signed-off-by: Chen Dai <[email protected]>

Author: dai-chen

async-query-core/src/main/java/org/opensearch/sql/spark/dispatcher/IndexDMLHandler.java

@@ -138,8 +138,6 @@ private FlintIndexOp getIndexOp(
       case ALTER:
         return flintIndexOpFactory.getAlter(
             indexQueryDetails.getFlintIndexOptions(), dispatchQueryRequest.getDatasource());
-      case VACUUM:
-        return flintIndexOpFactory.getVacuum(dispatchQueryRequest.getDatasource());
       default:
         throw new IllegalStateException(
             String.format(

async-query-core/src/main/java/org/opensearch/sql/spark/flint/operation/FlintIndexOpFactory.java

@@ -36,15 +36,6 @@ public FlintIndexOpAlter getAlter(FlintIndexOptions flintIndexOptions, String da
         asyncQueryScheduler);
   }
 
-  public FlintIndexOpVacuum getVacuum(String datasource) {
-    return new FlintIndexOpVacuum(
-        flintIndexStateModelService,
-        datasource,
-        flintIndexClient,
-        emrServerlessClientFactory,
-        asyncQueryScheduler);
-  }
-
   public FlintIndexOpCancel getCancel(String datasource) {
     return new FlintIndexOpCancel(
         flintIndexStateModelService, datasource, emrServerlessClientFactory);

async-query-core/src/main/java/org/opensearch/sql/spark/flint/operation/FlintIndexOpVacuum.java

@@ -236,32 +236,12 @@ public void createDropIndexQueryWithScheduler() {
   public void createVacuumIndexQuery() {
     givenSparkExecutionEngineConfigIsSupplied();
     givenValidDataSourceMetadataExist();
+    givenSessionExists();
     when(queryIdProvider.getQueryId(any(), eq(asyncQueryRequestContext))).thenReturn(QUERY_ID);
-    String indexName = "flint_datasource_name_table_name_index_name_index";
-    givenFlintIndexMetadataExists(indexName);
-
-    CreateAsyncQueryResponse response =
-        asyncQueryExecutorService.createAsyncQuery(
-            new CreateAsyncQueryRequest(
-                "VACUUM INDEX index_name ON table_name", DATASOURCE_NAME, LangType.SQL),
-            asyncQueryRequestContext);
-
-    assertEquals(QUERY_ID, response.getQueryId());
-    assertNull(response.getSessionId());
-    verifyGetQueryIdCalled();
-    verify(flintIndexClient).deleteIndex(indexName);
-    verifyCreateIndexDMLResultCalled();
-    verifyStoreJobMetadataCalled(DML_QUERY_JOB_ID, JobType.BATCH);
-  }
-
-  @Test
-  public void createVacuumIndexQueryWithScheduler() {
-    givenSparkExecutionEngineConfigIsSupplied();
-    givenValidDataSourceMetadataExist();
-    when(queryIdProvider.getQueryId(any(), eq(asyncQueryRequestContext))).thenReturn(QUERY_ID);
-
-    String indexName = "flint_datasource_name_table_name_index_name_index";
-    givenFlintIndexMetadataExistsWithExternalScheduler(indexName);
+    when(sessionIdProvider.getSessionId(any())).thenReturn(SESSION_ID);
+    givenSessionExists(); // called twice
+    when(awsemrServerless.startJobRun(any()))
+        .thenReturn(new StartJobRunResult().withApplicationId(APPLICATION_ID).withJobRunId(JOB_ID));
 
     CreateAsyncQueryResponse response =
         asyncQueryExecutorService.createAsyncQuery(
@@ -270,14 +250,12 @@ public void createVacuumIndexQueryWithScheduler() {
             asyncQueryRequestContext);
 
     assertEquals(QUERY_ID, response.getQueryId());
-    assertNull(response.getSessionId());
+    assertEquals(SESSION_ID, response.getSessionId());
     verifyGetQueryIdCalled();
-
-    verify(flintIndexClient).deleteIndex(indexName);
-    verifyCreateIndexDMLResultCalled();
-    verifyStoreJobMetadataCalled(DML_QUERY_JOB_ID, JobType.BATCH);
-
-    verify(asyncQueryScheduler).removeJob(indexName);
+    verifyGetSessionIdCalled();
+    verify(leaseManager).borrow(any());
+    verifyStartJobRunCalled();
+    verifyStoreJobMetadataCalled(JOB_ID, JobType.INTERACTIVE);
   }
 
   @Test

async-query-core/src/test/java/org/opensearch/sql/spark/asyncquery/AsyncQueryCoreIntegTest.java

@@ -41,11 +41,6 @@ void getAlter() {
     assertNotNull(flintIndexOpFactory.getAlter(new FlintIndexOptions(), DATASOURCE_NAME));
   }
 
-  @Test
-  void getVacuum() {
-    assertNotNull(flintIndexOpFactory.getDrop(DATASOURCE_NAME));
-  }
-
   @Test
   void getCancel() {
     assertNotNull(flintIndexOpFactory.getDrop(DATASOURCE_NAME));