Spring RequestContextHolder和WebTestClient

Question

我在控制器中使用了SpringRequestContextHolder，它工作得很好。但是在单元测试中，我使用WebTestClient获得java.lang.IllegalStateException。这里有一个例子：

package demo.reactive.controller;

import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.context.request.RequestContextHolder;
import reactor.core.publisher.Mono;

@RestController
public class FooController {

  @GetMapping("/foo")
  public ResponseEntity<Mono<String>> foo() {

    String sessionId = RequestContextHolder.currentRequestAttributes().getSessionId();

    return ResponseEntity.ok(Mono.just(sessionId));
  }
}

package demo.reactive;

import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.autoconfigure.web.reactive.WebFluxTest;
import org.springframework.test.web.reactive.server.WebTestClient;
import demo.reactive.controller.FooController;

@WebFluxTest(FooController.class)
class DemoReactiveApplicationTests {

  @Autowired private WebTestClient client;

  @Test
  void contextLoads() {
    client.get().uri("/foo").exchange().expectStatus().isOk();
  }
}

如何使单元测试工作。或者甚至不需要在WebFlux中使用RequestContextHolder？

Answer 1

SpringRequestContextHolder不能与Spring WebFlux一起使用。RequestContextHolderprimary使用java.lang.ThreadLocal工作。这显然不适合反应堆结构。

话虽如此，我们可以使用Reactor的Mono#subscriberContext()运算符实现一个替代方案。创建可重用解决方案的最佳方法是实现springwebfilter以获取请求属性并将其设置在反应器上下文中。

public class WebRequestAttributesContextFilter implements WebFilter {

  public static final String WEB_REQUEST_ATTRIBUTES = "WebRequestAttributes";

  @Override
  public Mono<Void> filter(ServerWebExchange serverWebExchange, WebFilterChain webFilterChain) {
    return webFilterChain.filter(serverWebExchange)
        .subscriberContext(context -> context.put(WEB_REQUEST_ATTRIBUTES, serverWebExchange.getAttributes()));
  }
}

然后可以从reactor上下文中的请求处理管道中的任何位置读取属性（在您的情况中为sessionId)，如下所示：

@GetMapping("/foo")
public Mono<String> helloWorld() {
  return Mono.subscriberContext().map(context -> (String) context.<Map<String, Object>>get(
          WebRequestAttributesContextFilter.WEB_REQUEST_ATTRIBUTES
      ).get("session"));
}