如何 Mock Java WebClient 以进行单元测试

本文旨在解决 Java WebClient 单元测试中 Mock 不生效的问题。通过分析常见错误原因，提供了一种通过可 Mock 方法构建 WebClient 的解决方案，并指出了 Mockito eq() 方法使用的必要性，帮助开发者编写可靠的 WebClient 单元测试。

在进行 Java WebClient 的单元测试时，Mock WebClient 是一个常见的需求。但有时 Mock 却无法生效，导致测试结果与预期不符。本文将分析可能的原因，并提供解决方案，帮助你正确地 Mock WebClient，编写可靠的单元测试。

问题分析：WebClient 实例未被 Mock 替换

最常见的问题是 getResponse 方法中使用的 WebClient 实例并非测试用例中 Mock 的 webClientMock。 如果你在 getResponse 方法内部直接创建 WebClient 实例，Mock 的 webClientMock 就不会被使用到。

立即学习“Java免费学习笔记（深入）”；

解决方案：通过可 Mock 方法构建 WebClient

为了解决这个问题，可以将 WebClient 的构建过程提取到一个单独的方法中，并使用 Mockito 桩截 (stubbing) 该方法。

1. 创建 WebClient 构建方法：

   在你的 Service 类中，创建一个用于构建 WebClient 实例的方法。建议添加 @VisibleForTesting 注解，明确该方法是为了方便测试而存在。

   

   Krikey AI

   68

   查看详情

   import org.springframework.web.reactive.function.client.WebClient;
   import com.google.common.annotations.VisibleForTesting;
   
   public class SomeService {
   
       @VisibleForTesting
       WebClient buildWebClient() {
           // 使用 WebClientBuilder 构建 WebClient
           return WebClient.builder().baseUrl("your_base_url").build();
       }
   
       public String getResponse(String requestBody){
           WebClient client = buildWebClient(); // 使用构建方法获取 WebClient 实例
           WebClient.RequestHeadersSpec<?> request =
               client.post().body(BodyInserters.fromValue(requestBody));
   
           String resp =
               request.retrieve().bodyToMono(String.class)
                   .doOnError(
                       WebClientResponseException.class,
                       err -> {
                         // do something
                       })
                   .block();
   
           return resp;
       }
   }

   登录后复制

2. 在测试用例中 Mock 构建方法：

   使用 Mockito.doReturn() 方法，桩截 someServiceSpy 的 buildWebClient() 方法，使其返回 Mock 的 webClientMock 实例。

   import org.junit.jupiter.api.Test;
   import org.mockito.Mockito;
   import org.springframework.web.reactive.function.BodyInserters;
   import org.springframework.web.reactive.function.client.WebClient;
   import reactor.core.publisher.Mono;
   
   import static org.junit.jupiter.api.Assertions.assertEquals;
   import static org.mockito.Mockito.when;
   
   public class SomeServiceTest {
   
       @Test
       public void testGetResponse() {
           // 创建 Mock 对象
           WebClient webClientMock = Mockito.mock(WebClient.class);
           WebClient.RequestBodyUriSpec requestBodyUriMock = Mockito.mock(WebClient.RequestBodyUriSpec.class);
           WebClient.RequestHeadersSpec requestHeadersMock = Mockito.mock(WebClient.RequestHeadersSpec.class);
           WebClient.ResponseSpec responseMock = Mockito.mock(WebClient.ResponseSpec.class);
   
           String requestBody = "your_request_body";
   
           SomeService someService = new SomeService();
           SomeService someServiceSpy = Mockito.spy(someService); // 创建 Spy 对象
   
           // 桩截 buildWebClient() 方法，返回 Mock 的 webClientMock
           Mockito.doReturn(webClientMock).when(someServiceSpy).buildWebClient();
   
           when(webClientMock.post()).thenReturn(requestBodyUriMock);
           when(requestBodyUriMock.body(BodyInserters.fromValue(requestBody))).thenReturn(requestHeadersMock);
           when(requestHeadersMock.retrieve()).thenReturn(responseMock);
           when(responseMock.bodyToMono(String.class)).thenReturn(Mono.just("response"));
   
           String response = someServiceSpy.getResponse(requestBody);
   
           assertEquals("response", response);
       }
   }

   登录后复制

注意事项：Mockito eq() 方法的使用

在 Mockito 中，当 Mock 方法的参数是复杂对象时，需要使用 eq() 方法进行精确匹配。否则，即使参数的值相同，Mockito 也可能认为参数不匹配，导致 Mock 失效。

因此，将以下代码：

when(requestBodyUriMock.body(BodyInserters.fromValue(requestBody))).thenReturn(requestHeadersMock);

修改为：

import static org.mockito.ArgumentMatchers.eq;

when(requestBodyUriMock.body(eq(BodyInserters.fromValue(requestBody)))).thenReturn(requestHeadersMock);

总结

通过将 WebClient 的构建过程提取到可 Mock 的方法中，并使用 Mockito.doReturn() 桩截该方法，可以确保在单元测试中使用 Mock 的 WebClient 实例。同时，对于复杂参数的 Mock，务必使用 eq() 方法进行精确匹配，以保证 Mock 的正确性。遵循这些原则，你就能编写出更可靠的 Java WebClient 单元测试。

以上就是如何 Mock Java WebClient 以进行单元测试的详细内容，更多请关注php中文网其它相关文章！