package de.juplo.kafka.wordcount.query;
-import com.fasterxml.jackson.databind.ObjectMapper;
import de.juplo.kafka.wordcount.top10.TestRanking;
+import de.juplo.kafka.wordcount.top10.TestUser;
import de.juplo.kafka.wordcount.users.TestUserData;
import lombok.extern.slf4j.Slf4j;
import org.apache.kafka.common.serialization.StringSerializer;
import org.junit.jupiter.api.Test;
import org.springframework.kafka.support.serializer.JsonSerializer;
+import java.util.Map;
+
import static de.juplo.kafka.wordcount.query.QueryApplicationConfiguration.serializationConfig;
{
public static final String TOP10_IN = "TOP10-IN";
public static final String USERS_IN = "USERS-IN";
- public static final String STORE_NAME = "TOPOLOGY-TEST";
+ public static final String RANKING_STORE_NAME = "TOPOLOGY-TEST-RANKINGS";
+ public static final String USERS_STORE_NAME = "TOPOLOGY-TEST-USERS";
TopologyTestDriver testDriver;
- TestInputTopic<String, TestRanking> top10In;
+ TestInputTopic<TestUser, TestRanking> top10In;
TestInputTopic<String, TestUserData> userIn;
Topology topology = QueryStreamProcessor.buildTopology(
USERS_IN,
TOP10_IN,
- Stores.inMemoryKeyValueStore(STORE_NAME),
- new ObjectMapper());
+ Stores.inMemoryKeyValueStore(USERS_STORE_NAME),
+ Stores.inMemoryKeyValueStore(RANKING_STORE_NAME));
testDriver = new TopologyTestDriver(topology, serializationConfig());
top10In = testDriver.createInputTopic(
TOP10_IN,
- new StringSerializer(),
- new JsonSerializer());
+ jsonSerializer(TestUser.class, true),
+ jsonSerializer(TestRanking.class,false));
userIn = testDriver.createInputTopic(
USERS_IN,
new StringSerializer(),
- new JsonSerializer());
+ jsonSerializer(TestUserData.class, false).noTypeInfo());
}
.getTop10Messages()
.forEach(kv -> top10In.pipeInput(kv.key, kv.value));
- KeyValueStore<String, String> store = testDriver.getKeyValueStore(STORE_NAME);
- TestData.assertExpectedState(store);
+ KeyValueStore<String, UserRanking> store = testDriver.getKeyValueStore(RANKING_STORE_NAME);
+ TestData.assertExpectedState(user -> store.get(user));
}
@AfterEach
{
testDriver.close();
}
+
+ private <T> JsonSerializer<T> jsonSerializer(Class<T> type, boolean isKey)
+ {
+ JsonSerializer<T> jsonSerializer = new JsonSerializer<>();
+ jsonSerializer.configure(
+ Map.of(
+ JsonSerializer.TYPE_MAPPINGS,
+ "user:" + TestUser.class.getName() + "," +
+ "ranking:" + TestRanking.class.getName()),
+ isKey);
+ return jsonSerializer;
+ }
}