frankvicky commented on code in PR #16227:
URL: https://github.com/apache/kafka/pull/16227#discussion_r1631833896
##########
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) {
Review Comment:
I keep this because the test case `iterator` has it' own test logic but I
will try to rewrite 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
