Reuse iterator to return list adhering to table semantics in case of

TableOutput
bakdata · Apr 11, 2024 · 7461741 · 7461741
1 parent b1ed9cb
commit 7461741
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/...-kafka-streams-tests/src/main/java/com/bakdata/fluent_kafka_streams_tests/BaseOutput.java b/...-kafka-streams-tests/src/main/java/com/bakdata/fluent_kafka_streams_tests/BaseOutput.java
@@ -24,8 +24,8 @@
 
 package com.bakdata.fluent_kafka_streams_tests;
 
+import java.util.ArrayList;
 import java.util.List;
-import java.util.stream.Collectors;
 import org.apache.kafka.clients.producer.ProducerRecord;
 import org.apache.kafka.common.serialization.Serde;
 import org.apache.kafka.streams.TestOutputTopic;
@@ -118,12 +118,9 @@ public TestOutput<K, V> asStream() {
 
     @Override
     public List<ProducerRecord<K, V>> toList() {
-        return this.testOutputTopic
-                .readRecordsToList()
-                .stream()
-                .map(testRecord -> new ProducerRecord<>(this.topic, 0, testRecord.timestamp(), testRecord.key(),
-                        testRecord.value(), testRecord.getHeaders()))
-                .collect(Collectors.toList());
+        final List<ProducerRecord<K, V>> list = new ArrayList<>();
+        this.iterator().forEachRemaining(list::add);
+        return list;
     }
 
     // ==================