def test_should_fail_kafka_unavailable(self, kafka_client): kafka = mock.Mock() kafka_client.side_effect = client.KafkaUnavailableError() kafka_client.return_value = kafka kafka_health = kc.KafkaHealthCheck() result = kafka_health.health_check() self.assertFalse(result.healthy) kafka.close.assert_not_called()
def test_should_fail_missing_topic(self, kafka_client): kafka = mock.Mock() kafka.topics = ['topic1'] kafka_client.return_value = kafka kafka_health = kc.KafkaHealthCheck() result = kafka_health.health_check() self.assertFalse(result.healthy) kafka.close.assert_called_once()
def test_should_pass(self, kafka_client): kafka = mock.Mock() kafka.topic_partitions = (self.mocked_topics, self.mocked_event_topic, self.mocked_alarm_state_topic) kafka_client.return_value = kafka kafka_health = kc.KafkaHealthCheck() result = kafka_health.health_check() self.assertTrue(result.healthy) kafka.close.assert_called_once()
def __init__(self): super(HealthChecks, self).__init__() self._kafka_check = kafka_check.KafkaHealthCheck() self._alarm_db_check = alarms_db_check.AlarmsDbHealthCheck() self._metrics_db_check = metrics_db_check.MetricsDbCheck()