summaryrefslogtreecommitdiff
path: root/test/test_consumer.py
blob: d5b4fdd6954386eda1ea3102f1627792532a2240 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
import os
import random
import struct
import unittest2

from mock import MagicMock, patch

from kafka.consumer import SimpleConsumer

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