test/test_consumer.py


1
2
3
4
5
6
7
8
9
10


from mock import MagicMock
from . import unittest

from kafka.consumer import SimpleConsumer

class TestKafkaConsumer(unittest.TestCase):
    def test_non_integer_partitions(self):
        with self.assertRaises(AssertionError):
            SimpleConsumer(MagicMock(), 'group', 'topic', partitions = [ '0' ])