Apply attributes to the underlying HTTP request

See gh-29958
This commit is contained in:
PhilKes
2023-02-11 14:12:18 +01:00
committed by rstoyanchev
parent 282ee02419
commit 052b6357c8
26 changed files with 374 additions and 24 deletions

View File

@@ -18,6 +18,7 @@ package org.springframework.test.web.reactive.server;
import java.net.URI;
import java.time.Duration;
import java.util.function.Function;
import org.junit.jupiter.api.Test;
import reactor.core.publisher.Mono;
@@ -48,7 +49,22 @@ public class WiretapConnectorTests {
public void captureAndClaim() {
ClientHttpRequest request = new MockClientHttpRequest(HttpMethod.GET, "/test");
ClientHttpResponse response = new MockClientHttpResponse(HttpStatus.OK);
ClientHttpConnector connector = (method, uri, fn) -> fn.apply(request).then(Mono.just(response));
ClientHttpConnector connector = new ClientHttpConnector() {
@Override
public Mono<ClientHttpResponse> connect(HttpMethod method, URI uri, Function<? super ClientHttpRequest, Mono<Void>> requestCallback) {
return requestCallback.apply(request).then(Mono.just(response));
}
@Override
public void setApplyAttributes(boolean applyAttributes) {
}
@Override
public boolean getApplyAttributes() {
return false;
}
};
ClientRequest clientRequest = ClientRequest.create(HttpMethod.GET, URI.create("/test"))
.header(WebTestClient.WEBTESTCLIENT_REQUEST_ID, "1").build();