如何在Spring Boot测试的BootstrapContext中编程式注册Bean？

除了通过spring.factories指定BootstrapRegistryInitializer，还有以下几种可行方案：

1. 自定义SpringBootContextBootstrapper并结合@BootstrapWith #

你可以继承SpringBootContextBootstrapper，重写createBootstrapContext方法，直接在里面向BootstrapRegistry注册所需Bean或初始化逻辑，无需依赖外部的BootstrapRegistryInitializer。

示例代码：

public class CustomTestBootstrapper extends SpringBootContextBootstrapper {
    @Override
    protected BootstrapContext createBootstrapContext() {
        BootstrapContext bootstrapContext = super.createBootstrapContext();
        // 直接注册目标Bean
        bootstrapContext.getRegistry().register(YourEarlyBean.class, () -> new YourEarlyBean());
        // 也可以注册自定义的BootstrapRegistryInitializer实现
        bootstrapContext.getRegistry().register(BootstrapRegistryInitializer.class, YourCustomInitializer::new);
        return bootstrapContext;
    }
}

然后在测试类上指定这个自定义Bootstrapper：

@SpringBootTest
@BootstrapWith(CustomTestBootstrapper.class)
public class YourIntegrationTest {
    // 测试逻辑
}

2. 使用TestContextCustomizer自定义测试上下文 #

通过实现TestContextCustomizer和TestContextCustomizerFactory，可以在测试上下文初始化的早期阶段介入，获取BootstrapContext并注册Bean。

首先实现TestContextCustomizer：

public class EarlyBeanTestContextCustomizer implements TestContextCustomizer {
    @Override
    public void customizeContext(TestContext testContext, MergedContextConfiguration mergedConfig) {
        BootstrapContext bootstrapContext = testContext.getBootstrapContext();
        bootstrapContext.getRegistry().register(YourEarlyBean.class, YourEarlyBean::new);
    }
}

然后实现工厂类来提供这个Customizer：

public class EarlyBeanCustomizerFactory implements TestContextCustomizerFactory {
    @Override
    public TestContextCustomizer createTestContextCustomizer(Class<?> testClass, List<ContextCustomizerFactory> factories) {
        return new EarlyBeanTestContextCustomizer();
    }
}

最后通过spring.factories注册这个工厂（这种方式更通用，无需在每个测试类上额外标注）：

org.springframework.test.context.ContextCustomizerFactory=com.yourpackage.EarlyBeanCustomizerFactory

3. 结合配置类与BootstrapContext间接注入 #

如果你的Bean需要依赖BootstrapContext中的资源，可以在配置类中直接注入BootstrapContext来创建Bean，只需确保配置类被提前加载：

@Configuration
public class EarlyBeanConfig {
    @Bean
    public YourEarlyBean yourEarlyBean(BootstrapContext bootstrapContext) {
        // 利用BootstrapContext中已注册的资源初始化Bean
        return new YourEarlyBean(bootstrapContext.getRegistry().get(RequiredDependency.class));
    }
}

可以通过@Import(EarlyBeanConfig.class)或者测试类上的@ContextConfiguration(classes = EarlyBeanConfig.class)确保配置类被加载。

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