counter: 1.2.10 - Replaced helper-class `Message` with `KeyValue`

[demos/kafka/wordcount] / src / test / java / de / juplo / kafka / wordcount / counter / TestData.java

diff --git a/src/test/java/de/juplo/kafka/wordcount/counter/TestData.java b/src/test/java/de/juplo/kafka/wordcount/counter/TestData.java
index 4b5482b..5798fc3 100644 (file)
--- a/src/test/java/de/juplo/kafka/wordcount/counter/TestData.java
+++ b/src/test/java/de/juplo/kafka/wordcount/counter/TestData.java
@@ -1,9 +1,6 @@
 package de.juplo.kafka.wordcount.counter;
 
-import com.fasterxml.jackson.core.JsonProcessingException;
-import com.fasterxml.jackson.databind.ObjectMapper;
-import org.springframework.util.LinkedMultiValueMap;
-import org.springframework.util.MultiValueMap;
+import org.apache.kafka.streams.KeyValue;
 
 import java.util.List;
 import java.util.function.BiConsumer;
@@ -13,97 +10,98 @@ import static org.assertj.core.api.Assertions.assertThat;
 
 class TestData
 {
-       static void writeInputData(BiConsumer<String, String> consumer)
+       static void writeInputData(BiConsumer<String, Word> consumer)
        {
                consumer.accept(
                                "peter",
-                               "Hallo");
+                               Word.of("peter","Hallo"));
                consumer.accept(
                                "klaus",
-                               "Müsch");
+                               Word.of("klaus","Müsch"));
                consumer.accept(
                                "peter",
-                               "Welt");
+                               Word.of("peter","Welt"));
                consumer.accept(
                                "klaus",
-                               "Müsch");
+                               Word.of("klaus","Müsch"));
                consumer.accept(
                                "klaus",
-                               "s");
+                               Word.of("klaus","s"));
                consumer.accept(
                                "peter",
-                               "Boäh");
+                               Word.of("peter","Boäh"));
                consumer.accept(
                                "peter",
-                               "Welt");
+                               Word.of("peter","Welt"));
                consumer.accept(
                                "peter",
-                               "Boäh");
+                               Word.of("peter","Boäh"));
                consumer.accept(
                                "klaus",
-                               "s");
+                               Word.of("klaus","s"));
                consumer.accept(
                                "peter",
-                               "Boäh");
+                               Word.of("peter","Boäh"));
                consumer.accept(
                                "klaus",
-                               "s");
+                               Word.of("klaus","s"));
        }
 
-       static void assertExpectedResult(List<Message> receivedMessages, ObjectMapper mapper)
+       static void assertExpectedResult(List<KeyValue<Word, WordCount>> receivedMessages)
        {
-               MultiValueMap<String, Message> received = new LinkedMultiValueMap<>();
-
-               receivedMessages.forEach(message ->
-               {
-                       try
-                       {
-                               Key key = mapper.readValue(message.getKey(), Key.class);
-                               received.add(key.getUsername(), message);
-                       }
-                       catch (JsonProcessingException e)
-                       {
-                               throw new RuntimeException(e);
-                       }
-               });
-
-               assertThat(received).hasSize(2);
-               assertThat(received.get("klaus")).containsExactly(klaus1, klaus2, klaus3, klaus4, klaus5);
-               assertThat(received.get("peter")).containsExactly(peter1, peter2, peter3, peter4, peter5, peter6);
+               assertThat(receivedMessages).hasSize(11);
+               assertThat(receivedMessages).containsSubsequence(
+                               expectedMessages[0]); // Hallo
+               assertThat(receivedMessages).containsSubsequence(
+                               expectedMessages[1],
+                               expectedMessages[3]); // Müsch
+               assertThat(receivedMessages).containsSubsequence(
+                               expectedMessages[2],
+                               expectedMessages[6]);
+               assertThat(receivedMessages).containsSubsequence(
+                               expectedMessages[4],
+                               expectedMessages[8],
+                               expectedMessages[10]); // s
+               assertThat(receivedMessages).containsSubsequence(
+                               expectedMessages[5],
+                               expectedMessages[7],
+                               expectedMessages[9]); // Boäh
        }
 
-       static Message peter1 = Message.of(
-                       "{\"username\":\"peter\",\"word\":\"Hallo\"}",
-                       "1");
-       static Message peter2 = Message.of(
-                       "{\"username\":\"peter\",\"word\":\"Welt\"}",
-                       "1");
-       static Message peter3 = Message.of(
-                       "{\"username\":\"peter\",\"word\":\"Boäh\"}",
-                       "1");
-       static Message peter4 = Message.of(
-                       "{\"username\":\"peter\",\"word\":\"Welt\"}",
-                       "2");
-       static Message peter5 = Message.of(
-                       "{\"username\":\"peter\",\"word\":\"Boäh\"}",
-                       "2");
-       static Message peter6 = Message.of(
-                       "{\"username\":\"peter\",\"word\":\"Boäh\"}",
-                       "3");
-
-       static Message klaus1 = Message.of(
-                       "{\"username\":\"klaus\",\"word\":\"Müsch\"}",
-                       "1");
-       static Message klaus2 = Message.of(
-                       "{\"username\":\"klaus\",\"word\":\"Müsch\"}",
-                       "2");
-       static Message klaus3 = Message.of(
-                       "{\"username\":\"klaus\",\"word\":\"s\"}",
-                       "1");
-       static Message klaus4 = Message.of(
-                       "{\"username\":\"klaus\",\"word\":\"s\"}",
-                       "2");
-       static Message klaus5 = Message.of(
-                       "{\"username\":\"klaus\",\"word\":\"s\"}",
-                       "3");
+       static KeyValue<Word,WordCount>[] expectedMessages = new KeyValue[]
+       {
+                       KeyValue.pair(
+                                       Word.of("peter","Hallo"),
+                                       WordCount.of("peter","Hallo",1)),
+                       KeyValue.pair(
+                                       Word.of("klaus","Müsch"),
+                                       WordCount.of("klaus","Müsch",1)),
+                       KeyValue.pair(
+                                       Word.of("peter","Welt"),
+                                       WordCount.of("peter","Welt",1)),
+                       KeyValue.pair(
+                                       Word.of("klaus","Müsch"),
+                                       WordCount.of("klaus","Müsch",2)),
+                       KeyValue.pair(
+                                       Word.of("klaus","s"),
+                                       WordCount.of("klaus","s",1)),
+                       KeyValue.pair(
+                                       Word.of("peter","Boäh"),
+                                       WordCount.of("peter","Boäh",1)),
+                       KeyValue.pair(
+                                       Word.of("peter","Welt"),
+                                       WordCount.of("peter","Welt",2)),
+                       KeyValue.pair(
+                                       Word.of("peter","Boäh"),
+                                       WordCount.of("peter","Boäh",2)),
+                       KeyValue.pair(
+                                       Word.of("klaus","s"),
+                                       WordCount.of("klaus","s",2)),
+                       KeyValue.pair(
+                                       Word.of("peter","Boäh"),
+                                       WordCount.of("peter","Boäh",3)),
+                       KeyValue.pair(
+                                       Word.of("klaus","s"),
+                                       WordCount.of("klaus","s",3)),
+       };
 }