X-Git-Url: https://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fde%2Fjuplo%2Fkafka%2Fwordcount%2Fquery%2FTestData.java;h=44162a0d9ab760ca2e9c9a607661f91e99511842;hb=d835b70fc4d49a50f42da9c21b3de1dbcd18cbaf;hp=c190eed24af4ad24392e25ec2e64b388c6db6085;hpb=700f80444d14b201f7b696fb5b7bcab0d767f007;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 c190eed..44162a0 100644
--- a/src/test/java/de/juplo/kafka/wordcount/query/TestData.java
+++ b/src/test/java/de/juplo/kafka/wordcount/query/TestData.java
@@ -10,13 +10,14 @@ import java.util.Arrays;
 import java.util.function.Function;
 import java.util.stream.Stream;
 
+import static de.juplo.kafka.wordcount.query.QueryStreamProcessor.STATS_TYPE;
 import static org.assertj.core.api.Assertions.assertThat;
 
 
 class TestData
 {
-	static final TestUser PETER = TestUser.of("peter");
-	static final TestUser KLAUS = TestUser.of("klaus");
+	static final TestUser PETER = TestUser.of(STATS_TYPE, "peter");
+	static final TestUser KLAUS = TestUser.of(STATS_TYPE, "klaus");
 
 	static final Stream<KeyValue<TestUser, TestRanking>> getTop10Messages()
 	{
@@ -30,8 +31,8 @@ class TestData
 
 	static void assertExpectedState(Function<String, UserRanking> function)
 	{
-		assertRankingEqualsRankingFromLastMessage(PETER.getUsername(), function.apply(PETER.getUsername()));
-		assertRankingEqualsRankingFromLastMessage(KLAUS.getUsername(), function.apply(KLAUS.getUsername()));
+		assertRankingEqualsRankingFromLastMessage(PETER.getChannel(), function.apply(PETER.getChannel()));
+		assertRankingEqualsRankingFromLastMessage(KLAUS.getChannel(), function.apply(KLAUS.getChannel()));
 	}
 
 	private static void assertRankingEqualsRankingFromLastMessage(String user, UserRanking rankingJson)
@@ -42,7 +43,7 @@ class TestData
 	private static UserRanking getLastMessageFor(String user)
 	{
 		return getTop10Messages()
-				.filter(kv -> kv.key.getUsername().equals(user))
+				.filter(kv -> kv.key.getChannel().equals(user))
 				.map(kv -> kv.value)
 				.map(testRanking -> userRankingFor(user, testRanking))
 				.reduce(null, (left, right) -> right);
@@ -69,8 +70,8 @@ class TestData
 	private static Entry entryOf(TestEntry testEntry)
 	{
 		Entry entry = new Entry();
-		entry.setWord(testEntry.getWord());
-		entry.setCount(testEntry.getCount());
+		entry.setKey(testEntry.getKey());
+		entry.setCounter(testEntry.getCounter());
 		return entry;
 	}
 	private static KeyValue<TestUser, TestRanking>[] TOP10_MESSAGES = new KeyValue[]
@@ -136,10 +137,10 @@ class TestData
 	private static KeyValue<String, TestUserData>[] USERS_MESSAGES = new KeyValue[]
 	{
 			KeyValue.pair(
-					PETER.getUsername(),
-					TestUserData.of(PETER.getUsername(), "Peter", "Pan", TestUserData.Sex.MALE)),
+					PETER.getChannel(),
+					TestUserData.of(PETER.getChannel(), "Peter", "Pan", TestUserData.Sex.MALE)),
 			KeyValue.pair(
-					KLAUS.getUsername(),
-					TestUserData.of(KLAUS.getUsername(), "Klaus", "Klüse", TestUserData.Sex.OTHER)),
+					KLAUS.getChannel(),
+					TestUserData.of(KLAUS.getChannel(), "Klaus", "Klüse", TestUserData.Sex.OTHER)),
 	};
 }