X-Git-Url: https://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fde%2Fjuplo%2Fkafka%2Fwordcount%2Fquery%2FTestData.java;h=c190eed24af4ad24392e25ec2e64b388c6db6085;hb=700f80444d14b201f7b696fb5b7bcab0d767f007;hp=3fcd7c9638baca6c839238762a2a74b112ae9513;hpb=cfd58858b9861e8d0e4c1d30896505a50f63255b;p=demos%2Fkafka%2Fwordcount

diff --git a/src/test/java/de/juplo/kafka/wordcount/query/TestData.java b/src/test/java/de/juplo/kafka/wordcount/query/TestData.java
index 3fcd7c9..c190eed 100644
--- a/src/test/java/de/juplo/kafka/wordcount/query/TestData.java
+++ b/src/test/java/de/juplo/kafka/wordcount/query/TestData.java
@@ -1,8 +1,8 @@
 package de.juplo.kafka.wordcount.query;
 
-import com.fasterxml.jackson.databind.ObjectMapper;
 import de.juplo.kafka.wordcount.top10.TestEntry;
 import de.juplo.kafka.wordcount.top10.TestRanking;
+import de.juplo.kafka.wordcount.top10.TestUser;
 import de.juplo.kafka.wordcount.users.TestUserData;
 import org.apache.kafka.streams.KeyValue;
 
@@ -15,11 +15,10 @@ import static org.assertj.core.api.Assertions.assertThat;
 
 class TestData
 {
-	static final ObjectMapper objectMapper = new ObjectMapper();
-	static final String PETER = "peter";
-	static final String KLAUS = "klaus";
+	static final TestUser PETER = TestUser.of("peter");
+	static final TestUser KLAUS = TestUser.of("klaus");
 
-	static final Stream<KeyValue<String, TestRanking>> getTop10Messages()
+	static final Stream<KeyValue<TestUser, TestRanking>> getTop10Messages()
 	{
 		return Stream.of(TOP10_MESSAGES);
 	}
@@ -29,38 +28,21 @@ class TestData
 		return Stream.of(USERS_MESSAGES);
 	}
 
-	static void assertExpectedState(Function<String, String> function)
+	static void assertExpectedState(Function<String, UserRanking> function)
 	{
-		assertRankingEqualsRankingFromLastMessage(PETER, function.apply(PETER));
-		assertRankingEqualsRankingFromLastMessage(KLAUS, function.apply(KLAUS));
+		assertRankingEqualsRankingFromLastMessage(PETER.getUsername(), function.apply(PETER.getUsername()));
+		assertRankingEqualsRankingFromLastMessage(KLAUS.getUsername(), function.apply(KLAUS.getUsername()));
 	}
 
-	private static void assertRankingEqualsRankingFromLastMessage(String user, String userRankingJson)
+	private static void assertRankingEqualsRankingFromLastMessage(String user, UserRanking rankingJson)
 	{
-		assertThat(userRankingOf(userRankingJson)).isEqualTo(getLastMessageFor(user));
-	}
-
-	private static UserRanking userRankingOf(String json)
-	{
-		if (json == null)
-		{
-			return null;
-		}
-
-		try
-		{
-			return objectMapper.readValue(json, UserRanking.class);
-		}
-		catch (Exception e)
-		{
-			throw new RuntimeException(e);
-		}
+		assertThat(rankingJson).isEqualTo(getLastMessageFor(user));
 	}
 
 	private static UserRanking getLastMessageFor(String user)
 	{
 		return getTop10Messages()
-				.filter(kv -> kv.key.equals(user))
+				.filter(kv -> kv.key.getUsername().equals(user))
 				.map(kv -> kv.value)
 				.map(testRanking -> userRankingFor(user, testRanking))
 				.reduce(null, (left, right) -> right);
@@ -91,8 +73,7 @@ class TestData
 		entry.setCount(testEntry.getCount());
 		return entry;
 	}
-
-	private static KeyValue<String, TestRanking>[] TOP10_MESSAGES = new KeyValue[]
+	private static KeyValue<TestUser, TestRanking>[] TOP10_MESSAGES = new KeyValue[]
 	{
 			KeyValue.pair( // 0
 					PETER,
@@ -155,10 +136,10 @@ class TestData
 	private static KeyValue<String, TestUserData>[] USERS_MESSAGES = new KeyValue[]
 	{
 			KeyValue.pair(
-					PETER,
-					TestUserData.of(PETER, "Peter", "Pan", TestUserData.Sex.MALE)),
+					PETER.getUsername(),
+					TestUserData.of(PETER.getUsername(), "Peter", "Pan", TestUserData.Sex.MALE)),
 			KeyValue.pair(
-					KLAUS,
-					TestUserData.of(KLAUS, "Klaus", "Klüse", TestUserData.Sex.OTHER)),
+					KLAUS.getUsername(),
+					TestUserData.of(KLAUS.getUsername(), "Klaus", "Klüse", TestUserData.Sex.OTHER)),
 	};
 }