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

Question

我已经为云网关实现了自定义过滤器工厂。但是，我无法弄清楚编写单元测试用例的方法。在探索默认的过滤器工厂测试用例时，我发现大多数工厂测试类都扩展了 BaseWebClientTests以及测试包中的其他类。

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

Answer 1

这是我的结果，供您引用

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);
    }

}