使用KafkaTemplate的生产者服务能否用MockProducer做单元测试？

你的问题在于KafkaProducer依赖的是KafkaTemplate而非直接的Producer接口，所以不能直接传入MockProducer。解决方法是把MockProducer封装到KafkaTemplate实例中，再注入到KafkaProducer里。

完整测试代码示例 #

import org.apache.kafka.clients.producer.MockProducer
import org.apache.kafka.common.serialization.StringSerializer
import org.springframework.kafka.core.DefaultKafkaProducerFactory
import org.springframework.kafka.core.KafkaTemplate
import org.junit.Test
import kotlin.test.assertEquals
import kotlin.test.assertNotNull

@Test
fun verifyMessageSend() {
    // 1. 创建MockProducer实例
    val mockProducer = MockProducer(true, StringSerializer(), StringSerializer())
    
    // 2. 创建ProducerFactory，指定使用MockProducer
    val producerFactory = DefaultKafkaProducerFactory<String, String>(emptyMap()) { mockProducer }
    
    // 3. 基于MockProducer的Factory创建KafkaTemplate
    val kafkaTemplate = KafkaTemplate(producerFactory)
    
    // 4. 初始化被测试的KafkaProducer
    val kafkaProducer = KafkaProducer(kafkaTemplate)

    // 5. 执行发送操作
    val testMessage = "test_kafka_message"
    kafkaProducer.sendMessage(testMessage)

    // 6. 触发异步回调（模拟发送成功）
    mockProducer.complete()

    // 7. 验证发送结果
    val sentRecords = mockProducer.history()
    assertEquals(1, sentRecords.size, "应该只发送了一条消息")
    
    val sentRecord = sentRecords[0]
    assertEquals(AppConstants.TOPIC_NAME, sentRecord.topic(), "消息主题不匹配")
    assertEquals("abc", sentRecord.key(), "消息Key不匹配")
    assertEquals(testMessage, sentRecord.value(), "消息内容不匹配")
}

关键说明 #

• 包装MockProducer到KafkaTemplate：通过DefaultKafkaProducerFactory的producerSupplier参数，让工厂返回我们预先创建的MockProducer，这样KafkaTemplate就会使用这个Mock实例。
• 处理异步回调：因为kafkaTemplate.send()是异步操作，调用mockProducer.complete()可以触发成功回调；如果要测试失败场景，可调用mockProducer.error(RuntimeException("发送失败"))来模拟异常。
• 验证逻辑：通过mockProducer.history()可以获取所有发送的记录，从而验证消息的主题、Key、内容是否符合预期。

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