add additional test for #1976

This commit is contained in:
Bastien Gérard 2019-02-17 22:50:42 +01:00
parent a657f29439
commit f4873fee18

View File

@ -4628,8 +4628,6 @@ class QuerySetTest(unittest.TestCase):
self.assertEqual(doc_objects, Doc.objects.from_json(json_data))
def test_as_pymongo(self):
from decimal import Decimal
class LastLogin(EmbeddedDocument):
location = StringField()
ip = StringField()
@ -4694,6 +4692,24 @@ class QuerySetTest(unittest.TestCase):
}
})
def test_as_pymongo_returns_cls_attribute_when_using_inheritance(self):
class User(Document):
name = StringField()
meta = {'allow_inheritance': True}
User.drop_collection()
user = User(name="Bob Dole").save()
result = User.objects.as_pymongo().first()
self.assertEqual(
result,
{
'_cls': 'User',
'_id': user.id,
'name': 'Bob Dole'
}
)
def test_as_pymongo_json_limit_fields(self):
class User(Document):