如何在JUnit5中实现测试用例写入System.out/err即判定失败的扩展

现有路径可行性判定 #

你尝试的TestExecutionListener#reportingEntryPublished路径不可行：JUnit在发布报告条目阶段已经完成了测试用例的执行状态判定，该阶段仅支持追加报告数据，不支持回溯修改测试结果状态，因此这条路径无法满足需求。

更简洁的实现方案 #

JUnit5 提供的InvocationInterceptor扩展接口是最优解决方案，它专门用于包装测试执行逻辑，比你之前尝试的BeforeAllCallback/AfterAllCallback重定向方案复杂度低得多，也避免了流泄露、上下文传递等维护隐患。

核心实现逻辑 #

通过InvocationInterceptor拦截每个测试方法的执行周期，在执行前替换System.out/System.err为自定义缓冲区，执行后恢复原始流并校验缓冲区内容，存在输出则直接抛出断言错误标记测试失败，异常默认会被JUnit关联到对应测试用例，无需额外处理定位逻辑。

代码示例 #

import org.junit.jupiter.api.extension.*;
import java.io.ByteArrayOutputStream;
import java.io.PrintStream;
import java.nio.charset.StandardCharsets;
import java.lang.reflect.Method;

public class NoStdOutputExtension implements InvocationInterceptor {
    private static final ThreadLocal<PrintStream> ORIGINAL_OUT = new ThreadLocal<>();
    private static final ThreadLocal<PrintStream> ORIGINAL_ERR = new ThreadLocal<>();
    private static final ThreadLocal<ByteArrayOutputStream> CAPTURED_OUT = new ThreadLocal<>();
    private static final ThreadLocal<ByteArrayOutputStream> CAPTURED_ERR = new ThreadLocal<>();

    @Override
    public void interceptTestMethod(Invocation<Void> invocation, 
                                    ReflectiveInvocationContext<Method> invocationContext, 
                                    ExtensionContext extensionContext) throws Throwable {
        try {
            // 替换标准输出流
            ORIGINAL_OUT.set(System.out);
            ORIGINAL_ERR.set(System.err);
            CAPTURED_OUT.set(new ByteArrayOutputStream());
            CAPTURED_ERR.set(new ByteArrayOutputStream());
            System.setOut(new PrintStream(CAPTURED_OUT.get(), true, StandardCharsets.UTF_8));
            System.setErr(new PrintStream(CAPTURED_ERR.get(), true, StandardCharsets.UTF_8));

            // 执行测试方法
            invocation.proceed();

            // 校验输出内容
            String outContent = CAPTURED_OUT.get().toString(StandardCharsets.UTF_8);
            String errContent = CAPTURED_ERR.get().toString(StandardCharsets.UTF_8);
            if (!outContent.isBlank() || !errContent.isBlank()) {
                throw new AssertionError(
                    String.format("测试方法禁止输出到标准流，检测到内容：%nSystem.out：%s%nSystem.err：%s",
                    outContent, errContent)
                );
            }
        } finally {
            // 恢复原始流，避免影响其他测试
            System.setOut(ORIGINAL_OUT.get());
            System.setErr(ORIGINAL_ERR.get());
            ORIGINAL_OUT.remove();
            ORIGINAL_ERR.remove();
            CAPTURED_OUT.remove();
            CAPTURED_ERR.remove();
        }
    }

    // 按需可重写interceptBeforeEachMethod、interceptAfterEachMethod等方法，覆盖生命周期方法的输出校验
}

扩展启用方式 #

• 局部启用：在需要校验的测试类上添加@ExtendWith(NoStdOutputExtension.class)注解
• 全局启用：在src/test/resources/META-INF/services/org.junit.jupiter.api.extension.Extension文件中写入实现类的全限定名，或开启自动检测后给扩展类添加@AutoService(Extension.class)注解即可全局生效

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