diff options
Diffstat (limited to 'test/test_producer.py')
-rw-r--r-- | test/test_producer.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/test/test_producer.py b/test/test_producer.py index 80017a1..09d184f 100644 --- a/test/test_producer.py +++ b/test/test_producer.py @@ -55,7 +55,6 @@ def test_end_to_end(kafka_broker, compression): futures.append(producer.send(topic, 'msg %d' % i)) ret = [f.get(timeout=30) for f in futures] assert len(ret) == messages - producer.close() consumer.subscribe([topic]) @@ -67,6 +66,7 @@ def test_end_to_end(kafka_broker, compression): break assert msgs == set(['msg %d' % i for i in range(messages)]) + consumer.close() @pytest.mark.skipif(platform.python_implementation() != 'CPython', |