X-Git-Url: https://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fde%2Fjuplo%2Fkafka%2Fwordcount%2Fquery%2FTestData.java;h=3fcd7c9638baca6c839238762a2a74b112ae9513;hb=cfd58858b9861e8d0e4c1d30896505a50f63255b;hp=610bca04c05892a37641e79eb15ad1b6ca447d7f;hpb=2b22a006cc57203406c8589687a6c729ebdbf40c;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 610bca0..3fcd7c9 100644
--- a/src/test/java/de/juplo/kafka/wordcount/query/TestData.java
+++ b/src/test/java/de/juplo/kafka/wordcount/query/TestData.java
@@ -5,9 +5,9 @@ import de.juplo.kafka.wordcount.top10.TestEntry;
 import de.juplo.kafka.wordcount.top10.TestRanking;
 import de.juplo.kafka.wordcount.users.TestUserData;
 import org.apache.kafka.streams.KeyValue;
-import org.apache.kafka.streams.state.ReadOnlyKeyValueStore;
 
 import java.util.Arrays;
+import java.util.function.Function;
 import java.util.stream.Stream;
 
 import static org.assertj.core.api.Assertions.assertThat;
@@ -29,10 +29,10 @@ class TestData
 		return Stream.of(USERS_MESSAGES);
 	}
 
-	static void assertExpectedState(ReadOnlyKeyValueStore<String, String> store)
+	static void assertExpectedState(Function<String, String> function)
 	{
-		assertRankingEqualsRankingFromLastMessage(PETER, store.get(PETER));
-		assertRankingEqualsRankingFromLastMessage(KLAUS, store.get(KLAUS));
+		assertRankingEqualsRankingFromLastMessage(PETER, function.apply(PETER));
+		assertRankingEqualsRankingFromLastMessage(KLAUS, function.apply(KLAUS));
 	}
 
 	private static void assertRankingEqualsRankingFromLastMessage(String user, String userRankingJson)