there were two...

MorphiaOrg · Aug 26, 2023 · 40dab6b · 40dab6b
1 parent 5775654
commit 40dab6b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/core/src/test/java/dev/morphia/test/query/TestQuery.java b/core/src/test/java/dev/morphia/test/query/TestQuery.java
@@ -836,7 +836,7 @@ public void testMultipleConstraintsOnOneField() {
                 lt("size", 100));
 
         Map<String, Object> explain = query.explain();
-        Map<String, Object> inputStage = null;
+        Map<String, Object> inputStage;
 
         if (explain.get("explainVersion").equals("1")) {
             inputStage = walk(explain, List.of("queryPlanner", "winningPlan", "inputStage"));

diff --git a/core/src/test/java/dev/morphia/test/query/legacy/TestLegacyQuery.java b/core/src/test/java/dev/morphia/test/query/legacy/TestLegacyQuery.java
@@ -759,9 +759,13 @@ public void testMultipleConstraintsOnOneField() {
         query.field("size").lessThan(100);
 
         Map<String, Object> explain = query.explain();
-        Map<String, Object> queryPlanner = (Map<String, Object>) explain.get("queryPlanner");
-        Map<String, Object> winningPlan = (Map<String, Object>) queryPlanner.get("winningPlan");
-        Map<String, Object> inputStage = (Map<String, Object>) winningPlan.get("inputStage");
+        Map<String, Object> inputStage;
+
+        if (explain.get("explainVersion").equals("1")) {
+            inputStage = walk(explain, List.of("queryPlanner", "winningPlan", "inputStage"));
+        } else {
+            inputStage = walk(explain, List.of("queryPlanner", "winningPlan", "queryPlan", "inputStage"));
+        }
         assertEquals(inputStage.get("stage"), "IXSCAN");
     }