juplo.de Git - demos/kafka/wordcount/blob - src/test/java/de/juplo/kafka/wordcount/recorder/RecorderApplicationIT.java

   1 package de.juplo.kafka.wordcount.recorder;
   2
   3 import de.juplo.kafka.wordcount.splitter.TestRecording;
   4 import de.juplo.kafka.wordcount.splitter.TestUser;
   5 import lombok.extern.slf4j.Slf4j;
   6 import org.junit.jupiter.api.DisplayName;
   7 import org.junit.jupiter.api.Test;
   8 import org.springframework.beans.factory.annotation.Autowired;
   9 import org.springframework.boot.test.autoconfigure.web.servlet.AutoConfigureMockMvc;
  10 import org.springframework.boot.test.context.SpringBootTest;
  11 import org.springframework.boot.test.context.TestConfiguration;
  12 import org.springframework.context.annotation.Bean;
  13 import org.springframework.http.MediaType;
  14 import org.springframework.kafka.annotation.KafkaListener;
  15 import org.springframework.kafka.support.KafkaHeaders;
  16 import org.springframework.kafka.test.context.EmbeddedKafka;
  17 import org.springframework.messaging.handler.annotation.Header;
  18 import org.springframework.messaging.handler.annotation.Payload;
  19 import org.springframework.test.web.servlet.MockMvc;
  20 import org.springframework.test.web.servlet.MvcResult;
  21 import org.springframework.util.LinkedMultiValueMap;
  22 import org.springframework.util.MultiValueMap;
  23
  24 import java.time.Duration;
  25 import java.util.List;
  26
  27 import static de.juplo.kafka.wordcount.recorder.RecorderApplicationIT.TOPIC_OUT;
  28 import static org.assertj.core.api.Assertions.assertThat;
  29 import static org.awaitility.Awaitility.await;
  30 import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.asyncDispatch;
  31 import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.post;
  32 import static org.springframework.test.web.servlet.result.MockMvcResultHandlers.print;
  33 import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.jsonPath;
  34 import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
  35
  36
  37 @SpringBootTest(
  38                 properties = {
  39                                 "spring.kafka.consumer.auto-offset-reset=earliest",
  40                                 "spring.kafka.consumer.key-deserializer=org.springframework.kafka.support.serializer.JsonDeserializer",
  41                                 "spring.kafka.consumer.value-deserializer=org.springframework.kafka.support.serializer.JsonDeserializer",
  42                                 "spring.kafka.consumer.properties.spring.json.key.default.type=de.juplo.kafka.wordcount.splitter.TestUser",
  43                                 "spring.kafka.consumer.properties.spring.json.value.default.type=de.juplo.kafka.wordcount.splitter.TestRecording",
  44                                 "logging.level.root=WARN",
  45                                 "logging.level.de.juplo=DEBUG",
  46                                 "juplo.wordcount.recorder.bootstrap-server=${spring.embedded.kafka.brokers}",
  47                                 "juplo.wordcount.recorder.topic=" + TOPIC_OUT })
  48 @AutoConfigureMockMvc
  49 @EmbeddedKafka(topics = { TOPIC_OUT })
  50 @Slf4j
  51 class RecorderApplicationIT
  52 {
  53         static final String TOPIC_OUT = "out";
  54
  55         @Autowired
  56         MockMvc mockMvc;
  57         @Autowired
  58         Consumer consumer;
  59
  60
  61         @Test
  62         @DisplayName("Posted messages are accepted and sent to Kafka")
  63         void userMessagesAreAcceptedAndSentToKafka()
  64         {
  65                 MultiValueMap<TestUser, TestRecording> recordings = new LinkedMultiValueMap<>();
  66
  67                 TestData
  68                                 .getInputMessages()
  69                                 .forEach(kv ->
  70                                 {
  71                                         sendRedording(kv.key, kv.value);
  72                                         recordings.add(TestUser.of(kv.key), TestRecording.of(kv.key, kv.value));
  73                                 });
  74
  75
  76                 await("Received expected messages")
  77                                 .atMost(Duration.ofSeconds(5))
  78                                 .untilAsserted(() -> recordings.forEach((user, userRecordings) ->
  79                                                 assertThat(consumer.receivedFor(user)).containsExactlyElementsOf(userRecordings)));
  80         }
  81
  82         void sendRedording(String user, String sentence)
  83         {
  84                 try
  85                 {
  86                         MvcResult result = mockMvc
  87                                         .perform(post("/{user}", user)
  88                                                         .contentType(MediaType.TEXT_PLAIN)
  89                                                         .content(sentence))
  90                                         .andReturn();
  91
  92                         mockMvc.perform(asyncDispatch(result))
  93                                         .andDo(print())
  94                                         .andExpect(status().isOk())
  95                                         .andExpect(jsonPath("$.username").value(user))
  96                                         .andExpect(jsonPath("$.sentence").value(sentence));
  97                 }
  98                 catch (Exception e)
  99                 {
 100                         throw new RuntimeException(e);
 101                 }
 102         }
 103
 104
 105         static class Consumer
 106         {
 107                 private final MultiValueMap<TestUser, TestRecording> received = new LinkedMultiValueMap<>();
 108
 109                 @KafkaListener(groupId = "TEST", topics = TOPIC_OUT)
 110                 public synchronized void receive(
 111                                 @Header(KafkaHeaders.RECEIVED_KEY) TestUser user,
 112                                 @Payload TestRecording recording)
 113                 {
 114                         log.debug("Received message: {}={}", user, recording);
 115                         received.add(user, recording);
 116                 }
 117
 118                 synchronized List<TestRecording> receivedFor(TestUser user)
 119                 {
 120                         List<TestRecording> recordings = received.get(user);
 121                         return recordings == null
 122                                         ? List.of()
 123                                         : received.get(user).stream().toList();
 124                 }
 125         }
 126
 127         @TestConfiguration
 128         static class Configuration
 129         {
 130                 @Bean
 131                 Consumer consumer()
 132                 {
 133                         return new Consumer();
 134                 }
 135         }
 136 }