From 6fbdde63d8d39e00f9d33c8f5ed0d55caee09de6 Mon Sep 17 00:00:00 2001 From: Eddie Linder Date: Thu, 20 Dec 2018 00:57:02 +0200 Subject: [PATCH] Fix id in test_only_after_count ut --- tests/queryset/queryset.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/queryset/queryset.py b/tests/queryset/queryset.py index 586913aa..7490cad3 100644 --- a/tests/queryset/queryset.py +++ b/tests/queryset/queryset.py @@ -4736,19 +4736,19 @@ class QuerySetTest(unittest.TestCase): age = IntField() address = StringField() User.drop_collection() - User(name="User", age=50, - address="Moscow, Russia").save() + user = User(name="User", age=50, + address="Moscow, Russia").save() user_queryset = User.objects(age=50) 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() self.assertEqual(result, 1) 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):