1 package de.juplo.kafka.wordcount.counter;
3 import lombok.extern.slf4j.Slf4j;
4 import org.apache.kafka.streams.TestInputTopic;
5 import org.apache.kafka.streams.TestOutputTopic;
6 import org.apache.kafka.streams.Topology;
7 import org.apache.kafka.streams.TopologyTestDriver;
8 import org.apache.kafka.streams.state.Stores;
9 import org.junit.jupiter.api.Test;
10 import org.springframework.kafka.support.serializer.JsonDeserializer;
11 import org.springframework.kafka.support.serializer.JsonSerde;
12 import org.springframework.kafka.support.serializer.JsonSerializer;
13 import org.springframework.util.LinkedMultiValueMap;
14 import org.springframework.util.MultiValueMap;
17 import java.util.Properties;
19 import static de.juplo.kafka.wordcount.counter.TestData.convertToMap;
20 import static de.juplo.kafka.wordcount.counter.TestData.parseHeader;
21 import static org.springframework.kafka.support.mapping.AbstractJavaTypeMapper.DEFAULT_CLASSID_FIELD_NAME;
22 import static org.springframework.kafka.support.mapping.AbstractJavaTypeMapper.KEY_DEFAULT_CLASSID_FIELD_NAME;
26 public class CounterStreamProcessorTopologyTest
28 public final static String IN = "TEST-IN";
29 public final static String OUT = "TEST-OUT";
34 Topology topology = CounterStreamProcessor.buildTopology(
37 Stores.inMemoryKeyValueStore("TOPOLOGY-TEST"));
39 CounterApplicationConfiguriation applicationConfiguriation =
40 new CounterApplicationConfiguriation();
41 Properties streamProcessorProperties =
42 applicationConfiguriation.streamProcessorProperties(new CounterApplicationProperties());
43 Map<String, Object> propertyMap = convertToMap(streamProcessorProperties);
45 JsonSerde<?> keySerde = new JsonSerde<>();
46 keySerde.configure(propertyMap, true);
47 JsonSerde<?> valueSerde = new JsonSerde<>();
48 valueSerde.configure(propertyMap, false);
50 TopologyTestDriver testDriver = new TopologyTestDriver(topology, streamProcessorProperties);
52 TestInputTopic<String, Word> in = testDriver.createInputTopic(
54 (JsonSerializer<String>)keySerde.serializer(),
55 (JsonSerializer<Word>)valueSerde.serializer());
57 TestOutputTopic<Word, WordCounter> out = testDriver.createOutputTopic(
59 (JsonDeserializer<Word>)keySerde.deserializer(),
60 (JsonDeserializer<WordCounter>)valueSerde.deserializer());
62 TestData.writeInputData((key, value) -> in.pipeInput(key, value));
64 MultiValueMap<Word, WordCounter> receivedMessages = new LinkedMultiValueMap<>();
70 "OUT: {} -> {}, {}, {}",
73 parseHeader(record.headers(), KEY_DEFAULT_CLASSID_FIELD_NAME),
74 parseHeader(record.headers(), DEFAULT_CLASSID_FIELD_NAME));
75 receivedMessages.add(record.key(), record.value());
78 TestData.assertExpectedResult(receivedMessages);