diff options
| -rw-r--r-- | kafka/__init__.py | 12 | ||||
| -rw-r--r-- | test/unit.py | 29 |
2 files changed, 40 insertions, 1 deletions
diff --git a/kafka/__init__.py b/kafka/__init__.py index 1e01ea9..41038d7 100644 --- a/kafka/__init__.py +++ b/kafka/__init__.py @@ -1 +1,11 @@ -__all__ = ["kafka","codec"] +__title__ = 'kafka' +__version__ = '0.1-alpha' +__author__ = 'David Arthur' +__license__ = 'Apache License 2.0' +__copyright__ = 'Copyright 2012, David Arthur under Apache License, v2.0' + +from .client import ( + KafkaClient, KafkaException, + Message, ProduceRequest, FetchRequest, OffsetRequest +) +from .codec import gzip_encode, gzip_decode diff --git a/test/unit.py b/test/unit.py index 6ca3564..7cb1aed 100644 --- a/test/unit.py +++ b/test/unit.py @@ -12,6 +12,35 @@ STRLEN = 100 def random_string(): return os.urandom(random.randint(0, STRLEN)) +class TestPackage(unittest.TestCase): + def test_top_level_namespace(self): + import kafka as kafka1 + self.assertEquals(kafka1.KafkaClient.__name__, "KafkaClient") + self.assertEquals(kafka1.gzip_encode.__name__, "gzip_encode") + self.assertEquals(kafka1.client.__name__, "kafka.client") + self.assertEquals(kafka1.codec.__name__, "kafka.codec") + + def test_submodule_namespace(self): + import kafka.client as client1 + self.assertEquals(client1.__name__, "kafka.client") + self.assertEquals(client1.KafkaClient.__name__, "KafkaClient") + + from kafka import client as client2 + self.assertEquals(client2.__name__, "kafka.client") + self.assertEquals(client2.KafkaClient.__name__, "KafkaClient") + + from kafka.client import KafkaClient as KafkaClient1 + self.assertEquals(KafkaClient1.__name__, "KafkaClient") + + from kafka.codec import gzip_encode as gzip_encode1 + self.assertEquals(gzip_encode1.__name__, "gzip_encode") + + from kafka import KafkaClient as KafkaClient2 + self.assertEquals(KafkaClient2.__name__, "KafkaClient") + + from kafka import gzip_encode as gzip_encode2 + self.assertEquals(gzip_encode2.__name__, "gzip_encode") + class TestMisc(unittest.TestCase): def test_length_prefix(self): for i in xrange(ITERATIONS): |