SERVER-37503 Support $type with wildcard index

1 files changed, 23 insertions, 0 deletions

diff --git a/src/mongo/db/query/query_planner_wildcard_index_test.cpp b/src/mongo/db/query/query_planner_wildcard_index_test.cpp
index 025f1ce9fa5..03af6b93227 100644
--- a/src/mongo/db/query/query_planner_wildcard_index_test.cpp
+++ b/src/mongo/db/query/query_planner_wildcard_index_test.cpp
@@ -1817,4 +1817,27 @@ TEST_F(QueryPlannerWildcardTest, ContainedOrPushdownWorksWithWildcardIndex) {
         "{$_path: [['a','a',true,true]], a:[[1,1,true,true]]}}}}}");
 }
 
+TEST_F(QueryPlannerWildcardTest, TypeOfObjectWithWildcardIndex) {
+    addWildcardIndex(BSON("$**" << 1));
+    runQuery(fromjson("{a: {$type: 'object'}}"));
+
+    assertNumSolutions(1U);
+    assertSolutionExists(
+        "{fetch: {filter: {a: {$type: [3]}}, node:"
+        "{ixscan: {filter: null, pattern: {$_path: 1, a: 1}, bounds:"
+        "{$_path: [['a','a',true,true], ['a.','a/', true, false]], "
+        "a:[['MinKey','MaxKey',true,true]]}}}}}");
+}
+
+TEST_F(QueryPlannerWildcardTest, TypeOfArrayWithWildcardIndex) {
+    addWildcardIndex(BSON("$**" << 1));
+    runQuery(fromjson("{a: {$type: 'array'}}"));
+
+    assertNumSolutions(1U);
+    assertSolutionExists(
+        "{fetch: {filter: {a: {$type: [4]}}, node:"
+        "{ixscan: {filter: null, pattern: {$_path: 1, a: 1}, bounds:"
+        "{$_path: [['a','a',true,true], ['a.','a/', true, false]], "
+        "a:[['MinKey','MaxKey',true,true]]}}}}}");
+}
 }  // namespace mongo