Spring WebFlux控制器中非响应式验证的集成与单元测试

import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RestController;
import reactor.core.publisher.Mono;

@RestController
public class MangoController {

    private final MangoService serviceLayer;

    public MangoController(MangoService serviceLayer) {
        this.serviceLayer = serviceLayer;
    }

    @GetMapping("/mango/{id}")
    public Mono<Mango> getMango(@PathVariable("id") final String id) {
        // 问题所在：validateId() 是一个非响应式方法，会立即执行
        validateId(id); 
        return serviceLayer.someMonoData(); // 响应式流的定义
    }

    // 假设这是一个非响应式验证方法
    private void validateId(String id) {
        if ("invalid-id".equals(id) || id == null || id.isEmpty()) {
            throw new CustomBadRequestException("Invalid ID provided: " + id);
        }
        // 其他有效ID的验证逻辑
    }
}

// 假设的Service层接口和数据模型
interface MangoService {
    Mono<Mango> someMonoData();
}

class Mango {
    private String id;
    private String name;
    public Mango(String id, String name) { this.id = id; this.name = name; }
    // Getters, equals, hashCode...
    public String getId() { return id; }
    public String getName() { return name; }
    @Override public boolean equals(Object o) { if (this == o) return true; if (o == null || getClass() != o.getClass()) return false; Mango mango = (Mango) o; return id.equals(mango.id) && name.equals(mango.name); }
    @Override public int hashCode() { return java.util.Objects.hash(id, name); }
}

// 自定义异常
class CustomBadRequestException extends RuntimeException {
    public CustomBadRequestException(String message) {
        super(message);
    }
}

import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RestController;
import reactor.core.publisher.Mono;

@RestController
public class MangoController {

    private final MangoService serviceLayer;

    public MangoController(MangoService serviceLayer) {
        this.serviceLayer = serviceLayer;
    }

    @GetMapping("/mango/{id}")
    public Mono<Mango> getMango(@PathVariable("id") final String id) {
        // 改进方案：将非响应式验证包装到响应式流中
        return Mono.fromRunnable(() -> validateId(id)) // 验证现在是响应式流的一部分
                   .then(serviceLayer.someMonoData()); // 验证成功后才执行服务调用
    }

    private void validateId(String id) {
        if ("invalid-id".equals(id) || id == null || id.isEmpty()) {
            throw new CustomBadRequestException("Invalid ID provided: " + id);
        }
        // 其他有效ID的验证逻辑
    }
}

import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.autoconfigure.web.reactive.WebFluxTest;
import org.springframework.boot.test.mock.mockito.MockBean;
import org.springframework.http.MediaType;
import org.springframework.test.web.reactive.server.WebTestClient;
import reactor.core.publisher.Mono;

import static org.mockito.Mockito.when;
import static org.mockito.Mockito.never;
import static org.mockito.Mockito.verify;

// 假设MangoController是你的控制器类
@WebFluxTest(MangoController.class)
public class MangoControllerTest {

    @Autowired
    private WebTestClient webTestClient;

    @MockBean // 模拟服务层，避免实际的服务调用
    private MangoService serviceLayer;

    /**
     * 测试有效ID的场景：验证通过，服务层被调用，返回OK状态。
     */
    @Test
    void getMango_withValidId_shouldReturnOk() {
        // 模拟服务层的响应
        Mango expectedMango = new Mango("valid-id", "Sweet Mango");
        when(serviceLayer.someMonoData()).thenReturn(Mono.just(expectedMango));

        webTestClient.get().uri("/mango/valid-id")
                .accept(MediaType.APPLICATION_JSON)
                .exchange() // 执行请求
                .expectStatus().isOk() // 期望HTTP状态码为200 OK
                .expectBody(Mango.class).isEqualTo(expectedMango); // 期望响应体内容

        // 验证服务层的方法确实被调用了
        verify(serviceLayer).someMonoData();
    }

    /**
     * 测试无效ID的场景：验证失败（抛出CustomBadRequestException），服务层不被调用，返回BAD_REQUEST状态。
     */
    @Test
    void getMango_withInvalidId_shouldReturnBadRequest() {
        // 对于无效ID，我们不期望服务层被调用，所以不需要模拟其返回值
        // 如果validateId抛出异常，Mono.fromRunnable会发出错误信号，
        // 进而导致整个响应式流提前终止，serviceLayer.someMonoData()不会被订阅。

        webTestClient.get().uri("/mango/invalid-id")
                .accept(MediaType.APPLICATION_JSON)
                .exchange() // 执行请求
                .expectStatus().isBadRequest(); // 期望HTTP状态码为400 BAD_REQUEST

        // 验证服务层的方法没有被调用
        verify(serviceLayer, never()).someMonoData();
    }

    /**
     * 测试空ID的场景：验证失败，返回BAD_REQUEST状态。
     */
    @Test
    void getMango_withNullId_shouldReturnBadRequest() {
        webTestClient.get().uri("/mango/") // 假设空ID或缺失ID也会触发验证失败
                .accept(MediaType.APPLICATION_JSON)
                .exchange()
                .expectStatus().isBadRequest();
        verify(serviceLayer, never()).someMonoData();
    }
}