summaryrefslogtreecommitdiff
path: root/test/test_package.py
blob: eb530274fc58632a12210d24157f74813ac9449c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
from . import unittest


class TestPackage(unittest.TestCase):
    def test_top_level_namespace(self):
        import kafka as kafka1
        self.assertEqual(kafka1.KafkaConsumer.__name__, "KafkaConsumer")
        self.assertEqual(kafka1.consumer.__name__, "kafka.consumer")
        self.assertEqual(kafka1.codec.__name__, "kafka.codec")

    def test_submodule_namespace(self):
        import kafka.client as client1
        self.assertEqual(client1.__name__, "kafka.client")

        from kafka import client as client2
        self.assertEqual(client2.__name__, "kafka.client")

        from kafka.client import SimpleClient as SimpleClient1
        self.assertEqual(SimpleClient1.__name__, "SimpleClient")

        from kafka.codec import gzip_encode as gzip_encode1
        self.assertEqual(gzip_encode1.__name__, "gzip_encode")

        from kafka import SimpleClient as SimpleClient2
        self.assertEqual(SimpleClient2.__name__, "SimpleClient")

        from kafka.codec import snappy_encode
        self.assertEqual(snappy_encode.__name__, "snappy_encode")