X-Git-Url: http://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fde%2Fjuplo%2Fkafka%2FGenericApplicationTests.java;h=49ddb47e1f2db54dc9cf3e570c54a14cbfe5591a;hb=0c9a0c1cf9a0065012743efcd940d8721bc33c20;hp=124143c32ad92a35dcbae0f9994a3e65997f348e;hpb=316c89aa6aafbc339eda3727638c75f5489c0a99;p=demos%2Fkafka%2Ftraining

diff --git a/src/test/java/de/juplo/kafka/GenericApplicationTests.java b/src/test/java/de/juplo/kafka/GenericApplicationTests.java
index 124143c..49ddb47 100644
--- a/src/test/java/de/juplo/kafka/GenericApplicationTests.java
+++ b/src/test/java/de/juplo/kafka/GenericApplicationTests.java
@@ -2,7 +2,6 @@ package de.juplo.kafka;
 
 import com.mongodb.client.MongoClient;
 import lombok.extern.slf4j.Slf4j;
-import org.apache.kafka.clients.consumer.ConsumerRecord;
 import org.apache.kafka.clients.consumer.KafkaConsumer;
 import org.apache.kafka.clients.producer.KafkaProducer;
 import org.apache.kafka.clients.producer.ProducerRecord;
@@ -69,9 +68,9 @@ abstract class GenericApplicationTests<K, V>
 	@Autowired
 	KafkaListenerEndpointRegistry kafkaListenerEndpointRegistry;
 	@Autowired
-	TestRecordHandler<K, V> recordHandler;
+	TestRecordHandler recordHandler;
 	@Autowired
-	EndlessConsumer<K, V> endlessConsumer;
+	EndlessConsumer endlessConsumer;
 
 	KafkaProducer<Bytes, Bytes> testRecordProducer;
 	KafkaConsumer<Bytes, Bytes> offsetConsumer;
@@ -99,7 +98,7 @@ abstract class GenericApplicationTests<K, V>
 		await(numberOfGeneratedMessages + " records received")
 				.atMost(Duration.ofSeconds(30))
 				.pollInterval(Duration.ofSeconds(1))
-				.until(() -> recordHandler.receivedRecords.size() >= numberOfGeneratedMessages);
+				.until(() -> recordHandler.receivedMessages >= numberOfGeneratedMessages);
 
 		await("Offsets committed")
 				.atMost(Duration.ofSeconds(10))
@@ -141,7 +140,7 @@ abstract class GenericApplicationTests<K, V>
 
 		checkSeenOffsetsForProgress();
 		assertSeenOffsetsEqualCommittedOffsets(recordHandler.seenOffsets);
-		assertThat(recordHandler.receivedRecords.size())
+		assertThat(recordHandler.receivedMessages)
 				.describedAs("Received not all sent events")
 				.isLessThan(numberOfGeneratedMessages);
 
@@ -348,7 +347,7 @@ abstract class GenericApplicationTests<K, V>
 
 		oldOffsets = new HashMap<>();
 		recordHandler.seenOffsets = new HashMap<>();
-		recordHandler.receivedRecords = new HashSet<>();
+		recordHandler.receivedMessages = 0;
 
 		doForCurrentOffsets((tp, offset) ->
 		{