java - Kafka - 如何模拟消费者记录？

Question

我需要一些帮助来为我的 Java kafka 消费者构建 Junit 测试用例。

我的原始源代码有如下方法，需要为其创建一个单元测试用例。

@KafkaListener(topics = "${kafka-receiver-topic}")
public void receiveTopic(ConsumerRecord<?, ?> consumerRecord) throws Exception {
    JSONObject kafkaObject = new JSONObject(consumerRecord.value().toString());
}

Answer 1

ConsumerRecord 有一个公共(public)构造函数，因此您可以自己创建一个实例。

例如，稍微修改您的代码:

主要内容:

public JSONObject receiveTopic(ConsumerRecord<?, ?> consumerRecord) throws Exception {
    JSONObject kafkaObject = new JSONObject(consumerRecord.value().toString());
    return kafkaObject;
}

测试:

@Test
public void testReceiveTopic() {
    ConsumerRecord<String, String> record = new ConsumerRecord<>("topic", 0, 123L, "key", "value");
    JSONObject expected = <what you expected>
    assertEquals(expected, receiveTopic(record));
}