X-Git-Url: https://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fde%2Fjuplo%2Fkafka%2Fwordcount%2Fquery%2FQueryApplicationIT.java;h=a9cca100a9674be1ca491177e692da8b77e4013e;hb=57f47fca712981116b726e437f589ac727c5d0a7;hp=e38871ff4488bdc69e351af093aabeb86237e0da;hpb=dcb383aef4f4f454e3a1aed9cf1f2a599c7b1bb9;p=demos%2Fkafka%2Fwordcount

diff --git a/src/test/java/de/juplo/kafka/wordcount/query/QueryApplicationIT.java b/src/test/java/de/juplo/kafka/wordcount/query/QueryApplicationIT.java
index e38871f..a9cca10 100644
--- a/src/test/java/de/juplo/kafka/wordcount/query/QueryApplicationIT.java
+++ b/src/test/java/de/juplo/kafka/wordcount/query/QueryApplicationIT.java
@@ -1,5 +1,6 @@
 package de.juplo.kafka.wordcount.query;
 
+import com.fasterxml.jackson.databind.ObjectMapper;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.kafka.streams.state.KeyValueBytesStoreSupplier;
 import org.apache.kafka.streams.state.Stores;
@@ -32,7 +33,7 @@ import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.
 		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",
+				"spring.kafka.producer.properties.spring.json.type.mapping=userdata:de.juplo.kafka.wordcount.users.TestUserData,ranking:de.juplo.kafka.wordcount.top10.TestRanking",
 				"logging.level.root=WARN",
 				"logging.level.de.juplo=DEBUG",
 				"logging.level.org.apache.kafka.clients=INFO",
@@ -52,6 +53,8 @@ public class QueryApplicationIT
 	@Autowired
 	MockMvc mockMvc;
 	@Autowired
+	ObjectMapper objectMapper;
+	@Autowired
 	QueryStreamProcessor streamProcessor;
 
 
@@ -103,17 +106,19 @@ public class QueryApplicationIT
 				.untilAsserted(() -> TestData.assertExpectedState(user -> requestUserRankingFor(user)));
 	}
 
-	private String requestUserRankingFor(String user)
+	private UserRanking requestUserRankingFor(String user)
 	{
 		try
 		{
-			return mockMvc
-					.perform(MockMvcRequestBuilders.get("/{user}", user)
-							.contentType(MediaType.APPLICATION_JSON))
-					.andExpect(status().isOk())
-					.andReturn()
-					.getResponse()
-					.getContentAsString(StandardCharsets.UTF_8);
+			return objectMapper.readValue(
+					mockMvc
+							.perform(MockMvcRequestBuilders.get("/{user}", user)
+									.contentType(MediaType.APPLICATION_JSON))
+							.andExpect(status().isOk())
+							.andReturn()
+							.getResponse()
+							.getContentAsString(StandardCharsets.UTF_8),
+					UserRanking.class);
 		}
 		catch (Exception e)
 		{