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