list_indexes support for multiple inheritance

tests/document/class_methods.py

@@ -152,6 +152,43 @@ class ClassMethodsTest(unittest.TestCase):
         BlogPostWithTags._get_collection().drop_index('_cls_1_author_1_tags_1')
         self.assertEqual(BlogPost.compare_indexes(), { 'missing': [[('_cls', 1), ('author', 1), ('tags', 1)]], 'extra': [] })
 
+    def test_compare_indexes_multiple_subclasses(self):
+        """ Ensure that compare_indexes behaves correctly if called from a
+        class, which base class has multiple subclasses
+        """
+
+        class BlogPost(Document):
+            author = StringField()
+            title = StringField()
+            description = StringField()
+
+            meta = {
+                'allow_inheritance': True
+            }
+
+        class BlogPostWithTags(BlogPost):
+            tags = StringField()
+            tag_list = ListField(StringField())
+
+            meta = {
+                'indexes': [('author', 'tags')]
+            }
+
+        class BlogPostWithCustomField(BlogPost):
+            custom = DictField()
+
+            meta = {
+                'indexes': [('author', 'custom')]
+            }
+
+        BlogPost.ensure_indexes()
+        BlogPostWithTags.ensure_indexes()
+        BlogPostWithCustomField.ensure_indexes()
+
+        self.assertEqual(BlogPost.compare_indexes(), { 'missing': [], 'extra': [] })
+        self.assertEqual(BlogPostWithTags.compare_indexes(), { 'missing': [], 'extra': [] })
+        self.assertEqual(BlogPostWithCustomField.compare_indexes(), { 'missing': [], 'extra': [] })
+
     def test_list_indexes_inheritance(self):
         """ ensure that all of the indexes are listed regardless of the super-
         or sub-class that we call it from
@@ -190,7 +227,6 @@ class ClassMethodsTest(unittest.TestCase):
                          BlogPostWithTags.list_indexes())
         self.assertEqual(BlogPost.list_indexes(),
                          BlogPostWithTagsAndExtraText.list_indexes())
-        print BlogPost.list_indexes()
         self.assertEqual(BlogPost.list_indexes(),
                          [[('_cls', 1), ('author', 1), ('tags', 1)],
                          [('_cls', 1), ('author', 1), ('tags', 1), ('extra_text', 1)],