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))
|
self.assertEqual(doc_objects, Doc.objects.from_json(json_data))
|
||||||
|
|
||||||
def test_as_pymongo(self):
|
def test_as_pymongo(self):
|
||||||
from decimal import Decimal
|
|
||||||
|
|
||||||
class LastLogin(EmbeddedDocument):
|
class LastLogin(EmbeddedDocument):
|
||||||
location = StringField()
|
location = StringField()
|
||||||
ip = 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):
|
def test_as_pymongo_json_limit_fields(self):
|
||||||
|
|
||||||
class User(Document):
|
class User(Document):
|
||||||
|
Loading…
x
Reference in New Issue
Block a user