单元测试Spring WebClient Api调用

Question

我使用Spring Web模块编写RESTAPI，使用WebClient类调用外部API，并使用block（）方法等待响应。

我知道我应该使用Spring WebFlux以完全非阻塞的方式编写RESTAPI。但这不是一个选择。

我在编写单元测试和模拟WebClient时遇到问题。我在Stack Overflow中读到了一些WebClient模拟线程，但在我的案例中似乎没有一个能正常工作。

以下是我的代码

ResponseEntity<String> response = null;
        
        try {
            response = webClient
                    .post()
                    .uri(uriBuilder())
                    .bodyValue(request)
                    .retrieve()
                    .toEntity(String.class)
                    .block();
           } catch (WebClientResponseException e) {
 
            // Process different types of exceptions here

           if (e.getStatusCode() == HttpStatus.NOT_FOUND) {
           // process HttpStatus.NOT_FOUND exception here
           }

         }

谁能告诉我如何对这段代码进行单元测试吗？

请求类是一个POJO类。uriBuilded（）方法返回一个uri。

Answer 1

模仿WebClient可能非常棘手，而且容易出错。即使您可以涵盖一些简单的积极案例，也很难测试更复杂的场景，例如错误处理、重试。。。

我会推荐WireMock，它为测试Web客户端提供了一个非常好的API。这里有一些例子

stubFor(get("/api")
        .willReturn(aResponse()
                .withHeader("Content-Type", MediaType.APPLICATION_JSON_VALUE)
                .withStatus(200)
                .withBody(...)
        )
);

您可以通过提供不同的存根轻松测试积极和消极的场景

stubFor(get("/api")
        .willReturn(aResponse()
                .withHeader("Content-Type", MediaType.APPLICATION_JSON_VALUE)
                .withStatus(500)
        )
);

或使用场景测试重试逻辑，甚至使用延迟模拟超时。

此外，我建议在这个级别上避免使用try-catch，使用反应式操作符处理错误