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.function.BiConsumer;
   7
   8 import static org.assertj.core.api.Assertions.assertThat;
   9
  10
  11 class TestData
  12 {
  13         static void writeInputData(BiConsumer<String, Word> consumer)
  14         {
  15                 consumer.accept(
  16                                 "peter",
  17                                 Word.of("peter","Hallo"));
  18                 consumer.accept(
  19                                 "klaus",
  20                                 Word.of("klaus","Müsch"));
  21                 consumer.accept(
  22                                 "peter",
  23                                 Word.of("peter","Welt"));
  24                 consumer.accept(
  25                                 "klaus",
  26                                 Word.of("klaus","Müsch"));
  27                 consumer.accept(
  28                                 "klaus",
  29                                 Word.of("klaus","s"));
  30                 consumer.accept(
  31                                 "peter",
  32                                 Word.of("peter","Boäh"));
  33                 consumer.accept(
  34                                 "peter",
  35                                 Word.of("peter","Welt"));
  36                 consumer.accept(
  37                                 "peter",
  38                                 Word.of("peter","Boäh"));
  39                 consumer.accept(
  40                                 "klaus",
  41                                 Word.of("klaus","s"));
  42                 consumer.accept(
  43                                 "peter",
  44                                 Word.of("peter","Boäh"));
  45                 consumer.accept(
  46                                 "klaus",
  47                                 Word.of("klaus","s"));
  48         }
  49
  50         static void assertExpectedResult(List<KeyValue<Word, WordCount>> receivedMessages)
  51         {
  52                 assertThat(receivedMessages).hasSize(11);
  53                 assertThat(receivedMessages).containsSubsequence(
  54                                 expectedMessages[0]); // Hallo
  55                 assertThat(receivedMessages).containsSubsequence(
  56                                 expectedMessages[1],
  57                                 expectedMessages[3]); // Müsch
  58                 assertThat(receivedMessages).containsSubsequence(
  59                                 expectedMessages[2],
  60                                 expectedMessages[6]);
  61                 assertThat(receivedMessages).containsSubsequence(
  62                                 expectedMessages[4],
  63                                 expectedMessages[8],
  64                                 expectedMessages[10]); // s
  65                 assertThat(receivedMessages).containsSubsequence(
  66                                 expectedMessages[5],
  67                                 expectedMessages[7],
  68                                 expectedMessages[9]); // Boäh
  69         }
  70
  71         static KeyValue<Word,WordCount>[] expectedMessages = new KeyValue[]
  72         {
  73                         KeyValue.pair(
  74                                         Word.of("peter","Hallo"),
  75                                         WordCount.of("peter","Hallo",1)),
  76                         KeyValue.pair(
  77                                         Word.of("klaus","Müsch"),
  78                                         WordCount.of("klaus","Müsch",1)),
  79                         KeyValue.pair(
  80                                         Word.of("peter","Welt"),
  81                                         WordCount.of("peter","Welt",1)),
  82                         KeyValue.pair(
  83                                         Word.of("klaus","Müsch"),
  84                                         WordCount.of("klaus","Müsch",2)),
  85                         KeyValue.pair(
  86                                         Word.of("klaus","s"),
  87                                         WordCount.of("klaus","s",1)),
  88                         KeyValue.pair(
  89                                         Word.of("peter","Boäh"),
  90                                         WordCount.of("peter","Boäh",1)),
  91                         KeyValue.pair(
  92                                         Word.of("peter","Welt"),
  93                                         WordCount.of("peter","Welt",2)),
  94                         KeyValue.pair(
  95                                         Word.of("peter","Boäh"),
  96                                         WordCount.of("peter","Boäh",2)),
  97                         KeyValue.pair(
  98                                         Word.of("klaus","s"),
  99                                         WordCount.of("klaus","s",2)),
 100                         KeyValue.pair(
 101                                         Word.of("peter","Boäh"),
 102                                         WordCount.of("peter","Boäh",3)),
 103                         KeyValue.pair(
 104                                         Word.of("klaus","s"),
 105                                         WordCount.of("klaus","s",3)),
 106         };
 107 }