Merge pull request #507 from dpkp/deprecation_warnings

Add DeprecationWarnings to legacy classes

1 files changed, 7 insertions, 8 deletions

diff --git a/test/test_package.py b/test/test_package.py
index e91753c..eb53027 100644
--- a/test/test_package.py
+++ b/test/test_package.py
@@ -1,29 +1,28 @@
 from . import unittest
 
+
 class TestPackage(unittest.TestCase):
     def test_top_level_namespace(self):
         import kafka as kafka1
-        self.assertEqual(kafka1.KafkaClient.__name__, "KafkaClient")
-        self.assertEqual(kafka1.client.__name__, "kafka.client")
+        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")
-        self.assertEqual(client1.KafkaClient.__name__, "KafkaClient")
 
         from kafka import client as client2
         self.assertEqual(client2.__name__, "kafka.client")
-        self.assertEqual(client2.KafkaClient.__name__, "KafkaClient")
 
-        from kafka.client import KafkaClient as KafkaClient1
-        self.assertEqual(KafkaClient1.__name__, "KafkaClient")
+        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 KafkaClient as KafkaClient2
-        self.assertEqual(KafkaClient2.__name__, "KafkaClient")
+        from kafka import SimpleClient as SimpleClient2
+        self.assertEqual(SimpleClient2.__name__, "SimpleClient")
 
         from kafka.codec import snappy_encode
         self.assertEqual(snappy_encode.__name__, "snappy_encode")