def test_get_topic_partition_metadata_empty(self, mock_get_metadata):
mock_get_metadata.return_value = {}
actual = get_topic_partition_with_error(self.cluster_config, 5)
expected = set([])
assert actual == expected
mock_get_metadata.asserd_called_wity('some_list')
def test_get_topic_partition_metadata_leader_not_available(
self, mock_get_metadata):
mock_get_metadata.return_value = METADATA_RESPONSE_WITH_ERRORS
actual = get_topic_partition_with_error(self.cluster_config,
LEADER_NOT_AVAILABLE_ERROR)
expected = set([('topic0', 1)])
assert actual == expected
def test_get_topic_partition_metadata_leader_not_available(self, mock_get_metadata):
mock_get_metadata.return_value = METADATA_RESPONSE_WITH_ERRORS
with mock.patch(
'kafka_utils.util.metadata.ZK',
):
actual = get_topic_partition_with_error(self.cluster_config, LEADER_NOT_AVAILABLE_ERROR)
expected = set([('topic0', 1)])
assert actual == expected
def test_get_topic_partition_metadata_replica_not_available(self, mock_get_metadata):
mock_get_metadata.return_value = METADATA_RESPONSE_WITH_ERRORS
with mock.patch(
'kafka_utils.util.metadata.ZK',
):
actual = get_topic_partition_with_error(self.cluster_config, REPLICA_NOT_AVAILABLE_ERROR)
expected = {('topic0', 0), ('topic1', 1)}
assert actual == expected
def test_get_topic_partition_metadata_no_errors(self, mock_get_metadata):
mock_get_metadata.return_value = METADATA_RESPONSE_ALL_FINE
with mock.patch(
'kafka_utils.util.metadata.ZK',
):
actual = get_topic_partition_with_error(self.cluster_config, 5)
expected = set([])
assert actual == expected
def run_command(self):
"""Checks the number of offline partitions"""
offline = get_topic_partition_with_error(
self.cluster_config,
LEADER_NOT_AVAILABLE_ERROR,
)
errcode = status_code.OK if not offline else status_code.CRITICAL
out = _prepare_output(offline, self.args.verbose, self.args.head)
return errcode, out
def run_command(self):
"""Checks the number of offline partitions"""
offline = get_topic_partition_with_error(
self.cluster_config,
LEADER_NOT_AVAILABLE_ERROR,
)
errcode = status_code.OK if not offline else status_code.CRITICAL
out = _prepare_output(offline, self.args.verbose)
return errcode, out
def test_get_topic_partition_metadata_empty(self, mock_get_metadata):
mock_get_metadata.return_value = {}
with mock.patch(
'kafka_utils.util.metadata.ZK',
):
actual = get_topic_partition_with_error(self.cluster_config, 5)
expected = set([])
assert actual == expected
mock_get_metadata.asserd_called_wity('some_list')
def run_command(self):
"""replica_unavailability command, checks number of replicas not available
for communication over all brokers in the Kafka cluster."""
replica_unavailability = get_topic_partition_with_error(
self.cluster_config,
REPLICA_NOT_AVAILABLE_ERROR,
)
errcode = status_code.OK if not replica_unavailability else status_code.CRITICAL
out = _prepare_output(replica_unavailability, self.args.verbose)
return errcode, out
def run_command(self):
"""replica_unavailability command, checks number of replicas not available
for communication over all brokers in the Kafka cluster."""
fetch_unavailable_brokers = True
result = get_topic_partition_with_error(
self.cluster_config,
REPLICA_NOT_AVAILABLE_ERROR,
fetch_unavailable_brokers=fetch_unavailable_brokers,
)
if fetch_unavailable_brokers:
replica_unavailability, unavailable_brokers = result
else:
replica_unavailability = result
errcode = status_code.OK if not replica_unavailability else status_code.CRITICAL
out = _prepare_output(replica_unavailability, unavailable_brokers, self.args.verbose)
return errcode, out
