945ecbcee7
- drop custom PartitionUnavailable exception - raise UnknownTopicOrPartitionError or LeaderNotAvailableError - add tests for exception raises
14 lines
361 B
Python
14 lines
361 B
Python
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' ])
|