From: Kai Moritz <kai@juplo.de>
Date: Sun, 9 Jun 2024 15:03:59 +0000 (+0200)
Subject: query: 1.0.6 - Added `QueryApplicationIT`
X-Git-Tag: query-1.0.6~2
X-Git-Url: http://juplo.de/gitweb/?a=commitdiff_plain;h=4801746d9b09f418053974a5471bd3087bd4ea3c;p=demos%2Fkafka%2Fwordcount

query: 1.0.6 - Added `QueryApplicationIT`
---

diff --git a/pom.xml b/pom.xml
index d74c826..d7f3a30 100644
--- a/pom.xml
+++ b/pom.xml
@@ -70,6 +70,9 @@
 
 	<build>
 		<plugins>
+			<plugin>
+				<artifactId>maven-failsafe-plugin</artifactId>
+			</plugin>
 			<plugin>
 				<groupId>org.springframework.boot</groupId>
 				<artifactId>spring-boot-maven-plugin</artifactId>
diff --git a/src/main/java/de/juplo/kafka/wordcount/query/QueryStreamProcessor.java b/src/main/java/de/juplo/kafka/wordcount/query/QueryStreamProcessor.java
index b7a08f8..e151259 100644
--- a/src/main/java/de/juplo/kafka/wordcount/query/QueryStreamProcessor.java
+++ b/src/main/java/de/juplo/kafka/wordcount/query/QueryStreamProcessor.java
@@ -88,6 +88,11 @@ public class QueryStreamProcessor
 		return builder.build();
 	}
 
+	ReadOnlyKeyValueStore<String, String> getStore()
+	{
+		return streams.store(StoreQueryParameters.fromNameAndType(STORE_NAME, QueryableStoreTypes.keyValueStore()));
+	}
+
 	public Optional<URI> getRedirect(String username)
 	{
 		KeyQueryMetadata metadata = streams.queryMetadataForKey(STORE_NAME, username, Serdes.String().serializer());
diff --git a/src/test/java/de/juplo/kafka/wordcount/query/QueryApplicationIT.java b/src/test/java/de/juplo/kafka/wordcount/query/QueryApplicationIT.java
new file mode 100644
index 0000000..4e44cda
--- /dev/null
+++ b/src/test/java/de/juplo/kafka/wordcount/query/QueryApplicationIT.java
@@ -0,0 +1,98 @@
+package de.juplo.kafka.wordcount.query;
+
+import lombok.extern.slf4j.Slf4j;
+import org.apache.kafka.streams.state.KeyValueBytesStoreSupplier;
+import org.apache.kafka.streams.state.Stores;
+import org.junit.jupiter.api.BeforeAll;
+import org.junit.jupiter.api.DisplayName;
+import org.junit.jupiter.api.Test;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.boot.test.context.TestConfiguration;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Primary;
+import org.springframework.kafka.core.KafkaTemplate;
+import org.springframework.kafka.support.SendResult;
+import org.springframework.kafka.test.context.EmbeddedKafka;
+
+import java.time.Duration;
+import java.util.concurrent.CompletableFuture;
+
+import static de.juplo.kafka.wordcount.query.QueryStreamProcessor.STORE_NAME;
+import static org.awaitility.Awaitility.await;
+
+
+@SpringBootTest(
+		properties = {
+				"spring.kafka.producer.key-serializer=org.apache.kafka.common.serialization.StringSerializer",
+				"spring.kafka.producer.value-serializer=org.springframework.kafka.support.serializer.JsonSerializer",
+				"spring.kafka.producer.properties.spring.json.add.type.headers=false",
+				"logging.level.root=WARN",
+				"logging.level.de.juplo=DEBUG",
+				"logging.level.org.apache.kafka.clients=INFO",
+				"logging.level.org.apache.kafka.streams=INFO",
+				"juplo.wordcount.query.bootstrap-server=${spring.embedded.kafka.brokers}",
+				"juplo.wordcount.query.users-input-topic=" + QueryApplicationIT.TOPIC_USERS,
+				"juplo.wordcount.query.ranking-input-topic=" + QueryApplicationIT.TOPIC_TOP10 })
+@EmbeddedKafka(topics = { QueryApplicationIT.TOPIC_TOP10, QueryApplicationIT.TOPIC_USERS})
+@Slf4j
+public class QueryApplicationIT
+{
+	public static final String TOPIC_TOP10 = "top10";
+	public static final String TOPIC_USERS = "users";
+
+	@Autowired
+	QueryStreamProcessor streamProcessor;
+
+
+	@BeforeAll
+	public static void testSendMessage(
+			@Autowired KafkaTemplate<String, Object> kafkaTemplate)
+	{
+		TestData
+				.getUsersMessages()
+				.forEach(kv -> flush(kafkaTemplate.send(TOPIC_USERS, kv.key, kv.value)));
+		TestData
+				.getTop10Messages()
+				.forEach(kv -> flush(kafkaTemplate.send(TOPIC_TOP10, kv.key, kv.value)));
+	}
+
+	private static void flush(CompletableFuture<SendResult<String, Object>> future)
+	{
+		try
+		{
+			SendResult<String, Object> result = future.get();
+			log.info(
+					"Sent: {}={}, partition={}, offset={}",
+					result.getProducerRecord().key(),
+					result.getProducerRecord().value(),
+					result.getRecordMetadata().partition(),
+					result.getRecordMetadata().offset());
+		}
+		catch (Exception e)
+		{
+			throw new RuntimeException(e);
+		}
+	}
+
+	@DisplayName("Await the expected state in the state-store")
+	@Test
+	public void testAwaitExpectedState()
+	{
+		await("Expected state")
+				.atMost(Duration.ofSeconds(5))
+				.catchUncaughtExceptions()
+				.untilAsserted(() -> TestData.assertExpectedState(streamProcessor.getStore()));
+	}
+
+	@TestConfiguration
+	static class Configuration
+	{
+		@Primary
+		@Bean
+		KeyValueBytesStoreSupplier inMemoryStoreSupplier()
+		{
+			return Stores.inMemoryKeyValueStore(STORE_NAME);
+		}
+	}
+}
diff --git a/src/test/java/de/juplo/kafka/wordcount/query/TestData.java b/src/test/java/de/juplo/kafka/wordcount/query/TestData.java
index 82f7217..610bca0 100644
--- a/src/test/java/de/juplo/kafka/wordcount/query/TestData.java
+++ b/src/test/java/de/juplo/kafka/wordcount/query/TestData.java
@@ -42,6 +42,11 @@ class TestData
 
 	private static UserRanking userRankingOf(String json)
 	{
+		if (json == null)
+		{
+			return null;
+		}
+
 		try
 		{
 			return objectMapper.readValue(json, UserRanking.class);