pub/mongoengine
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix id in test_only_after_count ut

Browse Source
This commit is contained in:
Eddie Linder 2018-12-20 00:57:02 +02:00
parent b04dc90cdf
commit 6fbdde63d8
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/queryset/queryset.py
View File

@ -4736,19 +4736,19 @@ class QuerySetTest(unittest.TestCase):
age = IntField() age = IntField()
address = StringField() address = StringField()
User.drop_collection() User.drop_collection()
User(name="User", age=50, user = User(name="User", age=50,
address="Moscow, Russia").save() address="Moscow, Russia").save()
user_queryset = User.objects(age=50) user_queryset = User.objects(age=50)
result = user_queryset.only("name", "age").as_pymongo().first() result = user_queryset.only("name", "age").as_pymongo().first()
self.assertEqual(result, {"name": "User", "age": 50}) self.assertEqual(result, {"_id": user.id, "name": "User", "age": 50})
result = user_queryset.count() result = user_queryset.count()
self.assertEqual(result, 1) self.assertEqual(result, 1)
result = user_queryset.only("name", "age").as_pymongo().first() result = user_queryset.only("name", "age").as_pymongo().first()
self.assertEqual(result, {"name": "User", "age": 50}) self.assertEqual(result, {"_id": user.id, "name": "User", "age": 50})
def test_no_dereference(self): def test_no_dereference(self):