本文旨在帮助开发者解决在使用 WebClient 进行单元测试时，Mock 对象未生效的问题。通过分析常见的错误配置和提供可行的解决方案，确保 WebClient 的 Mock 能够正确地模拟外部服务，从而有效地进行单元测试。本文将重点介绍如何正确地注入 Mock WebClient 对象，并修正 Mockito 参数匹配的错误，从而保证单元测试的准确性和可靠性。
在进行 WebClient 单元测试时，一个常见的问题是 Mock 对象没有按照预期工作，导致测试结果不符合预期。 这种情况通常是由于 WebClient 实例没有被正确地替换为 Mock 对象，或者 Mockito 的参数匹配不正确导致的。
确保 WebClient 使用 Mock 对象
最常见的问题是 getResponse 方法中使用的 WebClient 实例不是测试中创建的 Mock 对象 webClientMock。 如果 WebClient 对象是在 getResponse 方法内部创建的，那么 Mockito 就无法直接替换它。
解决这个问题的方法是将 WebClient 的创建过程提取到一个可 Mock 的方法中。例如：
public String getResponse(String requestBody){
WebClient client = buildWebClient();
...
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;
}
@VisibleForTesting
WebClient buildWebClient() {
// build your webclient using the WebClientBuilder
return WebClient.builder().baseUrl("your_base_url").build();
}然后，在单元测试中，可以使用 Mockito.doReturn 来 Mock 这个方法，使其返回 Mock 的 WebClient 对象：
Mockito.doReturn(webClientMock).when(someServiceSpy).buildWebClient();
这样可以确保 getResponse 方法中使用的是 Mock 的 WebClient 对象。
修正 Mockito 参数匹配
另一个需要注意的问题是 Mockito 的参数匹配。在使用 when 方法时，需要确保参数匹配是精确的。对于 BodyInserters.fromValue(requestBody) 这样的复杂对象，不能直接使用 equals 方法进行匹配，而应该使用 eq 方法。
正确的 Mockito 代码如下：
when(requestBodyUriMock.body(eq(BodyInserters.fromValue(requestBody)))).thenReturn(requestHeadersMock);
注意： 需要静态导入 org.mockito.Matchers.eq。
完整的单元测试示例
下面是一个完整的单元测试示例，展示了如何正确地 Mock WebClient 并进行参数匹配：
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.Mockito;
import org.mockito.junit.jupiter.MockitoExtension;
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.ArgumentMatchers.eq;
import static org.mockito.Mockito.when;
@ExtendWith(MockitoExtension.class)
public class SomeServiceTest {
@Mock
private WebClient webClientMock;
@Mock
private WebClient.RequestBodyUriSpec requestBodyUriMock;
@Mock
private WebClient.RequestHeadersSpec requestHeadersMock;
@Mock
private WebClient.ResponseSpec responseMock;
@InjectMocks
private SomeService someService; // 假设你的服务类名为 SomeService
@Test
public void testGetResponse() {
String requestBody = "test request body";
// 设置 Mock 行为
when(webClientMock.post()).thenReturn(requestBodyUriMock);
when(requestBodyUriMock.body(eq(BodyInserters.fromValue(requestBody)))).thenReturn(requestHeadersMock);
when(requestHeadersMock.retrieve()).thenReturn(responseMock);
when(responseMock.bodyToMono(String.class)).thenReturn(Mono.just("response"));
// Mock buildWebClient 方法，确保使用 mockWebClient
SomeService someServiceSpy = Mockito.spy(someService);
Mockito.doReturn(webClientMock).when(someServiceSpy).buildWebClient();
// 调用服务方法
String response = someServiceSpy.getResponse(requestBody);
// 断言结果
assertEquals("response", response);
}
}
// 示例服务类
class SomeService {
public String getResponse(String requestBody){
WebClient client = buildWebClient();
WebClient.RequestHeadersSpec<?> request =
client.post().body(BodyInserters.fromValue(requestBody));
String resp =
request.retrieve().bodyToMono(String.class)
.block();
return resp;
}
// 需要被 Mock 的方法
protected WebClient buildWebClient() {
return WebClient.builder().baseUrl("http://example.com").build();
}
}总结
通过将 WebClient 的创建过程提取到可 Mock 的方法中，并使用 Mockito.doReturn 来替换 WebClient 实例，可以确保单元测试中使用的是 Mock 对象。同时，需要注意 Mockito 的参数匹配，使用 eq 方法来精确匹配参数。 遵循这些步骤，可以有效地解决 WebClient Mock 对象未生效的问题，从而编写更准确、可靠的单元测试。
