我正在尝试模拟Spring WebClient,但在模拟WebClient.builder()时遇到了问题。到目前为止,我在测试中定义的模拟没有被使用,我想是因为这个构建器没有返回我的模拟WebClient。我怎样才能让构建器返回我的mock?
WebClient webClient = WebClient.builder()
.exchangeStrategies(ExchangeStrategies.builder()
.codecs(configurer -> configurer
.defaultCodecs()
.maxInMemorySize(16 * 1024 * 1024))
.build())
.clientConnector(new ReactorClientHttpConnector(HttpClient.newConnection().compress(true)))
.build();
并且WebClient是这样使用的:
response = webClient
.get()
.uri(URLDecoder.decode(someEndpoint.replace("\"", ""), CLIENT_ENCODING))
.header(AUTHORIZATION, BEARER + accessToken)
.accept(MediaType.APPLICATION_JSON)
.retrieve().bodyToMono(JsonNode.class).block();
当我遵循这些解决方案时,我没有看到任何关于模拟的问题:How to mock Spring WebFlux WebClient?,但是模拟没有被使用。我如何模拟构建器/从构建器返回我的模拟?
发布于 2021-11-05 19:03:16
您需要模拟WebClient
的静态方法builder()
。事实上,如果这个方法没有被mocked,mockito就不能编辑这个方法的行为。因此,不会使用模拟的webClient。要做到这一点,如果您使用maven,则需要在pom.xml
中包含此依赖项。
<dependency>
<groupId>org.mockito</groupId>
<artifactId>mockito-inline</artifactId>
<version>3.4.6</version>
<scope>test</scope>
</dependency>
这将在Mockito
上提供mockStatic()
方法,使用方法如下:
MockedStatic<WebClient> webClientStatic = Mockito.mockStatic(WebClient.class);
webClientStatic.when(WebClient::builder).thenReturn(webClientBuilder);
webClientBuilder
必须被模仿,因为我们需要改变它的方法build()
的行为来返回你被模仿的webClient。
Mockito.when(webClientBuilder.build()).thenReturn(mockedWebClient);
要在一个代码中恢复所有内容:
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.Mockito;
import org.mockito.junit.MockitoJUnitRunner;
import org.springframework.http.client.reactive.ReactorClientHttpConnector;
import org.springframework.web.reactive.function.client.ExchangeStrategies;
import org.springframework.web.reactive.function.client.WebClient;
import reactor.netty.http.client.HttpClient;
import org.mockito.MockedStatic;
@RunWith(MockitoJUnitRunner.class)
public class WebClientTest {
@Mock
private WebClient.Builder webClientBuilder;
@Mock
private WebClient mockedWebClient;
@Before
public void setup() {
Mockito.when(webClientBuilder.exchangeStrategies(Mockito.any(ExchangeStrategies.class))).thenReturn(webClientBuilder);
Mockito.when(webClientBuilder.clientConnector(Mockito.any(ReactorClientHttpConnector.class))).thenReturn(webClientBuilder);
Mockito.when(webClientBuilder.build()).thenReturn(mockedWebClient);
}
@Test
public void testWebClient() {
final MockedStatic<WebClient> webClientStatic = Mockito.mockStatic(WebClient.class);
webClientStatic.when(WebClient::builder).thenReturn(webClientBuilder);
final WebClient webClient = WebClient.builder()
.exchangeStrategies(ExchangeStrategies.builder()
.codecs(configurer -> configurer
.defaultCodecs()
.maxInMemorySize(16 * 1024 * 1024))
.build())
.clientConnector(new ReactorClientHttpConnector(HttpClient.newConnection().compress(true)))
.build();
//It throws error if webClient is not mockedWebClient
Assert.assertEquals(webClient, mockedWebClient);
}
}
您可以对其进行测试。
https://stackoverflow.com/questions/69857313
复制相似问题