83 lines
3.0 KiB
Java
83 lines
3.0 KiB
Java
|
package com.deviceinsight.kafka.health;
|
||
|
|
|
||
|
|
import static com.deviceinsight.kafka.health.KafkaConsumingHealthIndicatorTest.TOPIC;
|
||
|
|
import static org.assertj.core.api.Assertions.assertThat;
|
||
|
|
|
||
|
|
import kafka.server.KafkaServer;
|
||
|
|
import org.apache.kafka.clients.consumer.Consumer;
|
||
|
|
import org.apache.kafka.common.serialization.StringDeserializer;
|
||
|
|
import org.junit.jupiter.api.AfterAll;
|
||
|
|
import org.junit.jupiter.api.BeforeAll;
|
||
|
|
import org.junit.jupiter.api.Test;
|
||
|
|
import org.junit.jupiter.api.TestInstance;
|
||
|
|
import org.junit.jupiter.api.extension.ExtendWith;
|
||
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
||
|
|
import org.springframework.boot.actuate.health.Health;
|
||
|
|
import org.springframework.boot.actuate.health.Status;
|
||
|
|
import org.springframework.boot.autoconfigure.kafka.KafkaProperties;
|
||
|
|
import org.springframework.kafka.core.DefaultKafkaConsumerFactory;
|
||
|
|
import org.springframework.kafka.test.EmbeddedKafkaBroker;
|
||
|
|
import org.springframework.kafka.test.context.EmbeddedKafka;
|
||
|
|
import org.springframework.kafka.test.core.BrokerAddress;
|
||
|
|
import org.springframework.kafka.test.utils.KafkaTestUtils;
|
||
|
|
import org.springframework.test.context.junit.jupiter.SpringExtension;
|
||
|
|
|
||
|
|
import java.time.Duration;
|
||
|
|
import java.util.Collections;
|
||
|
|
import java.util.HashMap;
|
||
|
|
import java.util.Map;
|
||
|
|
|
||
|
|
@ExtendWith(SpringExtension.class)
|
||
|
|
@EmbeddedKafka(topics = {TOPIC})
|
||
|
|
@TestInstance(TestInstance.Lifecycle.PER_CLASS)
|
||
|
|
public class KafkaConsumingHealthIndicatorTest {
|
||
|
|
|
||
|
|
static final String TOPIC = "health-checks";
|
||
|
|
|
||
|
|
private Consumer<String, String> consumer;
|
||
|
|
|
||
|
|
@Autowired
|
||
|
|
private EmbeddedKafkaBroker embeddedKafkaBroker;
|
||
|
|
|
||
|
|
@BeforeAll
|
||
|
|
public void setUp() {
|
||
|
|
Map<String, Object> consumerConfigs =
|
||
|
|
new HashMap<>(KafkaTestUtils.consumerProps("consumer", "false", embeddedKafkaBroker));
|
||
|
|
consumer = new DefaultKafkaConsumerFactory<>(consumerConfigs, new StringDeserializer(),
|
||
|
|
new StringDeserializer()).createConsumer();
|
||
|
|
consumer.subscribe(Collections.singletonList(TOPIC));
|
||
|
|
consumer.poll(Duration.ofSeconds(1));
|
||
|
|
}
|
||
|
|
|
||
|
|
@AfterAll
|
||
|
|
public void tearDown() {
|
||
|
|
consumer.close();
|
||
|
|
embeddedKafkaBroker.getKafkaServers().forEach(KafkaServer::shutdown);
|
||
|
|
embeddedKafkaBroker.getKafkaServers().forEach(KafkaServer::awaitShutdown);
|
||
|
|
}
|
||
|
|
|
||
|
|
@Test
|
||
|
|
public void kafkaIsDown() throws Exception {
|
||
|
|
KafkaHealthProperties kafkaHealthProperties = new KafkaHealthProperties();
|
||
|
|
kafkaHealthProperties.setTopic(TOPIC);
|
||
|
|
|
||
|
|
final KafkaProperties kafkaProperties = new KafkaProperties();
|
||
|
|
BrokerAddress[] brokerAddresses = embeddedKafkaBroker.getBrokerAddresses();
|
||
|
|
kafkaProperties.setBootstrapServers(Collections.singletonList(brokerAddresses[0].toString()));
|
||
|
|
|
||
|
|
KafkaConsumingHealthIndicator healthIndicator =
|
||
|
|
new KafkaConsumingHealthIndicator(kafkaHealthProperties, kafkaProperties.buildConsumerProperties(),
|
||
|
|
kafkaProperties.buildProducerProperties());
|
||
|
|
healthIndicator.subscribeToTopic();
|
||
|
|
|
||
|
|
Health health = healthIndicator.health();
|
||
|
|
assertThat(health.getStatus()).isEqualTo(Status.UP);
|
||
|
|
|
||
|
|
shutdownKafka();
|
||
|
|
}
|
||
|
|
|
||
|
|
private void shutdownKafka() {
|
||
|
|
this.embeddedKafkaBroker.destroy();
|
||
|
|
}
|
||
|
|
}
|