The node.bases has not to be tweaked otherwise leads to false positive unused-import due to the fact that six.with_metaclass is not consumed. Adds a unittest to check that bases attribute holds a call node and that ancestors attributes returns the correct class hierarchy.

1 files changed, 28 insertions, 0 deletions

diff --git a/tests/unittest_inference.py b/tests/unittest_inference.py
index c107eadb..10dd49d7 100644
--- a/tests/unittest_inference.py
+++ b/tests/unittest_inference.py
@@ -3793,6 +3793,34 @@ class InferenceTest(resources.SysPathSetup, unittest.TestCase):
         self.assertIsInstance(inferred, nodes.ClassDef)
         self.assertEqual(inferred.name, "B")
 
+    def test_With_metaclass_subclasses_inheritance(self):
+        ast_node = extract_node(
+            """
+        class A(type):
+            def test(cls):
+                return cls
+
+        class C:
+            pass
+
+        import six
+        class B(six.with_metaclass(A, C)):
+            pass
+
+        B #@
+        """
+        )
+        inferred = next(ast_node.infer())
+        self.assertIsInstance(inferred, nodes.ClassDef)
+        self.assertEqual(inferred.name, "B")
+        bases = inferred.bases
+        self.assertIsInstance(bases[0], nodes.Call)
+        ancestors = tuple(inferred.ancestors())
+        self.assertIsInstance(ancestors[0], nodes.ClassDef)
+        self.assertEqual(ancestors[0].name, "C")
+        self.assertIsInstance(ancestors[1], nodes.ClassDef)
+        self.assertEqual(ancestors[1].name, "object")
+
     def test_With_metaclass_with_partial_imported_name(self):
         ast_node = extract_node(
             """