首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >如何模拟Spring和WebClient

如何模拟Spring和WebClient
EN

Stack Overflow用户
提问于 2021-11-05 17:46:37
回答 1查看 944关注 0票数 1

我正在尝试模拟Spring WebClient,但在模拟WebClient.builder()时遇到了问题。到目前为止,我在测试中定义的模拟没有被使用,我想是因为这个构建器没有返回我的模拟WebClient。我怎样才能让构建器返回我的mock?

代码语言:javascript
运行
复制
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是这样使用的:

代码语言:javascript
运行
复制
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?,但是模拟没有被使用。我如何模拟构建器/从构建器返回我的模拟?

EN

回答 1

Stack Overflow用户

发布于 2021-11-05 19:03:16

您需要模拟WebClient的静态方法builder()。事实上,如果这个方法没有被mocked,mockito就不能编辑这个方法的行为。因此,不会使用模拟的webClient。要做到这一点,如果您使用maven,则需要在pom.xml中包含此依赖项。

代码语言:javascript
运行
复制
<dependency>
    <groupId>org.mockito</groupId>
    <artifactId>mockito-inline</artifactId>
    <version>3.4.6</version>
    <scope>test</scope>
</dependency>

这将在Mockito上提供mockStatic()方法,使用方法如下:

代码语言:javascript
运行
复制
MockedStatic<WebClient> webClientStatic = Mockito.mockStatic(WebClient.class);
webClientStatic.when(WebClient::builder).thenReturn(webClientBuilder);

webClientBuilder必须被模仿,因为我们需要改变它的方法build()的行为来返回你被模仿的webClient。

代码语言:javascript
运行
复制
Mockito.when(webClientBuilder.build()).thenReturn(mockedWebClient);

要在一个代码中恢复所有内容:

代码语言:javascript
运行
复制
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);
    }
}

您可以对其进行测试。

票数 1
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/69857313

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档