Ensure that generated enum values have the correct base classesenum-fixes

Enum values should share the same base classes as their defining class.
If this is not the case it may lead to wrong inference results when an
enum member is used - e.g. for the following snippet:

class X(enum.IntEnum):
    one = 1

print([1, 2][X.one])

pylint will detect a "invalid-sequence-index" error as the __index__
method of X.one is not detected.

1 files changed, 33 insertions, 0 deletions

diff --git a/astroid/tests/unittest_brain.py b/astroid/tests/unittest_brain.py
index 91a9a79..b0e0608 100644
--- a/astroid/tests/unittest_brain.py
+++ b/astroid/tests/unittest_brain.py
@@ -252,5 +252,38 @@ class EnumBrainTest(unittest.TestCase):
         meth = one.getattr('mymethod')[0]
         self.assertIsInstance(meth, astroid.Function)
 
+    def test_enum_multiple_base_classes(self):
+        module = AstroidBuilder().string_build(dedent("""
+        import enum
+
+        class Mixin:
+            pass
+
+        class MyEnum(Mixin, enum.Enum):
+            one = 1
+        """))
+        enum = next(module['MyEnum'].infer())
+        one = enum['one']
+
+        clazz = one.getattr('__class__')[0]
+        self.assertTrue(clazz.is_subtype_of('.Mixin'),
+                        'Enum instance should share base classes with generating class')
+
+    def test_int_enum(self):
+        module = AstroidBuilder().string_build(dedent("""
+        import enum
+
+        class MyEnum(enum.IntEnum):
+            one = 1
+        """))
+
+        enum = next(module['MyEnum'].infer())
+        one = enum['one']
+
+        clazz = one.getattr('__class__')[0]
+        int_type = '{}.{}'.format(bases.BUILTINS, 'int')
+        self.assertTrue(clazz.is_subtype_of(int_type),
+                        'IntEnum based enums should be a subtype of int')
+
 if __name__ == '__main__':
     unittest.main()