def test_check_kafka_connection(self):
"""Test check kafka connections."""
with patch("kafka.BrokerConnection.connect_blocking",
return_value=False):
result = check_kafka_connection()
self.assertFalse(result)
with patch("kafka.BrokerConnection.connect_blocking",
return_value=True):
with patch("kafka.BrokerConnection.close") as mock_close:
result = check_kafka_connection()
mock_close.assert_called()
self.assertTrue(result)
def readiness_check(self):
"""Set the readiness check response."""
status = 424
msg = "not ready"
if self.ready:
if not check_kafka_connection():
response = ProbeResponse(status, "kafka connection error")
self._write_response(response)
self.logger.info(response.json)
return
if not check_sources_connection():
response = ProbeResponse(status, "sources-api not ready")
self._write_response(response)
self.logger.info(response.json)
return
status = 200
msg = "ok"
self._write_response(ProbeResponse(status, msg))
def is_kafka_connected(): # pragma: no cover
"""
Check connectability to Kafka messenger.
This method will block sources integration initialization until
Kafka is connected.
"""
count = 0
result = False
while not result:
result = check_kafka_connection()
if result:
LOG.info("Test connection to Kafka was successful.")
else:
LOG.error("Unable to connect to Kafka server.")
KAFKA_CONNECTION_ERRORS_COUNTER.inc()
backoff(count)
count += 1
return result
