X-Git-Url: https://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fde%2Fjuplo%2Fkafka%2Fwordcount%2Ftop10%2FTestData.java;h=f4e557c6a03406a88b018325d0121b9af165d701;hb=d1ac99074b22f357152644d4593b234540d1c952;hp=d8c603ae1ba6c688376456b47702de0a91bb0b1c;hpb=e32a74e6a989ba8fc1061cffd1bed814edb4f72f;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 d8c603a..f4e557c 100644
--- a/src/test/java/de/juplo/kafka/wordcount/top10/TestData.java
+++ b/src/test/java/de/juplo/kafka/wordcount/top10/TestData.java
@@ -9,9 +9,6 @@ import org.apache.kafka.streams.state.ReadOnlyKeyValueStore;
 import org.springframework.util.LinkedMultiValueMap;
 import org.springframework.util.MultiValueMap;
 
-import java.util.Map;
-import java.util.Properties;
-import java.util.stream.Collectors;
 import java.util.stream.Stream;
 
 import static org.assertj.core.api.Assertions.assertThat;
@@ -69,8 +66,44 @@ class TestData
 
 	static void assertExpectedState(ReadOnlyKeyValueStore<User, Ranking> store)
 	{
-		assertThat(store.get(EXPECTED_MESSAGES[9].key)).isEqualTo(EXPECTED_MESSAGES[9].value);
-		assertThat(store.get(EXPECTED_MESSAGES[10].key)).isEqualTo(EXPECTED_MESSAGES[10].value);
+		assertRankingEqualsRankingFromLastMessage(PETER, store.get(PETER));
+		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));
+	}
+
+	static Ranking getLastMessageFor(User user)
+	{
+		return getLastMessageFor(user, expectedMessages());
+	}
+
+	static Ranking getLastMessageFor(User user, MultiValueMap<User, Ranking> messagesForUsers)
+	{
+		return messagesForUsers
+				.get(user)
+				.stream()
+				.reduce(null, (left, right) -> right);
 	}
 
 	static KeyValue<User, Ranking>[] EXPECTED_MESSAGES = new KeyValue[]
@@ -142,18 +175,6 @@ class TestData
 		return expectedMessages;
 	}
 
-	static Map<String, Object> convertToMap(Properties properties)
-	{
-		return properties
-				.entrySet()
-				.stream()
-				.collect(
-						Collectors.toMap(
-								entry -> (String)entry.getKey(),
-								entry -> entry.getValue()
-						));
-	}
-
 	static String parseHeader(Headers headers, String key)
 	{
 		Header header = headers.lastHeader(key);