splitter: 1.2.0 - A domain-class (``User``) is used as key

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

diff --git a/src/test/java/de/juplo/kafka/wordcount/splitter/TestData.java b/src/test/java/de/juplo/kafka/wordcount/splitter/TestData.java
index 89cebaa..f89b099 100644 (file)
--- a/src/test/java/de/juplo/kafka/wordcount/splitter/TestData.java
+++ b/src/test/java/de/juplo/kafka/wordcount/splitter/TestData.java
@@ -1,12 +1,14 @@
 package de.juplo.kafka.wordcount.splitter;
 
-import de.juplo.kafka.wordcount.counter.TestWord;
+import de.juplo.kafka.wordcount.counter.TestOutputUser;
+import de.juplo.kafka.wordcount.counter.TestOutputWord;
+import de.juplo.kafka.wordcount.recorder.TestInputRecording;
+import de.juplo.kafka.wordcount.recorder.TestInputUser;
 import org.apache.kafka.streams.KeyValue;
 import org.springframework.util.LinkedMultiValueMap;
 import org.springframework.util.MultiValueMap;
 
 import java.time.Duration;
-import java.util.function.BiConsumer;
 import java.util.stream.Stream;
 
 import static org.assertj.core.api.Assertions.assertThat;
@@ -15,71 +17,82 @@ import static org.awaitility.Awaitility.await;
 
 public class TestData
 {
-       static void writeInputData(BiConsumer<String, Recording> consumer)
-       {
-               Recording recording;
-
-               recording = new Recording();
-               recording.setUser("peter");
-               recording.setSentence("Hallo Welt!");
-               consumer.accept(recording.getUser(), recording);
+       static final TestInputUser PETER = TestInputUser.of("peter");
+       static final TestInputUser KLAUS = TestInputUser.of("klaus");
 
-               recording = new Recording();
-               recording.setUser("klaus");
-               recording.setSentence("Müsch gäb's auch!");
-               consumer.accept(recording.getUser(), recording);
 
-               recording = new Recording();
-               recording.setUser("peter");
-               recording.setSentence("Boäh, echt! ß mal nä Nümmäh!");
-               consumer.accept(recording.getUser(), recording);
+       static final Stream<KeyValue<TestInputUser, TestInputRecording>> getInputMessages()
+       {
+               return Stream.of(INPUT_MESSAGES);
        }
 
-       static void assertExpectedMessages(MultiValueMap<String, TestWord> receivedMessages)
+       private static final KeyValue<TestInputUser, TestInputRecording>[] INPUT_MESSAGES = new KeyValue[]
+       {
+                       new KeyValue<>(
+                                       PETER,
+                                       TestInputRecording.of(PETER.getUser(), "Hallo Welt!")),
+                       new KeyValue<>(
+                                       KLAUS,
+                                       TestInputRecording.of(KLAUS.getUser(), "Müsch gäb's auch!")),
+                       new KeyValue<>(
+                                       PETER,
+                                       TestInputRecording.of(PETER.getUser(), "Boäh, echt! ß mal nä Nümmäh!")),
+       };
+
+       static void assertExpectedMessages(MultiValueMap<TestOutputUser, TestOutputWord> receivedMessages)
        {
-               MultiValueMap<String, TestWord> expected = new LinkedMultiValueMap<>();
-               expectedMessages.forEach(keyValue -> expected.add(keyValue.key, keyValue.value));
                await("Received expected messages")
                                .atMost(Duration.ofSeconds(5))
-                               .untilAsserted(() -> expected.forEach((user, word) ->
+                               .untilAsserted(() -> expectedMessages().forEach((user, word) ->
                                                assertThat(receivedMessages.get(user)).containsExactlyElementsOf(word)));
        }
 
-       static Stream<KeyValue<String, TestWord>> expectedMessages = Stream.of(
+       private static final KeyValue<TestOutputUser, TestOutputWord>[] EXPECTED_MESSAGES = new KeyValue[]
+       {
                        KeyValue.pair(
-                                       "peter",
-                                       TestWord.of("peter", "Hallo")),
+                                       TestOutputUser.of(PETER.getUser()),
+                                       TestOutputWord.of(PETER.getUser(), "Hallo")),
                        KeyValue.pair(
-                                       "peter",
-                                       TestWord.of("peter", "Welt")),
+                                       TestOutputUser.of(PETER.getUser()),
+                                       TestOutputWord.of(PETER.getUser(), "Welt")),
                        KeyValue.pair(
-                                       "klaus",
-                                       TestWord.of("klaus", "Müsch")),
+                                       TestOutputUser.of(KLAUS.getUser()),
+                                       TestOutputWord.of(KLAUS.getUser(), "Müsch")),
                        KeyValue.pair(
-                                       "klaus",
-                                       TestWord.of("klaus", "gäb")),
+                                       TestOutputUser.of(KLAUS.getUser()),
+                                       TestOutputWord.of(KLAUS.getUser(), "gäb")),
                        KeyValue.pair(
-                                       "klaus",
-                                       TestWord.of("klaus", "s")),
+                                       TestOutputUser.of(KLAUS.getUser()),
+                                       TestOutputWord.of(KLAUS.getUser(), "s")),
                        KeyValue.pair(
-                                       "klaus",
-                                       TestWord.of("klaus", "auch")),
+                                       TestOutputUser.of(KLAUS.getUser()),
+                                       TestOutputWord.of(KLAUS.getUser(), "auch")),
                        KeyValue.pair(
-                                       "peter",
-                                       TestWord.of("peter", "Boäh")),
+                                       TestOutputUser.of(PETER.getUser()),
+                                       TestOutputWord.of(PETER.getUser(), "Boäh")),
                        KeyValue.pair(
-                                       "peter",
-                                       TestWord.of("peter", "echt")),
+                                       TestOutputUser.of(PETER.getUser()),
+                                       TestOutputWord.of(PETER.getUser(), "echt")),
                        KeyValue.pair(
-                                       "peter",
-                                       TestWord.of("peter", "ß")),
+                                       TestOutputUser.of(PETER.getUser()),
+                                       TestOutputWord.of(PETER.getUser(), "ß")),
                        KeyValue.pair(
-                                       "peter",
-                                       TestWord.of("peter", "mal")),
+                                       TestOutputUser.of(PETER.getUser()),
+                                       TestOutputWord.of(PETER.getUser(), "mal")),
                        KeyValue.pair(
-                                       "peter",
-                                       TestWord.of("peter", "nä")),
+                                       TestOutputUser.of(PETER.getUser()),
+                                       TestOutputWord.of(PETER.getUser(), "nä")),
                        KeyValue.pair(
-                                       "peter",
-                                       TestWord.of("peter", "Nümmäh")));
+                                       TestOutputUser.of(PETER.getUser()),
+                                       TestOutputWord.of(PETER.getUser(), "Nümmäh")),
+       };
+
+       static MultiValueMap<TestOutputUser, TestOutputWord> expectedMessages()
+       {
+               MultiValueMap<TestOutputUser, TestOutputWord> expectedMessages = new LinkedMultiValueMap<>();
+               Stream
+                               .of(EXPECTED_MESSAGES)
+                               .forEach(keyValue -> expectedMessages.add(keyValue.key, keyValue.value));
+               return expectedMessages;
+       }
 }