blob: b33e53719a231461942a3786e3dbfe875f41534b (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
from mock import MagicMock
from . import unittest
from kafka.consumer import SimpleConsumer
from kafka.consumer.new import 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()
|
