如何为使用ZeroCopyHttpOutputMessage的Reactive端点编写集成测试？

原因说明 #

MockServerHttpResponse是Spring测试套件提供的轻量级模拟实现，核心用于验证响应状态、头部、内容等业务逻辑正确性，并未实现ZeroCopyHttpOutputMessage这类针对底层传输性能优化的接口——单元测试场景下，性能优化并非核心关注点，因此Spring没有为该Mock类添加此接口实现。

解决方法 #

方案1：自定义支持零拷贝的Mock响应类 #

实现一个继承MockServerHttpResponse并实现ZeroCopyHttpOutputMessage的自定义Mock类，在测试中替换默认的响应实例：

public class ZeroCopyMockServerHttpResponse extends MockServerHttpResponse implements ZeroCopyHttpOutputMessage {
    @Override
    public Mono<Void> writeWith(File file, long position, long count) {
        try {
            // 模拟零拷贝逻辑，将文件内容写入Mock响应缓冲区
            byte[] content = Files.readAllBytes(file.toPath());
            return writeWith(ByteBuffer.wrap(content, (int) position, (int) count));
        } catch (IOException e) {
            return Mono.error(e);
        }
    }
}

配置WebTestClient时使用自定义响应：

WebTestClient webTestClient = WebTestClient.bindToController(new YourFileController())
        .configureClient()
        .responseProvider(exchange -> Mono.just(new ZeroCopyMockServerHttpResponse()))
        .build();

方案2：使用真实服务器环境测试 #

通过@SpringBootTest启动内嵌真实服务器，让WebTestClient连接真实服务器实例，此时会使用对应服务器（如Netty、Tomcat）的ServerHttpResponse实现，这类实现天然支持ZeroCopyHttpOutputMessage：

@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
public class FileControllerTest {
    @Autowired
    private WebTestClient webTestClient;

    @Test
    void testFileDownload() {
        webTestClient.get().uri("/download")
                .exchange()
                .expectStatus().isOk()
                .expectHeader().contentType(MediaType.APPLICATION_OCTET_STREAM);
    }
}

方案3：端点代码添加兼容性降级 #

在业务代码中先判断响应实例是否实现ZeroCopyHttpOutputMessage，未实现时使用普通文件传输逻辑降级，避免强制转换报错：

@GetMapping("/download")
public Mono<Void> downloadFile(ServerHttpResponse response) {
    File targetFile = new File("/path/to/target/file");
    response.getHeaders().setContentLength(targetFile.length());
    response.getHeaders().setContentType(MediaType.APPLICATION_OCTET_STREAM);

    if (response instanceof ZeroCopyHttpOutputMessage) {
        // 生产环境使用零拷贝传输
        return ((ZeroCopyHttpOutputMessage) response).writeWith(targetFile, 0, targetFile.length());
    } else {
        // 测试环境降级为普通DataBuffer传输
        return response.writeWith(DataBufferUtils.read(targetFile, response.bufferFactory(), 4096));
    }
}

内容的提问来源于stack exchange，提问作者Lakatos Gyula