diff options
Diffstat (limited to 'test/test_producer.py')
| -rw-r--r-- | test/test_producer.py | 13 |
1 files changed, 3 insertions, 10 deletions
diff --git a/test/test_producer.py b/test/test_producer.py index f62b97a..227d4ad 100644 --- a/test/test_producer.py +++ b/test/test_producer.py @@ -2,6 +2,7 @@ import collections import logging +import threading import time from mock import MagicMock, patch @@ -15,15 +16,7 @@ from kafka.common import ( from kafka.producer.base import Producer, _send_upstream from kafka.protocol import CODEC_NONE -import threading -try: - from queue import Empty, Queue -except ImportError: - from Queue import Empty, Queue -try: - xrange -except NameError: - xrange = range +from six.moves import queue, xrange class TestKafkaProducer(unittest.TestCase): @@ -130,7 +123,7 @@ class TestKafkaProducerSendUpstream(unittest.TestCase): def setUp(self): self.client = MagicMock() - self.queue = Queue() + self.queue = queue.Queue() def _run_process(self, retries_limit=3, sleep_timeout=1): # run _send_upstream process with the queue |