X-Git-Url: https://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fde%2Fjuplo%2Fkafka%2Fwordcount%2Fquery%2FTestData.java;h=1fe34d9eecf50c609910dd02be5c5d0b919ef986;hb=57f47fca712981116b726e437f589ac727c5d0a7;hp=3fcd7c9638baca6c839238762a2a74b112ae9513;hpb=dcb383aef4f4f454e3a1aed9cf1f2a599c7b1bb9;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..1fe34d9 100644
--- a/src/test/java/de/juplo/kafka/wordcount/query/TestData.java
+++ b/src/test/java/de/juplo/kafka/wordcount/query/TestData.java
@@ -1,6 +1,5 @@
 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.users.TestUserData;
@@ -15,7 +14,6 @@ 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";
 
@@ -29,32 +27,15 @@ 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));
 	}
 
-	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)