From 4c383daf8a9d7aaa5049a98d7d6da19c85793d2d Mon Sep 17 00:00:00 2001 From: Dana Powers Date: Thu, 8 Mar 2018 07:50:13 -0800 Subject: Close KafkaConsumer instances during tests (#1410) --- test/test_consumer_group.py | 3 +++ 1 file changed, 3 insertions(+) (limited to 'test/test_consumer_group.py') diff --git a/test/test_consumer_group.py b/test/test_consumer_group.py index b930748..f9a41a4 100644 --- a/test/test_consumer_group.py +++ b/test/test_consumer_group.py @@ -44,6 +44,7 @@ def test_consumer(kafka_broker, version): assert len(consumer._client._conns) > 0 node_id = list(consumer._client._conns.keys())[0] assert consumer._client._conns[node_id].state is ConnectionStates.CONNECTED + consumer.close() @pytest.mark.skipif(version() < (0, 9), reason='Unsupported Kafka Version') @@ -153,6 +154,7 @@ def test_paused(kafka_broker, topic): consumer.unsubscribe() assert set() == consumer.paused() + consumer.close() @pytest.mark.skipif(version() < (0, 9), reason='Unsupported Kafka Version') @@ -183,3 +185,4 @@ def test_heartbeat_thread(kafka_broker, topic): assert consumer._coordinator.heartbeat.last_poll == last_poll consumer.poll(timeout_ms=100) assert consumer._coordinator.heartbeat.last_poll > last_poll + consumer.close() -- cgit v1.2.1