From 843682fed975e5ee2ee6871e95d1cb6c420b0d9e Mon Sep 17 00:00:00 2001
From: Kai Moritz <kai@juplo.de>
Date: Sun, 2 Jun 2024 11:00:36 +0200
Subject: [PATCH] top10: 1.2.1 - Refined assertion of expected state

---
 .../juplo/kafka/wordcount/top10/TestData.java | 22 +++++++++++++++++--
 1 file changed, 20 insertions(+), 2 deletions(-)

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..eb30298 100644
--- a/src/test/java/de/juplo/kafka/wordcount/top10/TestData.java
+++ b/src/test/java/de/juplo/kafka/wordcount/top10/TestData.java
@@ -69,8 +69,26 @@ 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 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[]
-- 
2.20.1