WebTestClient变异筛选器java.lang.IllegalStateException:没有“WebTestClient-请求-Id”标题

Question

我试图使用ExchangeFilterFunctions编写WebTestClient的单元测试，但是在对webTestClient进行变异和添加筛选器时会出现此错误。

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

添加以下测试中的注释行并导致上述错误：

  @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");
  }

AddHeaderExchangeFilter

public class AddHeaderExchangeFilter implements ExchangeFilterFunction {

  @Override
  public Mono<ClientResponse> filter(ClientRequest request, ExchangeFunction next) {

    ClientRequest newRequest = ClientRequest
            .create(request.method(), request.url())
            .header("FILTER-HEADER-KEY", "FILTER-HEADER-VALUE")
            .build();

    return next.exchange(newRequest);
  }

}

Answer 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()
                        );