diff options
Diffstat (limited to 'test/test_consumer.py')
-rw-r--r-- | test/test_consumer.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/test/test_consumer.py b/test/test_consumer.py index 9060919..7b8f370 100644 --- a/test/test_consumer.py +++ b/test/test_consumer.py @@ -2,9 +2,14 @@ from mock import MagicMock from . import unittest -from kafka.consumer import SimpleConsumer +from kafka import SimpleConsumer, KafkaConsumer +from kafka.common import KafkaConfigurationError class TestKafkaConsumer(unittest.TestCase): def test_non_integer_partitions(self): with self.assertRaises(AssertionError): SimpleConsumer(MagicMock(), 'group', 'topic', partitions = [ '0' ]) + + def test_broker_list_required(self): + with self.assertRaises(KafkaConfigurationError): + KafkaConsumer() |