top10: 1.2.1 - Added expectations for the number of received messages

[demos/kafka/wordcount] / src / test / java / de / juplo / kafka / wordcount / top10 / TestData.java

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 008cdde..a4f79ac 100644 (file)
--- 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,18 @@ 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));