def test_get_zookeeper_groups_error(self, mock_print):
with self.mock_kafka_info() as (mock_ZK, _):
obj = mock_ZK.return_value.__enter__.return_value
obj.__exit__.return_value = False
cluster_config = mock.Mock(zookeeper='some_ip')
obj.get_children.side_effect = NoNodeError("Boom!")
ListGroups.get_zookeeper_groups(cluster_config)
mock_print.assert_any_call(
"Error: No consumers node found in zookeeper",
file=sys.stderr,
)
def test_get_kafka_groups_error(self, mock_print):
with self.mock_kafka_info() as (_, mock_kafka_reader):
m = mock_kafka_reader.return_value
m.read_groups.side_effect = Exception("Boom!")
cluster_config = mock.Mock(zookeeper='some_ip', type='some_cluster_type')
cluster_config.configure_mock(name='some_cluster_name')
ListGroups.get_kafka_groups(cluster_config)
mock_print.assert_any_call(
"Error: No consumer offsets topic found in Kafka",
file=sys.stderr,
)
def test_print_groups(self, mock_print):
groups = ['group1', 'group2', 'group3']
cluster_config = mock.Mock(zookeeper='some_ip', type='some_cluster_type')
cluster_config.configure_mock(name='some_cluster_name')
expected_print = [
mock.call("Consumer Groups:"),
mock.call("\tgroup1"),
mock.call("\tgroup2"),
mock.call("\tgroup3"),
mock.call("3 groups found for cluster some_cluster_name "
"of type some_cluster_type"),
]
ListGroups.print_groups(groups, cluster_config)
assert mock_print.call_args_list == expected_print
def test_print_groups(self, mock_print):
groups = ['group1', 'group2', 'group3']
cluster_config = mock.Mock(zookeeper='some_ip', type='some_cluster_type')
cluster_config.configure_mock(name='some_cluster_name')
expected_print = [
mock.call("Consumer Groups:"),
mock.call("\tgroup1"),
mock.call("\tgroup2"),
mock.call("\tgroup3"),
mock.call("3 groups found for cluster some_cluster_name "
"of type some_cluster_type"),
]
ListGroups.print_groups(groups, cluster_config)
assert mock_print.call_args_list == expected_print
def test_get_zookeeper_groups(self):
with self.mock_kafka_info() as (mock_ZK, _):
expected_groups = ['group1', 'group2', 'group3']
obj = mock_ZK.return_value.__enter__.return_value
obj.get_children.return_value = expected_groups
cluster_config = mock.Mock(zookeeper='some_ip', type='some_cluster_type')
cluster_config.configure_mock(name='some_cluster_name')
assert ListGroups.get_zookeeper_groups(cluster_config) == expected_groups
assert obj.get_children.call_count == 1
def test_get_kafka_groups(self):
with self.mock_kafka_info() as (_, mock_kafka_reader):
expected_groups = {
'group_name': ['topic1', 'topic2'],
'group2': ['topic3', 'topic4']
}
m = mock_kafka_reader.return_value
m.read_groups.return_value = expected_groups
cluster_config = mock.Mock(zookeeper='some_ip', type='some_cluster_type')
cluster_config.configure_mock(name='some_cluster_name')
result = ListGroups.get_kafka_groups(cluster_config)
assert result == expected_groups.keys()
assert m.read_groups.call_count == 1
def test_get_kafka_groups(self):
with self.mock_kafka_info() as (mock_kafka_reader):
expected_groups = {
'group_name': ['topic1', 'topic2'],
'group2': ['topic3', 'topic4']
}
m = mock_kafka_reader.return_value
m.read_groups.return_value = expected_groups
cluster_config = mock.Mock(zookeeper='some_ip', type='some_cluster_type')
cluster_config.configure_mock(name='some_cluster_name')
result = ListGroups.get_kafka_groups(cluster_config)
assert result == list(expected_groups.keys())
assert m.read_groups.call_count == 1