57913f9f91
Bump version number to 0.9.1 Update readme to show supported Kafka/Python versions Validate arguments in consumer.py, add initial consumer unit test Make service kill() child processes when startup fails Add tests for util.py, fix Python 2.6 specific bug.
23 lines
640 B
Python
23 lines
640 B
Python
import os
|
|
import random
|
|
import struct
|
|
import unittest2
|
|
|
|
from mock import MagicMock, patch
|
|
|
|
from kafka import KafkaClient
|
|
from kafka.consumer import SimpleConsumer
|
|
from kafka.common import (
|
|
ProduceRequest, BrokerMetadata, PartitionMetadata,
|
|
TopicAndPartition, KafkaUnavailableError,
|
|
LeaderUnavailableError, PartitionUnavailableError
|
|
)
|
|
from kafka.protocol import (
|
|
create_message, KafkaProtocol
|
|
)
|
|
|
|
class TestKafkaConsumer(unittest2.TestCase):
|
|
def test_non_integer_partitions(self):
|
|
with self.assertRaises(AssertionError):
|
|
consumer = SimpleConsumer(MagicMock(), 'group', 'topic', partitions = [ '0' ])
|