X-Git-Url: https://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fde%2Fjuplo%2Fkafka%2Fwordcount%2Fcounter%2FCounterApplicationIT.java;h=cb0a5b78ce7a8d00b7320e9d89f7b93531caba51;hb=236aa3a7521bf44e2e397fa7b90373cc3cf59341;hp=31bf6a6d6c1f898793e51f9481f4a07d295687aa;hpb=3a0dbf30eaee8aade255360217d7ab6049a95041;p=demos%2Fkafka%2Fwordcount

diff --git a/src/test/java/de/juplo/kafka/wordcount/counter/CounterApplicationIT.java b/src/test/java/de/juplo/kafka/wordcount/counter/CounterApplicationIT.java
index 31bf6a6..cb0a5b7 100644
--- a/src/test/java/de/juplo/kafka/wordcount/counter/CounterApplicationIT.java
+++ b/src/test/java/de/juplo/kafka/wordcount/counter/CounterApplicationIT.java
@@ -69,11 +69,11 @@ public class CounterApplicationIT
 	{
 		TestData
 				.getInputMessages()
-				.forEach(word -> kafkaTemplate.send(TOPIC_IN, word.getUser(), word));
+				.forEach(kv -> kafkaTemplate.send(TOPIC_IN, kv.key, kv.value));
 
 		await("Expected messages")
 				.atMost(Duration.ofSeconds(10))
-				.untilAsserted(() -> TestData.assertExpectedMessages(consumer.getReceivedMessages()));
+				.untilAsserted(() -> consumer.enforceAssertion(TestData.expectedMessagesAssertion()));
 	}
 
 
@@ -90,9 +90,10 @@ public class CounterApplicationIT
 			received.add(word, counter);
 		}
 
-		synchronized MultiValueMap<TestOutputWord, TestOutputWordCounter> getReceivedMessages()
+		synchronized void enforceAssertion(
+				java.util.function.Consumer<MultiValueMap<TestOutputWord, TestOutputWordCounter>> assertion)
 		{
-			return received;
+			assertion.accept(received);
 		}
 	}