frankvicky commented on code in PR #16227:
URL: https://github.com/apache/kafka/pull/16227#discussion_r1631832885
##########
clients/src/test/java/org/apache/kafka/clients/consumer/ConsumerRecordsTest.java:
##########
@@ -59,4 +49,72 @@ public void iterator() throws Exception {
}
assertEquals(2, c);
}
+
+ @Test
+ public void testRecordsWithNullTopic() {
+ String nullTopic = null;
+ Map<TopicPartition, List<ConsumerRecord<Integer, String>>> records =
new LinkedHashMap<>();
+ ConsumerRecords<Integer, String> consumerRecords = new
ConsumerRecords<>(records);
+ IllegalArgumentException exception =
assertThrows(IllegalArgumentException.class, () ->
consumerRecords.records(nullTopic));
+ assertEquals("Topic must be non-null.", exception.getMessage());
+ }
+
+ @Test
+ public void testRecords() {
+ String[] topics = {"topic1", "topic2", "topic3", "topic4"};
+ int recordSize = 3;
+ Map<TopicPartition, List<ConsumerRecord<Integer, String>>>
partitionToRecords = buildTopicTestRecords(recordSize, topics);
+ ConsumerRecords<Integer, String> consumerRecords = new
ConsumerRecords<>(partitionToRecords);
+ for (String topic : topics) {
+ Iterable<ConsumerRecord<Integer, String>> records =
consumerRecords.records(topic);
+
+ int count = 0;
+ Iterator<ConsumerRecord<Integer, String>> iterator =
records.iterator();
+ for (; iterator.hasNext() && count < recordSize; count++) {
+ ConsumerRecord<Integer, String> record = iterator.next();
+ assertEquals(topic, record.topic());
+ assertEquals(count, record.partition());
+ assertEquals(count, record.offset());
+ assertEquals(count, record.key());
+ assertEquals(String.valueOf(count), record.value());
+ }
+ }
+ }
+
+ private Map<TopicPartition, List<ConsumerRecord<Integer, String>>>
buildSingleTopicTestRecords(String topic) {
+ Map<TopicPartition, List<ConsumerRecord<Integer, String>>> records =
new LinkedHashMap<>();
+ ConsumerRecord<Integer, String> record1 = new ConsumerRecord<>(topic,
1, 0, 0L, TimestampType.CREATE_TIME,
+ 0, 0, 1, "value1", new RecordHeaders(), Optional.empty());
+ ConsumerRecord<Integer, String> record2 = new ConsumerRecord<>(topic,
1, 1, 0L, TimestampType.CREATE_TIME,
+ 0, 0, 2, "value2", new RecordHeaders(), Optional.empty());
+
+ new ArrayList<>();
Review Comment:
Oops, I accidentally leave this, I will remove it.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: jira-unsubscr...@kafka.apache.org
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
