如何为Spring Kafka配置类编写JUnit测试？

下面提供几种可行的方案，根据你的测试场景选择：

方案1：通过测试属性覆盖+Spring上下文加载 #

如果getConfig()依赖配置文件中的属性（比如@Value注入的字段），可以直接在测试类中覆盖这些属性，让Spring自动创建符合预期的ConsumerFactory：

@SpringBootTest
@TestPropertySource(properties = {
    "kafka.bootstrap-servers=localhost:9092",
    "kafka.group-id=test-consumer-group",
    "kafka.key-deserializer=org.apache.kafka.common.serialization.StringDeserializer"
})
public class ConsumerConfigTest {

    @Autowired
    private ApplicationContext applicationContext;

    @Test
    void testConsumerFactoryCreation() {
        ConsumerFactory<String, SomeClass> consumerFactory = 
            applicationContext.getBean("consumerFactory", ConsumerFactory.class);
        
        // 验证配置是否符合预期
        Map<String, Object> configProps = consumerFactory.getConfigurationProperties();
        assert configProps.get("bootstrap.servers").equals("localhost:9092");
        assert configProps.get("group.id").equals("test-consumer-group");
    }
}

方案2：纯单元测试，手动模拟依赖+构建Bean #

如果不想启动Spring上下文，直接实例化ConsumerConfig并模拟其依赖字段，手动调用方法构建ConsumerFactory：

public class ConsumerConfigTest {

    @Test
    void testConsumerFactoryWithMockedDependencies() {
        // 模拟配置类中的依赖字段（比如从外部服务获取的配置项）
        SomeConfigService mockedConfigService = Mockito.mock(SomeConfigService.class);
        when(mockedConfigService.getKafkaBootstrapServers()).thenReturn("mock-kafka:9092");

        // 实例化配置类
        ConsumerConfig consumerConfig = new ConsumerConfig(mockedConfigService);
        
        // 获取配置Map并手动创建ConsumerFactory
        Map<String, Object> configMap = consumerConfig.getConfig();
        ConsumerFactory<String, SomeClass> consumerFactory = 
            new DefaultKafkaConsumerFactory<>(configMap);
        
        // 验证配置正确性
        assert consumerFactory.getConfigurationProperties().get("bootstrap.servers").equals("mock-kafka:9092");
    }
}

方案3：部分模拟配置类（@SpyBean） #

如果只想替换getConfig()方法的返回值，而保留配置类其他逻辑，可以用@SpyBean实现部分模拟：

@SpringBootTest
public class ConsumerConfigTest {

    @SpyBean
    private ConsumerConfig consumerConfig;

    @Autowired
    private ApplicationContext applicationContext;

    @Test
    void testConsumerFactoryWithMockedConfig() {
        // 自定义模拟的配置Map
        Map<String, Object> mockedConfig = new HashMap<>();
        mockedConfig.put("bootstrap.servers", "spy-mock:9092");
        mockedConfig.put("group.id", "spy-test-group");
        
        // 让getConfig()返回模拟的Map
        doReturn(mockedConfig).when(consumerConfig).getConfig();

        // 从上下文获取Bean并验证
        ConsumerFactory<String, SomeClass> consumerFactory = 
            applicationContext.getBean("consumerFactory", ConsumerFactory.class);
        assert consumerFactory.getConfigurationProperties().equals(mockedConfig);
    }
}

注意事项 #

• 如果getConfig()是private方法，Mockito无法直接模拟，建议将配置逻辑抽为protected/public方法，或者通过模拟其依赖的字段/服务来间接控制返回值。
• 纯单元测试场景下，避免启动Spring上下文可以大幅提升测试速度。

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