java - 如何为 Spring Kafka Listener 创建集成测试
问题描述
我有一个微服务发送到另一个应该消费的微服务消息。
所以,kafka 配置工作,一切工作,但我需要为这段代码创建一个集成测试,我不知道如何。
我的 KafkaConsumer.Class 带有组件注释:
private static final Logger logger = LoggerFactory.getLogger(KafkaReactionConsumerMessageComponent.class);
private final ReactionsService reactionsService;
public KafkaReactionConsumerMessageComponent(ReactionsService reactionsService) {
this.reactionsService = reactionsService;
}
@KafkaListener(topics = "reaction-topic", clientIdPrefix = "string", groupId = "magpie-trending")
public void consumingReactionMessages(ConsumerRecord<String, String> cr,
@Payload String payload){
logger.info("[JSON] received Payload: {}", payload);
try {
ObjectMapper mapper = new ObjectMapper();
ReactionMessage message = mapper.readValue(payload, ReactionMessage.class);
if(StringUtils.equals("unloved", message.getReactionType())) {
reactionsService.deleteReactionsByUserIdAndPostId(message.getPost().getPostId(), message.getUser().getUserId());
logger.info("Deleted reactions from database with postId: {} and userId: {}", message.getPost().getPostId(), message.getUser().getUserId());
} else {
List<Reaction> reactions = creatReactions(message).stream()
.map(reactionsService::insertReaction).collect(Collectors.toList());
logger.info("Added reactions to database: {}", reactions);
}
} catch (Exception e){
logger.error("Cannot Deserialize payload to ReactionMessage");
}
}
我的集成测试是
private static final String TOPIC = "reaction-topic";
private final Logger logger = LoggerFactory.getLogger(KafkaDeletePostConsumerMessageComponent.class);
private final KafkaReactionConsumerMessageComponent kafkaReactionConsumerMessageComponent;
private final EmbeddedKafkaBroker embeddedKafkaBroker;
private Consumer<String, String> consumer;
@SuppressWarnings("SpringJavaAutowiringInspection")
@Autowired
public KafkaReactionMessageConsumerTest(KafkaReactionConsumerMessageComponent kafkaReactionConsumerMessageComponent,
EmbeddedKafkaBroker embeddedKafkaBroker) {
this.kafkaReactionConsumerMessageComponent = kafkaReactionConsumerMessageComponent;
this.embeddedKafkaBroker = embeddedKafkaBroker;
}
@BeforeEach
public void setUp() {
Map<String, Object> configs = new HashMap<>(KafkaTestUtils.consumerProps("consumer", "true", embeddedKafkaBroker));
consumer = new DefaultKafkaConsumerFactory<>(configs, new StringDeserializer(), new StringDeserializer()).createConsumer();
consumer.subscribe(Collections.singleton(TOPIC));
consumer.poll(Duration.ZERO);
}
@AfterEach
public void tearDown() {
consumer.close();
}
@Test
public void shoudlConsumeAndInsertInDatabaseReactionDomain() {
ReactionMessage reactionMessage = new ReactionMessage(new PostMessage("1", Set.of("a", "b", "c")),
new UserMessage("2"), LocalDateTime.now().toString(), "loved");
Map<String, Object> configs = new HashMap<>(KafkaTestUtils.producerProps(embeddedKafkaBroker));
Producer<String, String> producer = new DefaultKafkaProducerFactory<>(configs, new StringSerializer(), new StringSerializer()).createProducer();
producer.send(new ProducerRecord<>(TOPIC, "1", reactionMessage.toString()));
producer.flush();
assertEquals(3, mongoTemplate.getCollection("reactions").countDocuments());
}
抽象类:
@ExtendWith(SpringExtension.class)
@SpringBootTest
@AutoConfigureMockMvc
@EmbeddedKafka(brokerProperties={
"log.dir=out/embedded-kafka"
})
public abstract class AbstractMongoEmbeddedTest {
@Autowired
private static MongodExecutable mongodExecutable;
@Autowired
protected MongoTemplate mongoTemplate;
@BeforeEach
private void dropPostCollection(){
mongoTemplate.dropCollection(Reaction.class);
}
解决方案
由于您使用的是嵌入式 Kafka 代理,您可以简单地从集成测试中生成/使用所需的主题。
消费
消费可以通过一个简单的 jUnit 规则来完成。可以在此处找到用于此目的的规则。随意使用它。
您可以像这样使用它来断言已使用的消息:
assertThat(kafkaConsumeMessagesRule.pollMessage()).isEqualTo("your-expected-message");
生产
要生成消息,您只需org.springframework.kafka.core.KafkaTemplate
在集成测试中连接 a 并将消息发送到给定主题。
推荐阅读
- typescript - 使用 --noEmit 时使用 TypeScript 进行增量编译?
- websocket - 托管服务器端 Blazor 应用程序时 SignalR 出现问题
- angular - Angular 渲染 FormArray,在一页中包含 10k+ 行(每行 15 列)数据
- javascript - 单击多个图像可同时播放多个音频
- google-kubernetes-engine - 在 GKE 上使用 Istio 获取 Envoy 访问日志
- python - 在 Python 中嵌套时展平元组
- angular - ngbTooltip 不适用于动态创建的元素
- java - 如果按钮空闲,如何启动线程?
- java - 在 Java 中创建 UDF 以将一个数据框列映射到另一列
- java - 尝试在 Spoj 中提出问题,我编写的代码被接受,但显示内存使用为 4468mb。是什么导致如此高的内存使用率?