JUnit 5集成测试后Spring Boot应用未自动关闭的优雅关闭方案

一、导致应用无法自动终止的常见原因 #

• 后台线程未正确终止：应用内存在未被管理的自定义线程池、定时任务线程等，Spring容器关闭时这些线程仍处于运行状态，阻止JVM进程退出。
• 资源未完全释放：数据库连接池、Socket连接、文件句柄等资源未被正确关闭，第三方库或自定义组件持有这些资源导致上下文无法彻底销毁。
• 版本兼容性细节：Spring Boot 2.6.x与JUnit 5.8.x的组合下，可能存在测试上下文销毁钩子未被正常触发的情况，导致上下文关闭逻辑执行不完整。
• 默认上下文加载过重：@SpringBootTest默认启动完整应用上下文，若包含非懒加载的长生命周期Bean（如Web服务器、消息队列消费者），可能导致关闭延迟或失败。

二、优雅解决的配置与代码实现 #

1. 排查并终止后台线程 #

针对自定义线程池或定时任务，在测试结束后主动关闭：

@SpringBootTest
public class MyIntegrationTest {
    @Autowired(required = false)
    private ThreadPoolTaskExecutor taskExecutor;

    @Autowired(required = false)
    private ScheduledExecutorService scheduledExecutor;

    @AfterAll
    static void cleanUpThreads(ThreadPoolTaskExecutor taskExecutor, ScheduledExecutorService scheduledExecutor) {
        // 关闭线程池
        if (taskExecutor != null) {
            taskExecutor.shutdown();
            try {
                if (!taskExecutor.awaitTermination(10, TimeUnit.SECONDS)) {
                    taskExecutor.shutdownNow();
                }
            } catch (InterruptedException e) {
                taskExecutor.shutdownNow();
            }
        }
        // 关闭定时任务线程
        if (scheduledExecutor != null) {
            scheduledExecutor.shutdown();
            try {
                if (!scheduledExecutor.awaitTermination(10, TimeUnit.SECONDS)) {
                    scheduledExecutor.shutdownNow();
                }
            } catch (InterruptedException e) {
                scheduledExecutor.shutdownNow();
            }
        }
    }

    @Test
    public void testSomething() {
        // 测试逻辑
    }
}

2. 优化@SpringBootTest的web环境配置 #

如果测试不需要启动Web服务器，指定webEnvironment为NONE，减少不必要的资源加载：

@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.NONE)
public class MyIntegrationTest {
    @Test
    public void testSomething() {
        // 测试逻辑
    }
}

3. 强制上下文销毁时机 #

配合@DirtiesContext指定类级别的销毁模式，确保所有测试结束后彻底销毁上下文：

@SpringBootTest
@DirtiesContext(classMode = DirtiesContext.ClassMode.AFTER_CLASS)
public class MyIntegrationTest {
    @Test
    public void testSomething() {
        // 测试逻辑
    }
}

4. 自定义资源清理逻辑 #

通过ApplicationListener监听上下文关闭事件，主动释放资源：

@Component
public class ContextCleanupListener implements ApplicationListener<ContextClosedEvent> {
    @Override
    public void onApplicationEvent(ContextClosedEvent event) {
        // 释放数据库连接、关闭Socket等资源的逻辑
        // 例如：dataSource.close();
    }
}

5. 检查第三方依赖 #

排查是否有依赖库持有后台线程（如某些监控、日志组件），可临时排除依赖测试是否恢复正常，再针对性处理该依赖的资源释放逻辑。

内容的提问来源于stack exchange，提问作者Krisztofer Török