使用MockServerRequest断言WebFlux ServerResponse

import org.springframework.mock.http.server.reactive.MockServerHttpRequest;
import org.springframework.mock.web.server.MockServerWebExchange;
import org.springframework.web.reactive.function.server.ServerResponse;
import org.springframework.web.server.WebHandler;
import reactor.core.publisher.Mono;

public class WebFluxTest {

    public void testHandlerFunction() {
        // 创建一个模拟的HTTP请求
        MockServerHttpRequest request = MockServerHttpRequest.get("/api/user")
                .header("Content-Type", "application/json")
                .build();

        // 创建一个模拟的WebExchange
        MockServerWebExchange exchange = MockServerWebExchange.from(request);

        // 创建一个处理器函数
        WebHandler handler = exchange1 -> {
            // 处理请求并返回响应
            return ServerResponse.ok().bodyValue("Hello, World!");
        };

        // 使用MockServerRequest触发处理器函数的处理
        Mono<ServerResponse> responseMono = handler.handle(exchange);

        // 验证生成的ServerResponse是否符合预期
        responseMono.subscribe(response -> {
            // 验证状态码是否为200
            assert response.statusCode().value() == 200;

            // 验证响应体是否为"Hello, World!"
            assert response.body().equals("Hello, World!");
        });
    }
}