counter: 1.1.6 - Refactored the test-data into a separated class

[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
new file mode 100644 (file)
index 0000000..4b5482b
--- /dev/null
+++ b/src/test/java/de/juplo/kafka/wordcount/counter/TestData.java
@@ -0,0 +1,109 @@
+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 java.util.List;
+import java.util.function.BiConsumer;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+
+class TestData
+{
+       static void writeInputData(BiConsumer<String, String> consumer)
+       {
+               consumer.accept(
+                               "peter",
+                               "Hallo");
+               consumer.accept(
+                               "klaus",
+                               "Müsch");
+               consumer.accept(
+                               "peter",
+                               "Welt");
+               consumer.accept(
+                               "klaus",
+                               "Müsch");
+               consumer.accept(
+                               "klaus",
+                               "s");
+               consumer.accept(
+                               "peter",
+                               "Boäh");
+               consumer.accept(
+                               "peter",
+                               "Welt");
+               consumer.accept(
+                               "peter",
+                               "Boäh");
+               consumer.accept(
+                               "klaus",
+                               "s");
+               consumer.accept(
+                               "peter",
+                               "Boäh");
+               consumer.accept(
+                               "klaus",
+                               "s");
+       }
+
+       static void assertExpectedResult(List<Message> receivedMessages, ObjectMapper mapper)
+       {
+               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);
+       }
+
+       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");
+}