Fix test execution.

Author: afoucret

x-pack/plugin/esql/src/test/java/org/elasticsearch/xpack/esql/inference/bulk/BulkInferenceExecutorTests.java

@@ -27,6 +27,7 @@
 import java.util.Iterator;
 import java.util.List;
 import java.util.concurrent.CountDownLatch;
+import java.util.concurrent.TimeUnit;
 import java.util.concurrent.atomic.AtomicReference;
 
 import static org.hamcrest.Matchers.allOf;
@@ -141,7 +142,7 @@ public void testInferenceRunnerSometimesFails() throws Exception {
             assertThat(exception.get().getMessage(), equalTo("inference failure"));
         });
     }
-    
+
     public void testParallelBulkExecution() throws Exception {
         int batches = between(50, 100);
         CountDownLatch latch = new CountDownLatch(batches);
@@ -168,7 +169,7 @@ public void testParallelBulkExecution() throws Exception {
             });
         }
 
-        latch.await();
+        latch.await(10, TimeUnit.SECONDS);
     }
 
     private BulkInferenceExecutor bulkExecutor(InferenceRunner inferenceRunner) {
@@ -225,11 +226,7 @@ private void runWithRandomDelay(Runnable runnable) {
         if (randomBoolean()) {
             runnable.run();
         } else {
-            threadPool.schedule(
-                runnable,
-                TimeValue.timeValueNanos(between(1, 1_000)),
-                threadPool.executor(EsqlPlugin.ESQL_WORKER_THREAD_POOL_NAME)
-            );
+            threadPool.schedule(runnable, TimeValue.timeValueNanos(between(1, 1_000)), threadPool.generic());
         }
     }
 }