add additional test for #1976
This commit is contained in:
parent
a657f29439
commit
f4873fee18
@ -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):
|
||||
|
Loading…
x
Reference in New Issue
Block a user