juplo.de Git - demos/kafka/wordcount/blob - src/test/java/de/juplo/kafka/wordcount/counter/TestData.java

   1 package de.juplo.kafka.wordcount.counter;
   2
   3 import org.apache.kafka.streams.KeyValue;
   4
   5 import java.util.List;
   6 import java.util.Map;
   7 import java.util.Properties;
   8 import java.util.function.BiConsumer;
   9 import java.util.stream.Collectors;
  10
  11 import static org.assertj.core.api.Assertions.assertThat;
  12
  13
  14 class TestData
  15 {
  16         static void writeInputData(BiConsumer<String, Word> consumer)
  17         {
  18                 consumer.accept(
  19                                 "peter",
  20                                 Word.of("peter","Hallo"));
  21                 consumer.accept(
  22                                 "klaus",
  23                                 Word.of("klaus","Müsch"));
  24                 consumer.accept(
  25                                 "peter",
  26                                 Word.of("peter","Welt"));
  27                 consumer.accept(
  28                                 "klaus",
  29                                 Word.of("klaus","Müsch"));
  30                 consumer.accept(
  31                                 "klaus",
  32                                 Word.of("klaus","s"));
  33                 consumer.accept(
  34                                 "peter",
  35                                 Word.of("peter","Boäh"));
  36                 consumer.accept(
  37                                 "peter",
  38                                 Word.of("peter","Welt"));
  39                 consumer.accept(
  40                                 "peter",
  41                                 Word.of("peter","Boäh"));
  42                 consumer.accept(
  43                                 "klaus",
  44                                 Word.of("klaus","s"));
  45                 consumer.accept(
  46                                 "peter",
  47                                 Word.of("peter","Boäh"));
  48                 consumer.accept(
  49                                 "klaus",
  50                                 Word.of("klaus","s"));
  51         }
  52
  53         static void assertExpectedResult(List<KeyValue<Word, WordCount>> receivedMessages)
  54         {
  55                 assertThat(receivedMessages).hasSize(11);
  56                 assertThat(receivedMessages).containsSubsequence(
  57                                 expectedMessages[0]); // Hallo
  58                 assertThat(receivedMessages).containsSubsequence(
  59                                 expectedMessages[1],
  60                                 expectedMessages[3]); // Müsch
  61                 assertThat(receivedMessages).containsSubsequence(
  62                                 expectedMessages[2],
  63                                 expectedMessages[6]);
  64                 assertThat(receivedMessages).containsSubsequence(
  65                                 expectedMessages[4],
  66                                 expectedMessages[8],
  67                                 expectedMessages[10]); // s
  68                 assertThat(receivedMessages).containsSubsequence(
  69                                 expectedMessages[5],
  70                                 expectedMessages[7],
  71                                 expectedMessages[9]); // Boäh
  72         }
  73
  74         static KeyValue<Word,WordCount>[] expectedMessages = new KeyValue[]
  75         {
  76                         KeyValue.pair(
  77                                         Word.of("peter","Hallo"),
  78                                         WordCount.of("peter","Hallo",1)),
  79                         KeyValue.pair(
  80                                         Word.of("klaus","Müsch"),
  81                                         WordCount.of("klaus","Müsch",1)),
  82                         KeyValue.pair(
  83                                         Word.of("peter","Welt"),
  84                                         WordCount.of("peter","Welt",1)),
  85                         KeyValue.pair(
  86                                         Word.of("klaus","Müsch"),
  87                                         WordCount.of("klaus","Müsch",2)),
  88                         KeyValue.pair(
  89                                         Word.of("klaus","s"),
  90                                         WordCount.of("klaus","s",1)),
  91                         KeyValue.pair(
  92                                         Word.of("peter","Boäh"),
  93                                         WordCount.of("peter","Boäh",1)),
  94                         KeyValue.pair(
  95                                         Word.of("peter","Welt"),
  96                                         WordCount.of("peter","Welt",2)),
  97                         KeyValue.pair(
  98                                         Word.of("peter","Boäh"),
  99                                         WordCount.of("peter","Boäh",2)),
 100                         KeyValue.pair(
 101                                         Word.of("klaus","s"),
 102                                         WordCount.of("klaus","s",2)),
 103                         KeyValue.pair(
 104                                         Word.of("peter","Boäh"),
 105                                         WordCount.of("peter","Boäh",3)),
 106                         KeyValue.pair(
 107                                         Word.of("klaus","s"),
 108                                         WordCount.of("klaus","s",3)),
 109         };
 110
 111         static Map<String, Object> convertToMap(Properties properties)
 112         {
 113                 return properties
 114                                 .entrySet()
 115                                 .stream()
 116                                 .collect(
 117                                                 Collectors.toMap(
 118                                                                 entry -> (String)entry.getKey(),
 119                                                                 entry -> entry.getValue()
 120                                                 ));
 121         }
 122 }