root
/
spring-framework
mirror of https://github.com/spring-projects/spring-framework.git
1
0
Fork 0
Code Issues Actions 8 Packages Projects Releases Wiki Activity

Fix WebClientIntegrationTests flaky Jetty tests 

Ensure the port used by the client in malformedResponseChunksOnBodilessEntity
and malformedResponseChunksOnEntityWithBody has correctly been set.

Closes gh-29862
This commit is contained in:
Sébastien Deleuze 2023-01-20 11:19:01 +01:00
parent 86a40716bd
commit fb0aa5abb3
1 changed files with 11 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
spring-webflux/src/test/java/org/springframework/web/reactive/function/client/WebClientIntegrationTests.java
View File

@ -34,7 +34,6 @@ import java.time.Duration;
import java.util.Arrays; import java.util.Arrays;
import java.util.List; import java.util.List;
import java.util.Map; import java.util.Map;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.function.Consumer; import java.util.function.Consumer;
import java.util.function.Function; import java.util.function.Function;
import java.util.stream.Collectors; import java.util.stream.Collectors;
@ -50,6 +49,7 @@ import org.junit.jupiter.params.ParameterizedTest;
import org.junit.jupiter.params.provider.MethodSource; import org.junit.jupiter.params.provider.MethodSource;
import reactor.core.publisher.Flux; import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono; import reactor.core.publisher.Mono;
import reactor.core.publisher.Sinks;
import reactor.netty.http.client.HttpClient; import reactor.netty.http.client.HttpClient;
import reactor.netty.resources.ConnectionProvider; import reactor.netty.resources.ConnectionProvider;
import reactor.test.StepVerifier; import reactor.test.StepVerifier;
@ -1281,12 +1281,13 @@ class WebClientIntegrationTests {
private <T> Mono<T> doMalformedChunkedResponseTest( private <T> Mono<T> doMalformedChunkedResponseTest(
ClientHttpConnector connector, Function<ResponseSpec, Mono<T>> handler) { ClientHttpConnector connector, Function<ResponseSpec, Mono<T>> handler) {
AtomicInteger port = new AtomicInteger(); Sinks.One<Integer> portSink = Sinks.one();
Thread serverThread = new Thread(() -> { Thread serverThread = new Thread(() -> {
// No way to simulate a malformed chunked response through MockWebServer. // No way to simulate a malformed chunked response through MockWebServer.
try (ServerSocket serverSocket = new ServerSocket(0)) { try (ServerSocket serverSocket = new ServerSocket(0)) {
port.set(serverSocket.getLocalPort()); Sinks.EmitResult result = portSink.tryEmitValue(serverSocket.getLocalPort());
assertThat(result).isEqualTo(Sinks.EmitResult.OK);
Socket socket = serverSocket.accept(); Socket socket = serverSocket.accept();
InputStream is = socket.getInputStream(); InputStream is = socket.getInputStream();
@ -1310,12 +1311,13 @@ class WebClientIntegrationTests {
serverThread.start(); serverThread.start();
return portSink.asMono().flatMap(port -> {
WebClient client = WebClient.builder() WebClient client = WebClient.builder()
.clientConnector(connector) .clientConnector(connector)
.baseUrl("http://localhost:" + port) .baseUrl("http://localhost:" + port)
.build(); .build();
return handler.apply(client.post().retrieve()); return handler.apply(client.post().retrieve());
});
} }
private void prepareResponse(Consumer<MockResponse> consumer) { private void prepareResponse(Consumer<MockResponse> consumer) {