fix tester

Author: idocyabra

tests/queryset/test_queryset.py

@@ -25,6 +25,7 @@
     queryset_manager,
 )
 from tests.utils import (
+    requires_mongodb_gte_42,
     requires_mongodb_gte_44,
     requires_mongodb_lt_42,
 )
@@ -2217,6 +2218,19 @@ class BlogPost(Document):
         post.reload()
         assert post.tags == ["code", "mongodb"]
 
+    @requires_mongodb_gte_42
+    def test_aggregation_update(self):
+        """Ensure that the 'aggregation_update' update works correctly."""
+
+        class BlogPost(Document):
+            slug = StringField()
+            tags = ListField(StringField())
+
+        BlogPost.drop_collection()
+
+        post = BlogPost(slug="test")
+        post.save()
+
         BlogPost.objects(slug="test").update(
             __raw__=[{"$set": {"slug": {"$concat": ["$slug", " ", "$slug"]}}}],
             aggregation_update=True,

tests/utils.py

@@ -37,6 +37,10 @@ def requires_mongodb_lt_42(func):
     return _decorated_with_ver_requirement(func, (4, 2), oper=operator.lt)
 
 
+def requires_mongodb_gte_42(func):
+    return _decorated_with_ver_requirement(func, (4, 2), oper=operator.ge)
+
+
 def requires_mongodb_gte_44(func):
     return _decorated_with_ver_requirement(func, (4, 4), oper=operator.ge)