Skip to content

Commit 9068d1a

Browse files
committed
Fix tests for pymongo < 4.4
1 parent 34e7cd7 commit 9068d1a

File tree

1 file changed

+11
-5
lines changed

1 file changed

+11
-5
lines changed

tests/queryset/test_queryset.py

Lines changed: 11 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717
MONGODB_36,
1818
get_mongodb_version,
1919
)
20+
from mongoengine.pymongo_support import PYMONGO_VERSION
2021
from mongoengine.queryset import (
2122
DoesNotExist,
2223
MultipleObjectsReturned,
@@ -2701,13 +2702,18 @@ def test_order_by_using_raw(self):
27012702
]
27022703
assert names == [("User A", 20), ("User B", 30), ("User B", 25), ("User C", 40)]
27032704

2704-
names = [
2705-
(p.name, p.age)
2706-
for p in self.Person.objects.order_by(
2705+
if PYMONGO_VERSION >= (4, 4):
2706+
# Pymongo >= 4.4 allow to mix single key with tuples inside the list
2707+
qs = self.Person.objects.order_by(
27072708
__raw__=["name", ("age", pymongo.ASCENDING)]
27082709
)
2709-
]
2710-
assert names == [("User A", 20), ("User B", 25), ("User B", 30), ("User C", 40)]
2710+
names = [(p.name, p.age) for p in qs]
2711+
assert names == [
2712+
("User A", 20),
2713+
("User B", 25),
2714+
("User B", 30),
2715+
("User C", 40),
2716+
]
27112717

27122718
def test_order_by_using_raw_and_keys_raises_exception(self):
27132719
with pytest.raises(OperationError):

0 commit comments

Comments
 (0)