如何使用Junit 5编写包含System.exit()的测试用例方法？

业务代码执行System.exit()会直接终止当前JVM进程，导致测试流程被打断无法完成验证。我们可以通过拦截System.exit()的执行流程，避免JVM实际退出，同时捕获退出状态码做断言验证。

方案1：原生JUnit 5实现（无第三方依赖） #

核心思路是替换JVM的安全管理器，拦截System.exit()调用，抛出特定异常终止退出流程，测试结束后恢复原始安全管理器避免影响其他用例。

import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import static org.junit.jupiter.api.Assertions.*;
import java.lang.reflect.Method;

public class MyClassTest {
    private SecurityManager originalSecurityManager;

    // 自定义异常用于标识exit请求被拦截
    private static class ExitTrappedException extends SecurityException {
        private final int exitCode;
        public ExitTrappedException(int exitCode) {
            this.exitCode = exitCode;
        }
        public int getExitCode() {
            return exitCode;
        }
    }

    @BeforeEach
    void setUp() {
        originalSecurityManager = System.getSecurityManager();
        // 替换自定义安全管理器拦截exit请求
        System.setSecurityManager(new SecurityManager() {
            @Override
            public void checkExit(int status) {
                super.checkExit(status);
                throw new ExitTrappedException(status);
            }

            // 放开其他权限避免影响正常测试逻辑
            @Override
            public void checkPermission(java.security.Permission perm) {
            }
        });
    }

    @AfterEach
    void tearDown() {
        // 测试结束恢复原始安全管理器
        System.setSecurityManager(originalSecurityManager);
    }

    @Test
    void testMethod1WithZeroInputExitWithCode1() throws Exception {
        MyClass targetObj = new MyClass();
        // 反射获取私有method1方法
        Method targetMethod = MyClass.class.getDeclaredMethod("method1", int.class);
        targetMethod.setAccessible(true);

        // 捕获exit请求对应的自定义异常
        ExitTrappedException exitException = assertThrows(ExitTrappedException.class, () -> {
            targetMethod.invoke(targetObj, 0);
        });

        // 验证退出状态码符合预期
        assertEquals(1, exitException.getExitCode());
    }
}

注意事项 #

• Java 17及以上版本默认禁用了SecurityManager，启动测试时需要添加JVM参数-Djava.security.manager=allow才能正常使用该方案。
• 若不想通过反射调用私有方法，可将目标方法的访问权限调整为包私有，测试类和目标类放在同一个包下即可直接调用。

方案2：使用System Stubs工具库实现（简洁易维护） #

核心思路是借助封装好的系统桩工具库，自动处理System.exit()的拦截逻辑，无需手动编写安全管理器相关代码，兼容性更好。

1. 引入Maven依赖

<dependency>
    <groupId>uk.org.webcompere</groupId>
    <artifactId>system-stubs-jupiter</artifactId>
    <version>2.1.6</version>
    <scope>test</scope>
</dependency>

2. 测试代码示例

import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import uk.org.webcompere.systemstubs.jupiter.SystemStub;
import uk.org.webcompere.systemstubs.jupiter.SystemStubsExtension;
import uk.org.webcompere.systemstubs.security.SystemExit;
import static org.junit.jupiter.api.Assertions.*;
import java.lang.reflect.Method;

@ExtendWith(SystemStubsExtension.class)
public class MyClassTest {
    @SystemStub
    private SystemExit systemExit;

    @Test
    void testMethod1WithZeroInputExitWithCode1() throws Exception {
        MyClass targetObj = new MyClass();
        Method targetMethod = MyClass.class.getDeclaredMethod("method1", int.class);
        targetMethod.setAccessible(true);

        // 执行目标方法，自动拦截exit请求
        systemExit.execute(() -> targetMethod.invoke(targetObj, 0));
        // 直接验证退出状态码
        assertEquals(1, systemExit.getExitCode());
    }
}

方案优势 #

• 不需要手动维护安全管理器的替换和恢复逻辑，框架自动处理
• 兼容Java 8到Java 21所有主流版本，高版本Java不需要额外加启动参数
• 支持同时捕获System.out、System.err等其他系统级输出，适合集成测试多验证点的场景

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