Skip to content

Commit fcf25ff

Browse files
committed
Fix SearchableSnapshotsPrewarmingIntegTests (Backport #101203)
Closes #112422
1 parent 6e057bd commit fcf25ff

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

x-pack/plugin/searchable-snapshots/src/internalClusterTest/java/org/elasticsearch/xpack/searchablesnapshots/cache/full/SearchableSnapshotsPrewarmingIntegTests.java

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -272,11 +272,13 @@ public void testConcurrentPrewarming() throws Exception {
272272
final CountDownLatch startPrewarmingLatch = new CountDownLatch(1);
273273
final ThreadPool threadPool = getInstanceFromNode(ThreadPool.class);
274274
final int maxUploadTasks = threadPool.info(CACHE_PREWARMING_THREAD_POOL_NAME).getMax();
275+
final CountDownLatch maxUploadTasksCreated = new CountDownLatch(maxUploadTasks);
275276
for (int i = 0; i < maxUploadTasks; i++) {
276277
threadPool.executor(CACHE_PREWARMING_THREAD_POOL_NAME).execute(new AbstractRunnable() {
277278

278279
@Override
279280
protected void doRun() throws Exception {
281+
maxUploadTasksCreated.countDown();
280282
startPrewarmingLatch.await();
281283
}
282284

@@ -287,6 +289,7 @@ public void onFailure(Exception e) {
287289
});
288290
}
289291

292+
safeAwait(maxUploadTasksCreated);
290293
ExecutorService prewarmingExecutor = threadPool.executor(SearchableSnapshots.CACHE_PREWARMING_THREAD_POOL_NAME);
291294
assertThat(prewarmingExecutor, instanceOf(ThreadPoolExecutor.class));
292295
assertThat(((ThreadPoolExecutor) prewarmingExecutor).getActiveCount(), equalTo(maxUploadTasks));

0 commit comments

Comments
 (0)