我正在尝试使用WebTestClient编写ExchangeFilterFunctions的单元测试,但是在更改并将过滤器添加到webTestClient时遇到此错误。
values.yaml
添加了以下测试中的注释行,并导致上述错误:
java.lang.IllegalStateException: No "WebTestClient-Request-Id" header
at org.springframework.util.Assert.state(Assert.java:94)
at org.springframework.test.web.reactive.server.WiretapConnector.lambda$connect$2(WiretapConnector.java:80)
at reactor.core.publisher.FluxMap$MapSubscriber.onNext(FluxMap.java:100)
at reactor.core.publisher.Operators$MonoSubscriber.complete(Operators.java:1510)
at reactor.core.publisher.MonoProcessor.subscribe(MonoProcessor.java:457)
at reactor.core.publisher.MonoMap.subscribe(MonoMap.java:55)
at reactor.core.publisher.MonoPeek.subscribe(MonoPeek.java:71)
at reactor.core.publisher.MonoPeek.subscribe(MonoPeek.java:71)
at reactor.core.publisher.MonoMap.subscribe(MonoMap.java:55)
at reactor.core.publisher.MonoSwitchIfEmpty.subscribe(MonoSwitchIfEmpty.java:44)
at reactor.core.publisher.Mono.block(Mono.java:1517)
at org.springframework.test.web.reactive.server.DefaultWebTestClient$DefaultRequestBodyUriSpec.exchange(DefaultWebTestClient.java:283)
at com.ecs.springframework.reactive.web.filter.EcsMdcRequestExchangeFilterTest.whenExchangeFilterFunctionInjectedIntoWebTestClient_thenWebTestClientShouldIncludeHeader
AddHeaderExchangeFilter
@Test
void whenExchangeFilterFunctionInjectedIntoWebTestClient_thenWebTestClientShouldIncludeHeader() {
RouterFunction function =
RouterFunctions.route(
RequestPredicates.GET("/filter"), request -> ServerResponse.ok().build());
WebTestClient.bindToRouterFunction(function)
.build()
// .mutate()
// .filter(new AddHeaderExchangeFilter())
// .build()
.get()
.uri("/filter")
.exchange()
.expectStatus()
.isOk()
.expectHeader()
.valueMatches("FILTER-HEADER-KEY", "FILTER-HEADER-VALUE");
}
答案 0 :(得分:1)
您需要在AddHeaderExchangeFilter
上添加当前请求标头
public class AddHeaderExchangeFilter implements ExchangeFilterFunction {
@Override
public Mono<ClientResponse> filter(ClientRequest request, ExchangeFunction next) {
ClientRequest newRequest = ClientRequest.create(request.method(), request.url())
.headers(httpHeaders -> httpHeaders.addAll(request.headers()))
.header("FILTER-HEADER-KEY", "FILTER-HEADER-VALUE")
.build();
return next.exchange(newRequest);
}
}
这对于WebTestClient
正确的行为是必需的,并且在执行exchange()
时对其进行验证。
在过滤器中,您正在创建一个新请求,初始请求的标头将被忽略,并且不会传递给newRequest
。
旁注::我猜您尝试测试过滤器是否正常运行,但是您当前的测试不是验证请求的标头,而是验证响应的标头(因此您的测试将失败)。
我对您的建议是更改路由器以从请求中返回标头
RouterFunction function =
RouterFunctions.route(
RequestPredicates.GET("/filter"), request ->
ServerResponse.ok().headers(httpHeaders -> httpHeaders.addAll(request.headers().asHttpHeaders())).build()
);