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