Spring测试场景下：是否可仅让一个Bean依赖另一个Bean？

当然可以！Spring提供了多种灵活的方式来实现仅在测试场景下让一个Bean依赖另一个Bean，下面结合你的Kafka生产者Bean场景，给你详细拆解可行方案：

1. 使用Spring Profiles区分环境 #

这是最常用的方式，通过Profiles标记不同环境的Bean，测试时激活指定Profile即可。

比如你有一个正常环境的Kafka生产者Bean，和一个测试环境下依赖额外Bean的版本：

// 主环境的Kafka生产者Bean（默认加载）
@Bean
@Profile("!test")
public KafkaProducer<String, String> kafkaProducer(Environment env) {
    Properties props = new Properties();
    props.put(ProducerConfig.BOOTSTRAP_SERVERS_CONFIG, env.getProperty("kafka.bootstrap-servers"));
    // 其他生产环境配置...
    return new KafkaProducer<>(props);
}

// 测试环境的Kafka生产者Bean，依赖测试专用的ZkUtils辅助Bean
@Bean
@Profile("test")
public KafkaProducer<String, String> testKafkaProducer(Environment env, TestZkHelper testZkHelper) {
    Properties props = new Properties();
    props.put(ProducerConfig.BOOTSTRAP_SERVERS_CONFIG, testZkHelper.getTestBootstrapServers());
    // 测试环境专属配置...
    return new KafkaProducer<>(props);
}

// 测试专用辅助Bean，仅在test Profile下加载
@Bean
@Profile("test")
public TestZkHelper testZkHelper() {
    return new TestZkHelper();
}

然后在测试类中激活test Profile：

@SpringBootTest
@ActiveProfiles("test")
public class KafkaProducerTest {
    @Autowired
    private KafkaProducer<String, String> kafkaProducer;
    
    // 测试逻辑...
}

2. 使用条件注解控制依赖加载 #

利用Spring的@Conditional系列注解（比如@ConditionalOnProperty、@ConditionalOnBean），根据测试场景的条件触发Bean的依赖关系。

比如你可以通过配置属性控制是否加载测试依赖：

// 主Kafka生产者Bean，默认不依赖测试Bean
@Bean
public KafkaProducer<String, String> kafkaProducer(Environment env, 
        @Autowired(required = false) TestZkHelper testZkHelper) {
    Properties props = new Properties();
    if (testZkHelper != null) {
        // 测试场景下使用测试辅助Bean的配置
        props.put(ProducerConfig.BOOTSTRAP_SERVERS_CONFIG, testZkHelper.getTestBootstrapServers());
    } else {
        // 生产环境使用配置文件的配置
        props.put(ProducerConfig.BOOTSTRAP_SERVERS_CONFIG, env.getProperty("kafka.bootstrap-servers"));
    }
    // 其他配置...
    return new KafkaProducer<>(props);
}

// 仅在测试属性开启时加载测试辅助Bean
@Bean
@ConditionalOnProperty(name = "test.kafka.enable", havingValue = "true")
public TestZkHelper testZkHelper() {
    return new TestZkHelper();
}

然后在测试的application-test.properties中添加：

test.kafka.enable=true

3. 测试专用配置类 #

在测试目录（src/test/java）下创建独立的配置类，覆盖或补充主配置的Bean定义，Spring测试时会优先识别测试目录的配置（或通过@ContextConfiguration指定）。

比如测试配置类：

@Configuration
public class TestKafkaConfig {
    // 定义测试专用的辅助Bean
    @Bean
    public TestZkHelper testZkHelper() {
        return new TestZkHelper();
    }

    // 定义依赖测试辅助Bean的Kafka生产者Bean，覆盖主配置的同名Bean
    @Bean
    public KafkaProducer<String, String> kafkaProducer(TestZkHelper testZkHelper) {
        Properties props = new Properties();
        props.put(ProducerConfig.BOOTSTRAP_SERVERS_CONFIG, testZkHelper.getTestBootstrapServers());
        // 测试配置...
        return new KafkaProducer<>(props);
    }
}

然后在测试类中指定使用这个配置：

@SpringBootTest(classes = TestKafkaConfig.class)
public class KafkaProducerTest {
    @Autowired
    private KafkaProducer<String, String> kafkaProducer;
    
    // 测试逻辑...
}

以上几种方案都能轻松实现“仅测试场景下Bean依赖另一个Bean”的需求，你可以根据自己的测试复杂度和场景选择最合适的方式~

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