如何验证Spring Boot关闭钩子的调用？自定义SmartLifecycle关闭处理器的测试问题

看起来你已经自己揪出问题的根源啦！不过我还是帮你把这个问题的关键和解决思路梳理得更清晰些，方便后续复盘或者给其他遇到类似问题的开发者参考：

问题背景回顾 #

你为了让自定义库的生命周期和Spring Boot集成，用SmartLifecycle封装了专属的启动/关闭处理器，测试启动逻辑时一切正常：能验证启动处理器处于运行状态、对应的AtomicBoolean标记也正确翻转，但一开始调用ApplicationContextRunner的context.stop()却没触发关闭处理器的逻辑，一度怀疑是不是ApplicationContextRunner不支持这个操作。

核心问题原因 #

你最初的测试在调用context.stop()之前塞进了太多断言和Bean查找逻辑——这些额外操作干扰了Spring上下文的状态，导致stop()没有正确触发所有SmartLifecycle的关闭回调。简单来说就是：测试逻辑太繁杂，分散了对上下文关闭流程的聚焦，反而让预期的关闭行为没按触发。

修正后的测试方案 #

简化测试逻辑，只保留「启动验证」「关闭触发」「关闭验证」三个核心步骤，去掉冗余的中间断言，就能让context.stop()正常触发关闭处理器了：

@Test
void lifecycleRegistryIntegratesWithDedicatedSpringLifecycleBeans() {
    testContext.run(context -> {
        // 1. 验证启动逻辑已执行
        AtomicBoolean startHandlerInvoked = context.getBean("startHandlerInvoked", AtomicBoolean.class);
        assertTrue(startHandlerInvoked.get());
        
        AtomicBoolean shutdownHandlerInvoked = context.getBean("shutdownHandlerInvoked", AtomicBoolean.class);
        assertFalse(shutdownHandlerInvoked.get());

        // 触发上下文关闭（这一步现在会正常执行关闭逻辑了）
        context.stop();

        // 2. 等待并验证关闭逻辑已执行
        await().atMost(Duration.ofSeconds(5))
               .pollDelay(Duration.ofMillis(25))
               .until(shutdownHandlerInvoked::get);

        // 可选：验证关闭处理器的运行状态
        Map<String, SpringLifecycleShutdownHandler> shutdownHandlers = BeanFactoryUtils.beansOfTypeIncludingAncestors(
            context, SpringLifecycleShutdownHandler.class
        );
        for (SpringLifecycleShutdownHandler handler : shutdownHandlers.values()) {
            assertFalse(handler.isRunning());
        }
    });
}

额外优化建议 #

1. 拆分测试用例：把「启动逻辑验证」和「关闭逻辑验证」拆成两个独立的测试方法，每个测试只聚焦一件事，既能提升测试可读性，也能避免上下文状态被交叉干扰。
2. 检查SmartLifecycle实现：确保你的关闭处理器正确实现了SmartLifecycle的方法，比如isAutoStartup()返回true，getPhase()设置了合理的阶段值（如果有多个生命周期Bean，阶段值会影响执行顺序）。
3. 线程安全的状态标记：你用AtomicBoolean来标记处理器是否被调用的做法非常稳妥，能避免多线程场景下的状态不一致问题，继续保持~

内容来源于stack exchange