如何为 spring webflux 路由器编写单元测试？

Question

我有以下类型的 EntityRouter，现在我的问题是没有自动连接的处理程序，我如何模拟和编写 junit 单元测试（非集成）这样的端点？

注意：对于 controller，这对我来说在概念上很简单，因为如果我们使用 spring-web-mvc，我可以模拟通常在 controller 中自动装配的服务。

public class EntityRouter {

    @Bean
    public RouterFunction<ServerResponse> route(EntityHandler handler) {
        return RouterFunctions
                .route(GET("/getAllEntities").and(accept(MediaType.APPLICATION_JSON)), handler::findAll)
                .andRoute(GET("/getEntity/{id}").and(accept(MediaType.APPLICATION_STREAM_JSON)), handler::findById)
                .andRoute(POST("/createEntity").and(accept(MediaType.APPLICATION_JSON)), handler::save)
                .andRoute(DELETE("/deleteEntity/{id}").and(accept(MediaType.APPLICATION_JSON)), handler::delete);
    }

}

Answer 1

如果你模拟EntityHandler类就足够了：

@Test
public void testRoute() {
    EntityHandler mockHandler = mock(EntityHandler.class);

    RouterFunction<ServerResponse> routerFunction = new EntityRouter().route(mockHandler);

    // Set up a mock request
    ServerRequest request = mock(ServerRequest.class);
    when(request.method()).thenReturn(HttpMethod.GET);
    when(request.path()).thenReturn("/getAllEntities");
    when(request.headers()).thenReturn(new HttpHeaders());

    // Call the router function
    routerFunction.route(request).subscribe();

    // Verify that the correct method on the mock handler was called
    verify(mockHandler).findAll(request);
}

Answer 2

您可以将WebTestClient用于基于 stream 的响应式 rest 端点。 你可以在这里找到一些例子