如何通过Java agent终止Gradle测试进程，解决超时触发System.exit不生效问题

1. Gradle测试进程默认拦截System.exit调用：Gradle为了保证测试框架能正常收集测试结果、控制测试生命周期，默认会安装SecurityManager拦截System.exit()调用，调用时会直接抛出SecurityException，如果线程异常处理器没有处理对应异常，你根本感知不到调用失败，进程自然不会退出。
2. 静态CountDownLatch存在线程安全问题：你的latch是静态成员变量，如果Gradle开启了并行测试执行，或者前一个测试的超时检测线程还没结束，下一个测试的enter方法就把latch覆盖了，会导致超时判断逻辑完全错乱，可能出现真超时但latch.getCount()返回0的情况，自然不会执行System.exit逻辑。
3. 超时判断逻辑存在竞态条件：你在latch.await()超时后才判断latch.getCount() > 0，极端情况下可能刚超时还没走判断逻辑，测试方法就执行完成调用了countDown()，会导致超时逻辑漏触发。

1. 解除Gradle对System.exit的限制 #

Gradle的test任务可以通过配置放开System.exit调用权限，在你的build.gradle中添加如下配置即可，Gradle 7+版本可以直接使用setExitRule配置：

test {
    // 允许测试代码调用System.exit终止进程
    setExitRule(org.gradle.api.tasks.testing.Test.ExitRule.ALLOW)
}

低版本Gradle可以通过配置安全策略实现：

test {
    jvmArgs '-Djava.security.manager=allow'
}

2. 修复代码逻辑缺陷 #

2.1 避免使用静态latch变量 #

通过ByteBuddy的@Advice.Local存储当前测试调用生命周期内的局部变量，避免多测试执行时latch被覆盖的问题，同时给TestCallable增加构造函数传入对应latch，不要依赖静态变量。

2.2 替换为更可靠的进程终止方式 #

如果System.exit还是被shutdown钩子阻塞，可以直接调用Runtime.getRuntime().halt(1)，该方法会直接终止虚拟机，不会等待shutdown钩子执行，终止成功率更高。

2.3 修复超时判断逻辑 #

直接使用latch.await()的返回值判断是否超时，避免后续查count的竞态问题。

修改后代码示例：

// 改造后的TestCallable
public class TestCallable implements Runnable {
    private final CountDownLatch latch;
    // 构造函数传入当前测试对应的latch
    public TestCallable(CountDownLatch latch) {
        this.latch = latch;
    }
    @Override
    public void run()  {
        boolean isTimeout = false;
        try {
            isTimeout = !latch.await(10, TimeUnit.MINUTES);
        } catch (InterruptedException e) {
            Thread.currentThread().interrupt();
            return;
        }
        if(isTimeout) {
            System.err.println("Callable thread"
                    + Thread.currentThread().getName() +
                    "TIMEOUT OCCURED!!!!");
            // 直接终止进程，不会被阻塞
            Runtime.getRuntime().halt(1);
        }
    }
}

// 改造后的TimerAdvice
public class TimerAdvice {
    @Advice.OnMethodEnter
    static long enter(@Advice.Local("latch") CountDownLatch latch,
                      @Advice.Local("executor") ExecutorService executor) throws InterruptedException {
        System.out.println("Timer Advice Enter thread: " + Thread.currentThread().getName() + " time: " + Instant.now().toString());

        latch = new CountDownLatch(1);
        ThreadFactory factory = new MyThreadFactory(new MyExceptionHandler());
        executor = Executors.newFixedThreadPool(1, factory);
        executor.execute(new TestCallable(latch));

        return System.nanoTime();
    }

    @Advice.OnMethodExit (onThrowable = Throwable.class)
    static void onExit(@Advice.Local("latch") CountDownLatch latch,
                       @Advice.Local("executor") ExecutorService executor) {
        System.out.println("Timer Advice Exit thread: " + Thread.currentThread().getName() + " time: " + Instant.now().toString());
        System.out.println("Counting down");
        if (latch != null) {
            latch.countDown();
        }
        if (executor != null) {
            executor.shutdownNow();
        }
    }
}

3. 更简单的替代方案（无需自定义Agent） #

Gradle本身自带测试超时能力，无需自己实现字节码增强逻辑，直接配置即可：

test {
    // 单个测试用例超时时间，超过直接标记失败
    timeout = Duration.ofMinutes(10)
    // 如果超时后需要直接终止整个测试进程，增加监听器即可
    testListener {
        afterTest { descriptor, result ->
            if (result.resultType == TestResult.ResultType.FAILURE && result.exception instanceof TimeoutException) {
                Runtime.getRuntime().halt(1)
            }
        }
    }
}

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