Fix UT

Signed-off-by: Louis Chu <[email protected]>
opensearch-project · Jul 24, 2024 · c59f70b · c59f70b
1 parent 81c4c93
commit c59f70b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/...y/src/test/java/org/opensearch/sql/spark/scheduler/OpenSearchAsyncQuerySchedulerTest.java b/...y/src/test/java/org/opensearch/sql/spark/scheduler/OpenSearchAsyncQuerySchedulerTest.java
@@ -199,7 +199,7 @@ public void testUnscheduleJob() throws IOException {
   public void testUnscheduleJobWithIndexNotFound() {
     when(clusterService.state().routingTable().hasIndex(SCHEDULER_INDEX_NAME)).thenReturn(false);
 
-    assertThrows(IllegalArgumentException.class, () -> scheduler.unscheduleJob(TEST_JOB_ID));
+    assertThrows(IllegalStateException.class, () -> scheduler.unscheduleJob(TEST_JOB_ID));
   }
 
   @Test
@@ -237,7 +237,7 @@ public void testUpdateJobWithIndexNotFound() {
 
     when(clusterService.state().routingTable().hasIndex(SCHEDULER_INDEX_NAME)).thenReturn(false);
 
-    assertThrows(IllegalArgumentException.class, () -> scheduler.updateJob(request));
+    assertThrows(IllegalStateException.class, () -> scheduler.updateJob(request));
   }
 
   @Test
@@ -305,7 +305,7 @@ public void testRemoveJob() {
   public void testRemoveJobWithIndexNotFound() {
     when(clusterService.state().routingTable().hasIndex(SCHEDULER_INDEX_NAME)).thenReturn(false);
 
-    assertThrows(IllegalArgumentException.class, () -> scheduler.removeJob(TEST_JOB_ID));
+    assertThrows(IllegalStateException.class, () -> scheduler.removeJob(TEST_JOB_ID));
   }
 
   @Test