fix test for WorkflowTimeoutUtilityTests

Signed-off-by: Junwei Dai <[email protected]>
junweid62 · Jan 15, 2025 · 95816c9 · 95816c9
1 parent 15e052b
commit 95816c9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/test/java/org/opensearch/flowframework/util/WorkflowTimeoutUtilityTests.java b/src/test/java/org/opensearch/flowframework/util/WorkflowTimeoutUtilityTests.java
@@ -24,6 +24,7 @@
 import java.util.Collections;
 import java.util.concurrent.atomic.AtomicBoolean;
 
+import static org.opensearch.flowframework.common.CommonValue.PROVISION_WORKFLOW_THREAD_POOL;
 import static org.mockito.ArgumentMatchers.any;
 import static org.mockito.ArgumentMatchers.anyString;
 import static org.mockito.ArgumentMatchers.eq;
@@ -69,7 +70,7 @@ public void testScheduleTimeoutHandler() {
         verify(mockThreadPool, times(1)).schedule(
             any(Runnable.class),
             eq(TimeValue.timeValueMillis(timeout)),
-            eq(ThreadPool.Names.GENERIC)
+            eq(PROVISION_WORKFLOW_THREAD_POOL)
         );
     }