summaryrefslogtreecommitdiff
path: root/test/test_consumer.py
diff options
context:
space:
mode:
authorMark Roberts <wizzat@gmail.com>2014-04-25 10:55:04 -0700
committerMark Roberts <wizzat@gmail.com>2014-04-25 10:55:04 -0700
commit57913f9f914a959f52bc9040a172f8c9ff77e491 (patch)
treefe5cc6c14283a4c9d9175a748ef97f7d55df6fd7 /test/test_consumer.py
parent0e50f33ec678f6d656d488ce8a4537f95bba003e (diff)
downloadkafka-python-57913f9f914a959f52bc9040a172f8c9ff77e491.tar.gz
Various fixes
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.
Diffstat (limited to 'test/test_consumer.py')
-rw-r--r--test/test_consumer.py22
1 files changed, 22 insertions, 0 deletions
diff --git a/test/test_consumer.py b/test/test_consumer.py
new file mode 100644
index 0000000..778d76a
--- /dev/null
+++ b/test/test_consumer.py
@@ -0,0 +1,22 @@
+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' ])