diff options
author | Dana Powers <dana.powers@gmail.com> | 2018-03-08 07:50:13 -0800 |
---|---|---|
committer | GitHub <noreply@github.com> | 2018-03-08 07:50:13 -0800 |
commit | 4c383daf8a9d7aaa5049a98d7d6da19c85793d2d (patch) | |
tree | 2e7fb0874ce0d7dc154ca60dfc7f646577c7fc97 /test/test_consumer_group.py | |
parent | a6130d288f84af7ffe054bdf301bc40febe07719 (diff) | |
download | kafka-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.py | 3 |
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() |