diff options
| author | Jeff Widman <jeff@jeffwidman.com> | 2018-10-22 01:24:50 -0700 |
|---|---|---|
| committer | Jeff Widman <jeff@jeffwidman.com> | 2018-10-22 14:30:33 -0700 |
| commit | b83feeca2ec6f6ad745fb7ea47c6484304bb55d8 (patch) | |
| tree | 7580e387521b388f3af629b9d25278faa26ee349 /test/test_producer_legacy.py | |
| parent | a6be21e7b3a20ce2e25ef26140c43b59cc356f38 (diff) | |
| download | kafka-python-b83feeca2ec6f6ad745fb7ea47c6484304bb55d8.tar.gz | |
Vendor `six` consistently
Use vendored `six`, and also `six.moves.range` rather than `xrange`
Diffstat (limited to 'test/test_producer_legacy.py')
| -rw-r--r-- | test/test_producer_legacy.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/test/test_producer_legacy.py b/test/test_producer_legacy.py index 6d00116..ab80ee7 100644 --- a/test/test_producer_legacy.py +++ b/test/test_producer_legacy.py @@ -16,7 +16,7 @@ from kafka.protocol import CODEC_NONE from kafka.structs import ( ProduceResponsePayload, RetryOptions, TopicPartition) -from six.moves import queue, xrange +from kafka.vendor.six.moves import queue, range class TestKafkaProducer(unittest.TestCase): @@ -84,7 +84,7 @@ class TestKafkaProducer(unittest.TestCase): message_list = [message] * (queue_size + 1) producer.send_messages(topic, partition, *message_list) self.assertEqual(producer.queue.qsize(), queue_size) - for _ in xrange(producer.queue.qsize()): + for _ in range(producer.queue.qsize()): producer.queue.get() def test_producer_sync_fail_on_error(self): @@ -253,5 +253,5 @@ class TestKafkaProducerSendUpstream(unittest.TestCase): self.assertEqual(self.client.send_produce_request.call_count, 5) def tearDown(self): - for _ in xrange(self.queue.qsize()): + for _ in range(self.queue.qsize()): self.queue.get() |