Update bulkhead test (#13836)

This commit is contained in:
Adrian Bob 2023-04-24 20:32:49 +03:00 committed by GitHub
parent 90f2d165bf
commit c11605cc61
1 changed files with 6 additions and 10 deletions

View File

@ -14,7 +14,6 @@ import com.github.tomakehurst.wiremock.core.WireMockConfiguration;
import com.github.tomakehurst.wiremock.junit5.WireMockExtension; import com.github.tomakehurst.wiremock.junit5.WireMockExtension;
import java.net.URI; import java.net.URI;
import java.nio.charset.Charset; import java.nio.charset.Charset;
import java.util.ArrayList;
import java.util.List; import java.util.List;
import java.util.Map; import java.util.Map;
import java.util.concurrent.*; import java.util.concurrent.*;
@ -210,23 +209,20 @@ class ResilientAppControllerIntegrationTest {
EXTERNAL_SERVICE.stubFor(WireMock.get("/api/external").willReturn(ok())); EXTERNAL_SERVICE.stubFor(WireMock.get("/api/external").willReturn(ok()));
Map<Integer, Integer> responseStatusCount = new ConcurrentHashMap<>(); Map<Integer, Integer> responseStatusCount = new ConcurrentHashMap<>();
ExecutorService executorService = Executors.newFixedThreadPool(5); ExecutorService executorService = Executors.newFixedThreadPool(5);
CountDownLatch latch = new CountDownLatch(5);
List<Callable<Integer>> tasks = new ArrayList<>();
IntStream.rangeClosed(1, 5) IntStream.rangeClosed(1, 5)
.forEach( .forEach(
i -> i ->
tasks.add( executorService.execute(
() -> { () -> {
ResponseEntity<String> response = ResponseEntity<String> response =
restTemplate.getForEntity("/api/bulkhead", String.class); restTemplate.getForEntity("/api/bulkhead", String.class);
return response.getStatusCodeValue(); int statusCode = response.getStatusCodeValue();
responseStatusCount.merge(statusCode, 1, Integer::sum);
latch.countDown();
})); }));
latch.await();
List<Future<Integer>> futures = executorService.invokeAll(tasks);
for (Future<Integer> future : futures) {
int statusCode = future.get();
responseStatusCount.merge(statusCode, 1, Integer::sum);
}
executorService.shutdown(); executorService.shutdown();
assertEquals(2, responseStatusCount.keySet().size()); assertEquals(2, responseStatusCount.keySet().size());