Fixed MQL

mongodb-forks · Nov 15, 2024 · 65b29f9 · 65b29f9
1 parent 025026e
commit 65b29f9
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/tests/raw_query/tests.py b/tests/raw_query/tests.py
@@ -311,20 +311,23 @@ def test_annotations(self):
             {
                 "$lookup": {
                     "from": "raw_query_book",
-                    "localField": "id",
+                    "localField": "_id",
                     "foreignField": "author_id",
                     "as": "books",
                 }
             },
+            {"$addFields": {"book_count": {"$size": "$books"}}},
+            {"$project": {"books": 0}},
             {
-                "$project": {
-                    "first_name": 1,
-                    "last_name": 1,
-                    "dob": 1,
-                    "book_count": {"$size": "$books"},
+                "$group": {
+                    "_id": "$_id",
+                    "first_name": {"$first": "$first_name"},
+                    "last_name": {"$first": "$last_name"},
+                    "dob": {"$first": "$dob"},
+                    "book_count": {"$first": "$book_count"},
                 }
             },
-            {"$sort": {"id": 1}},
+            {"$sort": {"_id": 1}},
         ]
         expected_annotations = (
             ("book_count", 3),