summaryrefslogtreecommitdiff
path: root/test/test_consumer_group.py
diff options
context:
space:
mode:
authorDana Powers <dana.powers@gmail.com>2018-03-08 07:50:13 -0800
committerGitHub <noreply@github.com>2018-03-08 07:50:13 -0800
commit4c383daf8a9d7aaa5049a98d7d6da19c85793d2d (patch)
tree2e7fb0874ce0d7dc154ca60dfc7f646577c7fc97 /test/test_consumer_group.py
parenta6130d288f84af7ffe054bdf301bc40febe07719 (diff)
downloadkafka-python-4c383daf8a9d7aaa5049a98d7d6da19c85793d2d.tar.gz
Close KafkaConsumer instances during tests (#1410)
Diffstat (limited to 'test/test_consumer_group.py')
-rw-r--r--test/test_consumer_group.py3
1 files changed, 3 insertions, 0 deletions
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()