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=2f1e0c3a2b2b9a02577264c7dc43c311e904dba4;hpb=44f1ad5dcd50851ef5d93b1be759481d5a38f63a;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 2f1e0c3..cb0a5b7 100644
--- a/src/test/java/de/juplo/kafka/wordcount/counter/CounterApplicationIT.java
+++ b/src/test/java/de/juplo/kafka/wordcount/counter/CounterApplicationIT.java
@@ -73,7 +73,7 @@ public class CounterApplicationIT
 
 		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);
 		}
 	}