fixing incompatibility with mongoengine aggregation to support mongo 4.4

This commit is contained in:
Bastien Gerard
2021-02-28 14:07:15 +01:00
parent 2d1c9afbb7
commit 8ef771912d
5 changed files with 31 additions and 10 deletions

View File

@@ -7,6 +7,7 @@ import pytest
from mongoengine import *
from mongoengine.connection import get_db
from mongoengine.mongodb_support import MONGODB_44, get_mongodb_version
class TestIndexes(unittest.TestCase):
@@ -452,9 +453,11 @@ class TestIndexes(unittest.TestCase):
.get("stage")
== "IXSCAN"
)
mongo_db = get_mongodb_version()
PROJECTION_STR = "PROJECTION" if mongo_db < MONGODB_44 else "PROJECTION_COVERED"
assert (
query_plan.get("queryPlanner").get("winningPlan").get("stage")
== "PROJECTION"
== PROJECTION_STR
)
query_plan = Test.objects(a=1).explain()