From f126e5bfcc8f41ee5ea29b41ec6eabbc3f441647 Mon Sep 17 00:00:00 2001 From: Brian Sang Date: Fri, 31 May 2019 19:05:41 -0700 Subject: Sanity test for consumer.topics() and consumer.partitions_for_topic() --- test/test_consumer_group.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/test/test_consumer_group.py b/test/test_consumer_group.py index d7aaa88..ec26857 100644 --- a/test/test_consumer_group.py +++ b/test/test_consumer_group.py @@ -29,6 +29,15 @@ def test_consumer(kafka_broker, topic, version): assert consumer._client._conns[node_id].state is ConnectionStates.CONNECTED consumer.close() +@pytest.mark.skipif(not version(), reason="No KAFKA_VERSION set") +def test_consumer_topics(kafka_broker, topic, version): + consumer = KafkaConsumer(bootstrap_servers=get_connect_str(kafka_broker)) + # Necessary to drive the IO + consumer.poll(500) + consumer_topics = consumer.topics() + assert topic in consumer_topics + assert len(consumer.partitions_for_topic(topic)) > 0 + consumer.close() @pytest.mark.skipif(version() < (0, 9), reason='Unsupported Kafka Version') @pytest.mark.skipif(not version(), reason="No KAFKA_VERSION set") -- cgit v1.2.1