Spring自定义Condition获取Bean失败及测试问题求助

1. 依赖Bean的Conditional能否正常工作？ #

不能直接通过beanFactory.getBean()的方式实现，核心原因是Spring Condition的执行时机太早：Condition的matches方法是在Spring容器加载BeanDefinition的阶段执行的，此时大部分Bean还没完成实例化，甚至连BeanDefinition都可能还没全部注册完成，直接调用getBean()必然抛出NoSuchBeanException。

如果你的Condition需要依赖其他Bean的逻辑，有几种可行的替代方案：

• 依赖Environment而非Bean：把FeatureFlag的配置直接放到环境变量、配置文件里，在Condition的matches方法中通过ConditionContext.getEnvironment()读取配置，完全绕开Bean依赖。
• 检查BeanDefinition而非实例：如果FeatureFlagManager的BeanDefinition已经提前注册，可通过beanFactory.containsBeanDefinition("featureFlagManager")判断是否存在对应的定义，虽然不能拿到实例，但可以基于定义是否存在做条件判断。
• 使用DeferredImportSelector延迟条件判断：通过实现DeferredImportSelector，让条件判断延迟到所有常规BeanDefinition都注册完成之后，此时可以安全地获取已注册的Bean实例（但仍需注意Bean是否已实例化，必要时结合@Lazy）。
• 配合@ConditionalOnBean注解：如果你的目标Bean本身依赖FeatureFlagManager，可以在目标Bean上同时添加@ConditionalOnBean(FeatureFlagManager.class)和自定义的@ConditionalOnFeatureEnabled，确保FeatureFlagManager的BeanDefinition存在后再执行自定义条件判断。

2. 该如何对其进行测试？ #

针对自定义Condition的测试，推荐用Spring Boot测试框架，结合不同场景验证Bean的创建逻辑，下面是具体的测试思路和示例：

测试步骤： #

• 编写测试专用的配置类，分别注册FeatureFlagManager、使用@ConditionalOnFeatureEnabled的目标Bean。
• 分两种场景测试：FeatureFlag开启时目标Bean存在，关闭时目标Bean不存在。
• 可以用AnnotationConfigApplicationContext手动加载配置，或者用@SpringBootTest结合@TestPropertySource快速切换配置。

示例代码： #

// 测试配置类
@Configuration
class TestConfig {
    @Bean
    fun featureFlagManager(): FeatureFlagManager {
        return FeatureFlagManager().apply {
            // 根据测试场景设置开关状态
            enableFeature("test-feature")
        }
    }

    @Bean
    @ConditionalOnFeatureEnabled(feature = "test-feature")
    fun testBean(): TestBean {
        return TestBean()
    }
}

// 测试类
class FeatureEnabledConditionTest {

    @Test
    fun `test bean created when feature is enabled`() {
        val context = AnnotationConfigApplicationContext(TestConfig::class.java)
        assertTrue(context.containsBean(TestBean::class.java))
        context.close()
    }

    @Test
    fun `test bean not created when feature is disabled`() {
        val context = AnnotationConfigApplicationContext()
        // 注册修改后的FeatureFlagManager
        context.registerBean(FeatureFlagManager::class.java) {
            FeatureFlagManager().apply {
                disableFeature("test-feature")
            }
        }
        context.register(TestConfig::class.java)
        context.refresh()
        assertFalse(context.containsBean(TestBean::class.java))
        context.close()
    }
}

注意事项： #

• 如果你的Condition依赖Environment，测试时可以通过context.environment.setProperty("feature.test-feature", "true")快速设置环境变量。
• 避免在测试中过度依赖容器的内部状态，尽量通过containsBean()或getBean()的存在性来验证条件逻辑。

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