Wait for exchange source to complete before verifying results (#121603)…

… (#121607) Closes #118732
elastic · Feb 4, 2025 · f692fc8 · f692fc8
1 parent c65b9b9
commit f692fc8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...mpute/src/test/java/org/elasticsearch/compute/operator/exchange/ExchangeServiceTests.java b/...mpute/src/test/java/org/elasticsearch/compute/operator/exchange/ExchangeServiceTests.java
@@ -436,8 +436,8 @@ public void testExchangeSourceContinueOnFailure() {
             exchangeSink
         );
         assertThat(actualSeqNos, equalTo(expectedSeqNos));
+        safeGet(sourceCompletionFuture);
         assertThat(completedSinks.get() + failedSinks.get(), equalTo(totalSinks.get()));
-        sourceCompletionFuture.actionGet();
         if (failedRequests.get() > 0) {
             assertThat(failedSinks.get(), greaterThan(0));
         } else {