Merge pull request #2445 from bagerard/fix_count_mongomock

Use Queryset._query instead of Cursor.__spec for count()

docs/changelog.rst

@@ -7,6 +7,7 @@ Development
 ===========
 - (Fill this out as you fix issues and develop your features).
 - Fix LazyReferenceField dereferencing in embedded documents #2426
+- Fix regarding the recent use of Cursor.__spec in .count() that was interfering with mongomock #2425
 
 Changes in 0.21.0
 =================

mongoengine/queryset/base.py

@@ -420,7 +420,7 @@ class BaseQuerySet:
 
         count = count_documents(
             collection=self._cursor.collection,
-            filter=self._cursor._Cursor__spec,
+            filter=self._query,
             **kwargs,
         )
 

tests/document/test_instance.py

@@ -2900,39 +2900,23 @@ class TestDocumentInstance(MongoDBTestCase):
         # Checks
         assert ",".join([str(b) for b in Book.objects.all()]) == "1,2,3,4,5,6,7,8,9"
         # bob related books
-        assert (
+        bob_books_qs = Book.objects.filter(
-            ",".join(
-                [
-                    str(b)
-                    for b in Book.objects.filter(
             Q(extra__a=bob) | Q(author=bob) | Q(extra__b=bob)
         )
-                ]
+        assert [str(b) for b in bob_books_qs] == ["1", "2", "3", "4"]
-            )
+        assert bob_books_qs.count() == 4
-            == "1,2,3,4"
-        )
 
         # Susan & Karl related books
-        assert (
+        susan_karl_books_qs = Book.objects.filter(
-            ",".join(
-                [
-                    str(b)
-                    for b in Book.objects.filter(
             Q(extra__a__all=[karl, susan])
             | Q(author__all=[karl, susan])
             | Q(extra__b__all=[karl.to_dbref(), susan.to_dbref()])
         )
-                ]
+        assert [str(b) for b in susan_karl_books_qs] == ["1"]
-            )
+        assert susan_karl_books_qs.count() == 1
-            == "1"
-        )
 
         # $Where
-        assert (
+        custom_qs = Book.objects.filter(
-            ",".join(
-                [
-                    str(b)
-                    for b in Book.objects.filter(
             __raw__={
                 "$where": """
                                             function(){
@@ -2940,10 +2924,8 @@ class TestDocumentInstance(MongoDBTestCase):
                                                        this.name == '2';}"""
             }
         )
-                ]
+        assert [str(b) for b in custom_qs] == ["1", "2"]
-            )
+        assert custom_qs.count() == 2
-            == "1,2"
-        )
 
     def test_switch_db_instance(self):
         register_connection("testdb-1", "mongoenginetest2")

tests/queryset/test_transform.py

@@ -104,18 +104,17 @@ class TestTransform(unittest.TestCase):
         post = BlogPost(**data)
         post.save()
 
-        assert "postTitle" in BlogPost.objects(title=data["title"])._query
+        qs = BlogPost.objects(title=data["title"])
-        assert not ("title" in BlogPost.objects(title=data["title"])._query)
+        assert qs._query == {"postTitle": data["title"]}
-        assert BlogPost.objects(title=data["title"]).count() == 1
+        assert qs.count() == 1
 
-        assert "_id" in BlogPost.objects(pk=post.id)._query
+        qs = BlogPost.objects(pk=post.id)
-        assert BlogPost.objects(pk=post.id).count() == 1
+        assert qs._query == {"_id": post.id}
+        assert qs.count() == 1
 
-        assert (
+        qs = BlogPost.objects(comments__content="test")
-            "postComments.commentContent"
+        assert qs._query == {"postComments.commentContent": "test"}
-            in BlogPost.objects(comments__content="test")._query
+        assert qs.count() == 1
-        )
-        assert BlogPost.objects(comments__content="test").count() == 1
 
         BlogPost.drop_collection()