X-Git-Url: https://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fde%2Fjuplo%2Fkafka%2Fwordcount%2Ftop10%2FTestData.java;h=a4f79ac72c882fee5a0a9e444f5f570dfe11e15c;hb=fa19058953b388f893255d2bcc9351ac9eb3328b;hp=eb302982accae749b447e74b9a54a4777be8c5df;hpb=843682fed975e5ee2ee6871e95d1cb6c420b0d9e;p=demos%2Fkafka%2Fwordcount

diff --git a/src/test/java/de/juplo/kafka/wordcount/top10/TestData.java b/src/test/java/de/juplo/kafka/wordcount/top10/TestData.java
index eb30298..a4f79ac 100644
--- a/src/test/java/de/juplo/kafka/wordcount/top10/TestData.java
+++ b/src/test/java/de/juplo/kafka/wordcount/top10/TestData.java
@@ -73,6 +73,24 @@ class TestData
 		assertRankingEqualsRankingFromLastMessage(KLAUS, store.get(KLAUS));
 	}
 
+	static void assertExpectedNumberOfMessagesForUsers(MultiValueMap<User, Ranking> receivedMessages)
+	{
+		assertThat(countMessagesForUser(PETER, receivedMessages));
+		assertThat(countMessagesForUser(KLAUS, receivedMessages));
+	}
+
+	static int countMessagesForUser(User user, MultiValueMap<User, Ranking> messagesForUsers)
+	{
+		return messagesForUsers.get(user).size();
+	}
+
+
+	static void assertExpectedLastMessagesForUsers(MultiValueMap<User, Ranking> receivedMessages)
+	{
+		assertRankingEqualsRankingFromLastMessage(PETER, getLastMessageFor(PETER, receivedMessages));
+		assertRankingEqualsRankingFromLastMessage(KLAUS, getLastMessageFor(KLAUS, receivedMessages));
+	}
+
 	static void assertRankingEqualsRankingFromLastMessage(User user, Ranking ranking)
 	{
 		assertThat(ranking).isEqualTo(getLastMessageFor(user));