Merge pull request #158 from wizzat/add_tests

Improve Tests, fix connection error timeout, other issues

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' ])