X-Git-Url: https://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fde%2Fjuplo%2Fkafka%2Fwordcount%2Ftop10%2FTestData.java;h=fb030e39479145e561e834b2c62d8320401d24c9;hb=87f82fe35276666d298bc5100f0810b6aa6ce2d4;hp=f5218698b00c1a5b572185c1ccbbe10c357dcf12;hpb=1bd7726318005a9ed24a8c14b45311a05fa1a937;p=demos%2Fkafka%2Fwordcount 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 f521869..fb030e3 100644 --- a/src/test/java/de/juplo/kafka/wordcount/top10/TestData.java +++ b/src/test/java/de/juplo/kafka/wordcount/top10/TestData.java @@ -15,44 +15,44 @@ import static org.assertj.core.api.Assertions.assertThat; class TestData { - static void writeInputData(BiConsumer consumer) + static void writeInputData(BiConsumer consumer) { consumer.accept( - Word.of("peter","Hallo"), + Key.of("peter","Hallo"), Counter.of("peter","Hallo",1)); consumer.accept( - Word.of("klaus","Müsch"), + Key.of("klaus","Müsch"), Counter.of("klaus","Müsch",1)); consumer.accept( - Word.of("peter","Welt"), + Key.of("peter","Welt"), Counter.of("peter","Welt",1)); consumer.accept( - Word.of("klaus","Müsch"), + Key.of("klaus","Müsch"), Counter.of("klaus","Müsch",2)); consumer.accept( - Word.of("klaus","s"), + Key.of("klaus","s"), Counter.of("klaus","s",1)); consumer.accept( - Word.of("peter","Boäh"), + Key.of("peter","Boäh"), Counter.of("peter","Boäh",1)); consumer.accept( - Word.of("peter","Welt"), + Key.of("peter","Welt"), Counter.of("peter","Welt",2)); consumer.accept( - Word.of("peter","Boäh"), + Key.of("peter","Boäh"), Counter.of("peter","Boäh",2)); consumer.accept( - Word.of("klaus","s"), + Key.of("klaus","s"), Counter.of("klaus","s",2)); consumer.accept( - Word.of("peter","Boäh"), + Key.of("peter","Boäh"), Counter.of("peter","Boäh",3)); consumer.accept( - Word.of("klaus","s"), + Key.of("klaus","s"), Counter.of("klaus","s",3)); } - static void assertExpectedResult(List> receivedMessages) + static void assertExpectedResult(List> receivedMessages) { assertThat(receivedMessages).hasSize(11); assertThat(receivedMessages).containsSubsequence( @@ -73,42 +73,65 @@ class TestData expectedMessages[9]); // Boäh } - static KeyValue[] expectedMessages = new KeyValue[] - { - KeyValue.pair( - Word.of("peter","Hallo"), - Counter.of("peter","Hallo",1)), - KeyValue.pair( - Word.of("klaus","Müsch"), - Counter.of("klaus","Müsch",1)), - KeyValue.pair( - Word.of("peter","Welt"), - Counter.of("peter","Welt",1)), - KeyValue.pair( - Word.of("klaus","Müsch"), - Counter.of("klaus","Müsch",2)), - KeyValue.pair( - Word.of("klaus","s"), - Counter.of("klaus","s",1)), - KeyValue.pair( - Word.of("peter","Boäh"), - Counter.of("peter","Boäh",1)), - KeyValue.pair( - Word.of("peter","Welt"), - Counter.of("peter","Welt",2)), - KeyValue.pair( - Word.of("peter","Boäh"), - Counter.of("peter","Boäh",2)), - KeyValue.pair( - Word.of("klaus","s"), - Counter.of("klaus","s",2)), - KeyValue.pair( - Word.of("peter","Boäh"), - Counter.of("peter","Boäh",3)), - KeyValue.pair( - Word.of("klaus","s"), - Counter.of("klaus","s",3)), - }; + static KeyValue[] expectedMessages = new KeyValue[] + { + KeyValue.pair( + "peter", + Ranking.of( + Entry.of("Hallo", 1l))), + KeyValue.pair( + "klaus", + Ranking.of( + Entry.of("Müsch", 1l))), + KeyValue.pair( + "peter", + Ranking.of( + Entry.of("Hallo", 1l), + Entry.of("Welt", 1l))), + KeyValue.pair( + "klaus", + Ranking.of( + Entry.of("Müsch", 2l))), + KeyValue.pair( + "klaus", + Ranking.of( + Entry.of("Müsch", 2l), + Entry.of("s", 1l))), + KeyValue.pair( + "peter", + Ranking.of( + Entry.of("Hallo", 1l), + Entry.of("Welt", 1l), + Entry.of("Boäh", 1l))), + KeyValue.pair( + "peter", + Ranking.of( + Entry.of("Welt", 2l), + Entry.of("Hallo", 1l), + Entry.of("Boäh", 1l))), + KeyValue.pair( + "peter", + Ranking.of( + Entry.of("Welt", 2l), + Entry.of("Boäh", 2l), + Entry.of("Hallo", 1l))), + KeyValue.pair( + "klaus", + Ranking.of( + Entry.of("Müsch", 2l), + Entry.of("s", 2l))), + KeyValue.pair( + "peter", + Ranking.of( + Entry.of("Boäh", 3l), + Entry.of("Welt", 2l), + Entry.of("Hallo", 1l))), + KeyValue.pair( + "klaus", + Ranking.of( + Entry.of("s", 3l), + Entry.of("Müsch", 2l))), + }; static Map convertToMap(Properties properties) {