diff --git a/tests/queryset/test_queryset.py b/tests/queryset/test_queryset.py index 6310b9cd..4d281c60 100644 --- a/tests/queryset/test_queryset.py +++ b/tests/queryset/test_queryset.py @@ -21,6 +21,7 @@ from mongoengine.queryset import ( QuerySetManager, queryset_manager, ) +from tests.utils import requires_mongodb_gte_44 class db_ops_tracker(query_counter): @@ -5656,6 +5657,7 @@ class TestQueryset(unittest.TestCase): qs = self.Person.objects().timeout(False) assert qs._cursor_args == {"no_cursor_timeout": True} + @requires_mongodb_gte_44 def test_allow_disk_use(self): qs = self.Person.objects() assert qs._cursor_args == {} @@ -5681,6 +5683,5 @@ class TestQueryset(unittest.TestCase): assert qs_disk[index] == qs[index] - if __name__ == "__main__": unittest.main() diff --git a/tests/utils.py b/tests/utils.py index 0899c208..adb0bdb4 100644 --- a/tests/utils.py +++ b/tests/utils.py @@ -1,3 +1,4 @@ +import operator import unittest import pytest @@ -33,6 +34,10 @@ def get_as_pymongo(doc): return doc.__class__.objects.as_pymongo().get(id=doc.id) +def requires_mongodb_gte_44(func): + return _decorated_with_ver_requirement(func, (4, 4), oper=operator.ge) + + def _decorated_with_ver_requirement(func, mongo_version_req, oper): """Return a MongoDB version requirement decorator.