diff options
Diffstat (limited to 'test/test_consumer.py')
| -rw-r--r-- | test/test_consumer.py | 16 |
1 files changed, 2 insertions, 14 deletions
diff --git a/test/test_consumer.py b/test/test_consumer.py index 778d76a..f70e292 100644 --- a/test/test_consumer.py +++ b/test/test_consumer.py @@ -1,22 +1,10 @@ -import os -import random -import struct import unittest2 -from mock import MagicMock, patch +from mock import MagicMock -from kafka import KafkaClient from kafka.consumer import SimpleConsumer -from kafka.common import ( - ProduceRequest, BrokerMetadata, PartitionMetadata, - TopicAndPartition, KafkaUnavailableError, - LeaderUnavailableError, PartitionUnavailableError -) -from kafka.protocol import ( - create_message, KafkaProtocol -) class TestKafkaConsumer(unittest2.TestCase): def test_non_integer_partitions(self): with self.assertRaises(AssertionError): - consumer = SimpleConsumer(MagicMock(), 'group', 'topic', partitions = [ '0' ]) + SimpleConsumer(MagicMock(), 'group', 'topic', partitions = [ '0' ]) |