Add Deprecated Class unittests (#4448)

* Add Deprecated Class unittests

1 files changed, 55 insertions, 0 deletions

diff --git a/tests/checkers/unittest_deprecated.py b/tests/checkers/unittest_deprecated.py
index a15596abc..f2a51aa05 100644
--- a/tests/checkers/unittest_deprecated.py
+++ b/tests/checkers/unittest_deprecated.py
@@ -15,6 +15,9 @@ class _DeprecatedChecker(DeprecatedMixin, BaseChecker):
     def deprecated_modules(self):
         return {"deprecated_module"}
 
+    def deprecated_classes(self, module):
+        return ["DeprecatedClass"] if module == "deprecated" else []
+
     def deprecated_arguments(self, method):
         if method == "myfunction1":
             # def myfunction1(arg1, deprecated_arg1='spam')
@@ -416,3 +419,55 @@ class TestDeprecatedChecker(CheckerTestCase):
             )
         ):
             self.checker.visit_importfrom(node)
+
+    def test_deprecated_class_import_from(self):
+        # Tests detecting deprecated class via import from
+        node = astroid.extract_node(
+            """
+        from .deprecated import DeprecatedClass
+        """
+        )
+        with self.assertAddsMessages(
+            Message(
+                msg_id="deprecated-class",
+                args=("DeprecatedClass", "deprecated"),
+                node=node,
+                confidence=UNDEFINED,
+            )
+        ):
+            self.checker.visit_importfrom(node)
+
+    def test_deprecated_class_import(self):
+        # Tests detecting deprecated class via import
+        node = astroid.extract_node(
+            """
+        import deprecated.DeprecatedClass
+        """
+        )
+        with self.assertAddsMessages(
+            Message(
+                msg_id="deprecated-class",
+                args=("DeprecatedClass", "deprecated"),
+                node=node,
+                confidence=UNDEFINED,
+            )
+        ):
+            self.checker.visit_import(node)
+
+    def test_deprecated_class_call(self):
+        # Tests detecting deprecated class via call
+        node = astroid.extract_node(
+            """
+        import deprecated
+        deprecated.DeprecatedClass()
+        """
+        )
+        with self.assertAddsMessages(
+            Message(
+                msg_id="deprecated-class",
+                args=("DeprecatedClass", "deprecated"),
+                node=node,
+                confidence=UNDEFINED,
+            )
+        ):
+            self.checker.visit_call(node)