fix-#397: Allow specifying the '_cls' as a field for indexes

2014-08-03 12:54:22 +03:00
parent 41cfe5d2ca
commit 9b30afeca9
9 changed files with 46 additions and 8 deletions
--- a/tests/document/class_methods.py
+++ b/tests/document/class_methods.py
@@ -36,9 +36,9 @@ class ClassMethodsTest(unittest.TestCase):
    def test_definition(self):
        """Ensure that document may be defined using fields.
        """
-        self.assertEqual(['age', 'id', 'name'],
+        self.assertEqual(['_cls', 'age', 'id', 'name'],
                         sorted(self.Person._fields.keys()))
-        self.assertEqual(["IntField", "ObjectIdField", "StringField"],
+        self.assertEqual(["IntField", "ObjectIdField", "StringField", "StringField"],
                        sorted([x.__class__.__name__ for x in
                                self.Person._fields.values()]))