如何让所有Embedded Kafka测试共用单个Broker及测试上下文？

一、复用Spring测试上下文 #

Spring测试框架会根据测试类的配置（注解、导入类等）自动缓存上下文，只要让所有测试类的核心配置保持一致，就能避免重复创建上下文：

• 把公共测试注解（@SpringBootTest、@ActiveProfiles、@ExtendWith等）提取到抽象基础测试类中，所有业务测试类直接继承该类。
• 统一@SpringBootTest的classes参数，不要在每个测试类中指定不同的启动类。
• 将公共的测试配置（如KafkaTemplate的基础配置）放到基础类的@TestConfiguration中，避免每个测试类单独导入不同配置。

二、共用EmbeddedKafka Broker #

默认@EmbeddedKafka会为每个测试类创建独立Broker，要实现共用，可通过以下方式：

1. 开启Broker共享（Spring Kafka 2.3+支持） #

在基础测试类上添加@EmbeddedKafka并设置shared = true，这样所有继承类会复用同一个Broker实例：

@ExtendWith(SpringExtension::class)
@SpringBootTest(classes = [Application::class])
@TestInstance(TestInstance.Lifecycle.PER_CLASS)
@EmbeddedKafka(
    partitions = 1,
    shared = true // 关键：开启Broker实例共享
)
@ActiveProfiles("test")
@Import(value = [BaseKafkaTest.BaseTestConfig::class])
abstract class BaseKafkaTest {

    @Autowired
    protected lateinit var kafkaEmbeddedKafka: EmbeddedKafkaBroker

    @Autowired
    protected lateinit var kafkaListenerEndpointRegistry: KafkaListenerEndpointRegistry

    @BeforeEach
    open fun setup() {
        // 统一等待监听器完成分区分配
        for (msgListenerContainer in kafkaListenerEndpointRegistry.listenerContainers) {
            ContainerTestUtils.waitForAssignment(msgListenerContainer, kafkaEmbeddedKafka.partitionsPerTopic)
        }
    }

    @TestConfiguration
    class BaseTestConfig {
        // 公共KafkaTemplate配置，不绑定固定Topic
        @Bean
        @Primary
        fun testKafkaTemplate(broker: EmbeddedKafkaBroker): KafkaTemplate<String, String> {
            return KafkaTemplate(DefaultKafkaProducerFactory(KafkaTestUtils.producerProps(broker)))
        }
    }
}

2. 动态创建独立Topic（满足测试隔离需求） #

因为共用了Broker，不能在@EmbeddedKafka中固定写死Topic，需在每个测试类中动态生成专属Topic：

class ApplicationListenerCaseTest : BaseKafkaTest() {

    // 为当前测试生成唯一Topic，保证测试隔离
    private val testTopic = "topic-${UUID.randomUUID()}"

    @Autowired
    private lateinit var testKafkaTemplate: KafkaTemplate<String, String>

    @BeforeEach
    override fun setup() {
        super.setup()
        // 动态创建当前测试的专属Topic
        kafkaEmbeddedKafka.addTopics(testTopic)
        // 绑定KafkaTemplate的默认Topic为当前测试的专属Topic
        testKafkaTemplate.defaultTopic = testTopic
    }

    @Test
    fun test() {
        // 使用专属Topic发送测试消息
        testKafkaTemplate.sendDefault("test message")
        // ... 业务测试逻辑
    }
}

3. 测试后清理（保证隔离性） #

如果需要彻底隔离测试数据，可在@AfterEach中清理当前测试的Topic数据：

@AfterEach
fun cleanUp() {
    // 清空当前Topic的所有消息
    KafkaTestUtils.clearTopic(kafkaEmbeddedKafka, testTopic)
    // 可选：关闭当前测试关联的监听器容器
    kafkaListenerEndpointRegistry.listenerContainers.forEach {
        if (it.destinationTopics.contains(testTopic)) {
            it.stop()
        }
    }
}

三、额外优化建议 #

• 确保测试类配置无差异：只要所有测试类的注解、导入配置一致，Spring会自动缓存上下文，避免重复初始化。
• 缩小上下文范围：如果不需要加载完整应用上下文，可改用@ContextConfiguration指定必要的配置类，替代@SpringBootTest，进一步提升启动速度。
• 复用Producer/Consumer：在基础类中创建全局的Producer/Consumer实例，避免每个测试重复创建客户端。

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