WebTestClient变异过滤器java.lang.IllegalStateException:没有“ WebTestClient-Request-Id”标头

时间:2019-12-03 16:15:19

标签: java spring spring-boot spring-webflux project-reactor

我正在尝试使用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");
  }

1 个答案:

答案 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()
                        );