使用@SpringBootTest结合@KafkaListener与@MockitoBean时的问题

环境配置 #

Kafka消费者代码 #

@KafkaListener(topics = "${pmp.kafka.import-catalog-processing.topic}", concurrency = "1", groupId = "${pmp.kafka.group-id}",
        containerFactory = "kafkaProductListenerContainerFactory", errorHandler = "kafkaErrorHandler")
public void listen(List<String> values, @Header(KafkaHeaders.RECEIVED_KEY) List<String> keys) { ...

该消费者依赖FileDownloadService完成业务处理。

集成测试基类代码 #

@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
@ActiveProfiles({"test"})
public abstract class BaseIntegrationTest {

@MockitoBean
protected FileDownloadService fileDownloadService;

通过@MockitoBean模拟FileDownloadService，避免依赖外部不可控的下载URL。

问题现状 #

• 单测单个集成测试用例时，FileDownloadService的Mock行为完全符合预期；
• 批量运行多个测试用例时，依赖@KafkaListener的测试出现异常；
• 排查确认：Kafka消费者listen方法中使用的FileDownloadService实例，和测试类中注入的Mock实例并非同一个；
• 推测原因：@KafkaListener对应的Bean在Spring上下文启动初期就绑定了初始的FileDownloadService实例，后续测试切换上下文时，消费者未重新绑定新的Mock实例；
• 已尝试@DirtiesContext等方案，无法解决多实例冲突问题。

可行解决办法 #

1. 强制Kafka容器在测试前重新初始化 #

在测试类中注入KafkaListenerEndpointRegistry，每个测试方法执行前重启Kafka容器，确保消费者重新绑定最新的Mock实例：

@Autowired
private KafkaListenerEndpointRegistry kafkaListenerEndpointRegistry;

@BeforeEach
void setUp() {
    // 停止所有Kafka监听容器
    kafkaListenerEndpointRegistry.getListenerContainers().forEach(MessageListenerContainer::stop);
    // 重启容器，触发Bean重新注入
    kafkaListenerEndpointRegistry.getListenerContainers().forEach(MessageListenerContainer::start);
    // 重置Mock行为，避免测试间干扰
    Mockito.reset(fileDownloadService);
}

2. 改用@Mock手动注册Mock实例（调整Bean加载顺序） #

如果@MockitoBean的实例替换时机晚于Kafka消费者初始化，可手动创建Mock并提前注册到上下文：

@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
@ActiveProfiles({"test"})
public abstract class BaseIntegrationTest {

    @Mock
    protected FileDownloadService fileDownloadService;

    @TestConfiguration
    static class TestConfig {
        @Bean
        public FileDownloadService fileDownloadService() {
            return Mockito.mock(FileDownloadService.class);
        }
    }

    @Autowired
    protected void injectMocks(FileDownloadService service) {
        this.fileDownloadService = service;
    }
}

这种方式能确保Mock实例在Spring上下文初始化时就被注册，Kafka消费者启动时直接绑定该Mock实例。

3. 禁用Kafka容器自动启动，手动控制启动时机 #

在测试配置中禁用Kafka容器自动启动，待Mock实例完全准备好后再启动容器：

@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
@ActiveProfiles({"test"})
@EnableKafkaListener(autoStartup = "false") // 禁用容器自动启动
public abstract class BaseIntegrationTest {

    @MockitoBean
    protected FileDownloadService fileDownloadService;

    @Autowired
    private KafkaListenerEndpointRegistry kafkaListenerEndpointRegistry;

    @BeforeEach
    void setUp() {
        // 启动容器前确保Mock已就绪
        kafkaListenerEndpointRegistry.getListenerContainers().forEach(container -> {
            container.start();
            // 等待容器完全启动，避免消费逻辑提前执行
            container.awaitStart(Duration.ofSeconds(5));
        });
        Mockito.reset(fileDownloadService);
    }

    @AfterEach
    void tearDown() {
        kafkaListenerEndpointRegistry.getListenerContainers().forEach(MessageListenerContainer::stop);
    }
}

4. 调整@DirtiesContext的刷新时机 #

如果之前使用默认的AFTER_CLASS模式，改为每个测试方法后刷新上下文，确保每个测试都使用全新的上下文和Mock实例：

@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
@ActiveProfiles({"test"})
@DirtiesContext(classMode = DirtiesContext.ClassMode.AFTER_EACH_TEST_METHOD)
public abstract class BaseIntegrationTest {

    @MockitoBean
    protected FileDownloadService fileDownloadService;
}

注意：这种方式会增加测试总执行时间，因为每个测试都要重新启动Spring上下文。

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