Update bulkhead test (eugenp#13836)

862569222 · Apr 24, 2023 · 91d1cdf · 91d1cdf
1 parent 113c62e
commit 91d1cdf
Showing 1 changed file with 6 additions and 10 deletions.
diff --git a/.../java/com/baeldung/resilience4j/eventendpoints/ResilientAppControllerIntegrationTest.java b/.../java/com/baeldung/resilience4j/eventendpoints/ResilientAppControllerIntegrationTest.java
@@ -14,7 +14,6 @@
 import com.github.tomakehurst.wiremock.junit5.WireMockExtension;
 import java.net.URI;
 import java.nio.charset.Charset;
-import java.util.ArrayList;
 import java.util.List;
 import java.util.Map;
 import java.util.concurrent.*;
@@ -210,23 +209,20 @@ void testBulkheadEvents() throws Exception {
     EXTERNAL_SERVICE.stubFor(WireMock.get("/api/external").willReturn(ok()));
     Map<Integer, Integer> responseStatusCount = new ConcurrentHashMap<>();
     ExecutorService executorService = Executors.newFixedThreadPool(5);
+    CountDownLatch latch = new CountDownLatch(5);
 
-    List<Callable<Integer>> tasks = new ArrayList<>();
     IntStream.rangeClosed(1, 5)
         .forEach(
             i ->
-                tasks.add(
+                executorService.execute(
                     () -> {
                       ResponseEntity<String> response =
                           restTemplate.getForEntity("/api/bulkhead", String.class);
-                      return response.getStatusCodeValue();
+                      int statusCode = response.getStatusCodeValue();
+                      responseStatusCount.merge(statusCode, 1, Integer::sum);
+                      latch.countDown();
                     }));
-
-    List<Future<Integer>> futures = executorService.invokeAll(tasks);
-    for (Future<Integer> future : futures) {
-      int statusCode = future.get();
-      responseStatusCount.merge(statusCode, 1, Integer::sum);
-    }
+    latch.await();
     executorService.shutdown();
 
     assertEquals(2, responseStatusCount.keySet().size());