时间戳

首页 > 解决方案 > Spring Cloud Gateway 自定义过滤器工厂的单元测试

java - Spring Cloud Gateway 自定义过滤器工厂的单元测试

问题描述

我已经为 Cloud Gateway 实现了自定义过滤器工厂。但是，我想不出编写单元测试用例的方法。在探索默认过滤器工厂测试用例时，我发现大多数工厂测试类扩展了 BaseWebClientTests和测试包中的其他类。

我的问题是我应该将那些中间测试类复制粘贴到我的本地测试包中吗？这里有什么社区推荐？

标签： javaspringspring-cloudspring-cloud-gateway

解决方案

这是我的结果，供您参考

class CustomGatewayFilterFactoryTest {

    @Autowired
    private CustomGatewayFilterFactory factory;

    private ServerWebExchange exchange;
    private GatewayFilterChain filterChain = mock(GatewayFilterChain.class);
    private ArgumentCaptor<ServerWebExchange> captor = ArgumentCaptor.forClass(ServerWebExchange.class);

    @BeforeEach
    void setup() {
        when(filterChain.filter(captor.capture())).thenReturn(Mono.empty());
    }

    @Test
    void customTest() {
        MockServerHttpRequest request = MockServerHttpRequest.get(DUMMY_URL).build();
        exchange = MockServerWebExchange.from(request);
        GatewayFilter filter = factory.apply(YOUR_FACTORY_CONFIG);
        filter.filter(exchange, filterChain);
        // filter.filter(exchange, filterChain).block(); if you have any reactive methods

        ServerHttpRequest actualRequest = captor.getValue().getRequest();

        // Now you can assert anything in the actualRequest
        assertEquals(request, actualRequest);
    }

}

推荐阅读