summaryrefslogtreecommitdiff
path: root/src/mongo/db/exec/inclusion_projection_executor_test.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/mongo/db/exec/inclusion_projection_executor_test.cpp')
-rw-r--r--src/mongo/db/exec/inclusion_projection_executor_test.cpp198
1 files changed, 90 insertions, 108 deletions
diff --git a/src/mongo/db/exec/inclusion_projection_executor_test.cpp b/src/mongo/db/exec/inclusion_projection_executor_test.cpp
index 2176652c437..de2464f7f11 100644
--- a/src/mongo/db/exec/inclusion_projection_executor_test.cpp
+++ b/src/mongo/db/exec/inclusion_projection_executor_test.cpp
@@ -549,23 +549,22 @@ TEST_F(InclusionProjectionExecutionTestWithoutFallBackToDefault,
ShouldApplyDottedInclusionToEachElementInArray) {
auto inclusion = makeInclusionProjectionWithDefaultPolicies(BSON("a.b" << true));
- vector<Value> nestedValues = {Value(1),
- Value(Document{}),
- Value(Document{{"b", 1}}),
- Value(Document{{"b", 1}, {"c", 2}}),
- Value(vector<Value>{}),
- Value(vector<Value>{Value(1), Value(Document{{"c", 1}})})};
-
// Drops non-documents and non-arrays. Applies projection to documents, recurses on
// nested arrays.
- vector<Value> expectedNestedValues = {Value(),
- Value(Document{}),
- Value(Document{{"b", 1}}),
- Value(Document{{"b", 1}}),
- Value(vector<Value>{}),
- Value(vector<Value>{Value(), Value(Document{})})};
- auto result = inclusion->applyTransformation(Document{{"a", nestedValues}});
- auto expectedResult = Document{{"a", expectedNestedValues}};
+ auto result = inclusion->applyTransformation(Document{{"a",
+ {1,
+ Document{},
+ Document{{"b", 1}},
+ Document{{"b", 1}, {"c", 2}},
+ vector<Value>{},
+ {1, Document{{"c", 1}}}}}});
+ auto expectedResult = Document{{"a",
+ {Value(),
+ Document{},
+ Document{{"b", 1}},
+ Document{{"b", 1}},
+ vector<Value>{},
+ {Value(), Document{}}}}};
ASSERT_DOCUMENT_EQ(result, expectedResult);
}
@@ -627,22 +626,21 @@ TEST_F(InclusionProjectionExecutionTestWithFallBackToDefault,
auto inclusion =
makeInclusionProjectionWithDefaultPolicies(BSON("a.b" << wrapInLiteral("COMPUTED")));
- vector<Value> nestedValues = {Value(1),
- Value(Document{}),
- Value(Document{{"b", 1}}),
- Value(Document{{"b", 1}, {"c", 2}}),
- Value(vector<Value>{}),
- Value(vector<Value>{Value(1), Value(Document{{"c", 1}})})};
- vector<Value> expectedNestedValues = {
- Value(Document{{"b", "COMPUTED"_sd}}),
- Value(Document{{"b", "COMPUTED"_sd}}),
- Value(Document{{"b", "COMPUTED"_sd}}),
- Value(Document{{"b", "COMPUTED"_sd}}),
- Value(vector<Value>{}),
- Value(vector<Value>{Value(Document{{"b", "COMPUTED"_sd}}),
- Value(Document{{"b", "COMPUTED"_sd}})})};
- auto result = inclusion->applyTransformation(Document{{"a", nestedValues}});
- auto expectedResult = Document{{"a", expectedNestedValues}};
+ auto result = inclusion->applyTransformation(Document{{"a",
+ {1,
+ Document{},
+ Document{{"b", 1}},
+ Document{{"b", 1}, {"c", 2}},
+ vector<Value>{},
+ {1, Document{{"c", 1}}}}}});
+ auto expectedResult =
+ Document{{"a",
+ {Document{{"b", "COMPUTED"_sd}},
+ Document{{"b", "COMPUTED"_sd}},
+ Document{{"b", "COMPUTED"_sd}},
+ Document{{"b", "COMPUTED"_sd}},
+ vector<Value>{},
+ {Document{{"b", "COMPUTED"_sd}}, Document{{"b", "COMPUTED"_sd}}}}}};
ASSERT_DOCUMENT_EQ(result, expectedResult);
}
@@ -651,26 +649,26 @@ TEST_F(InclusionProjectionExecutionTestWithFallBackToDefault,
auto inclusion = makeInclusionProjectionWithDefaultPolicies(
BSON("a.inc" << true << "a.comp" << wrapInLiteral("COMPUTED")));
- vector<Value> nestedValues = {Value(1),
- Value(Document{}),
- Value(Document{{"inc", 1}}),
- Value(Document{{"inc", 1}, {"c", 2}}),
- Value(Document{{"c", 2}, {"inc", 1}}),
- Value(Document{{"inc", 1}, {"c", 2}, {"comp", "original"_sd}}),
- Value(vector<Value>{}),
- Value(vector<Value>{Value(1), Value(Document{{"inc", 1}})})};
- vector<Value> expectedNestedValues = {
- Value(Document{{"comp", "COMPUTED"_sd}}),
- Value(Document{{"comp", "COMPUTED"_sd}}),
- Value(Document{{"inc", 1}, {"comp", "COMPUTED"_sd}}),
- Value(Document{{"inc", 1}, {"comp", "COMPUTED"_sd}}),
- Value(Document{{"inc", 1}, {"comp", "COMPUTED"_sd}}),
- Value(Document{{"inc", 1}, {"comp", "COMPUTED"_sd}}),
- Value(vector<Value>{}),
- Value(vector<Value>{Value(Document{{"comp", "COMPUTED"_sd}}),
- Value(Document{{"inc", 1}, {"comp", "COMPUTED"_sd}})})};
- auto result = inclusion->applyTransformation(Document{{"a", nestedValues}});
- auto expectedResult = Document{{"a", expectedNestedValues}};
+ auto result = inclusion->applyTransformation(
+ Document{{"a",
+ {1,
+ Document{},
+ Document{{"inc", 1}},
+ Document{{"inc", 1}, {"c", 2}},
+ Document{{"c", 2}, {"inc", 1}},
+ Document{{"inc", 1}, {"c", 2}, {"comp", "original"_sd}},
+ vector<Value>{},
+ {1, Document{{"inc", 1}}}}}});
+ auto expectedResult = Document{
+ {"a",
+ {Document{{"comp", "COMPUTED"_sd}},
+ Document{{"comp", "COMPUTED"_sd}},
+ Document{{"inc", 1}, {"comp", "COMPUTED"_sd}},
+ Document{{"inc", 1}, {"comp", "COMPUTED"_sd}},
+ Document{{"inc", 1}, {"comp", "COMPUTED"_sd}},
+ Document{{"inc", 1}, {"comp", "COMPUTED"_sd}},
+ vector<Value>{},
+ {Document{{"comp", "COMPUTED"_sd}}, Document{{"inc", 1}, {"comp", "COMPUTED"_sd}}}}}};
ASSERT_DOCUMENT_EQ(result, expectedResult);
}
@@ -916,18 +914,12 @@ TEST_F(InclusionProjectionExecutionTestWithoutFallBackToDefault,
auto inclusion = makeInclusionProjectionWithDefaultPolicies(BSON("a.b" << true));
// {a: [1, {b: 2, c: 3}, [{b: 4, c: 5}], {d: 6}]} => {a: [{b: 2}, [{b: 4}], {}]}
- auto result = inclusion->applyTransformation(
- Document{{"a",
- vector<Value>{Value(1),
- Value(Document{{"b", 2}, {"c", 3}}),
- Value(vector<Value>{Value(Document{{"b", 4}, {"c", 5}})}),
- Value(Document{{"d", 6}})}}});
+ auto result = inclusion->applyTransformation(Document{
+ {"a",
+ {1, Document{{"b", 2}, {"c", 3}}, {Document{{"b", 4}, {"c", 5}}}, Document{{"d", 6}}}}});
- auto expectedResult = Document{{"a",
- vector<Value>{Value(),
- Value(Document{{"b", 2}}),
- Value(vector<Value>{Value(Document{{"b", 4}})}),
- Value(Document{})}}};
+ auto expectedResult =
+ Document{{"a", {Value(), Document{{"b", 2}}, {Document{{"b", 4}}}, Document{}}}};
ASSERT_DOCUMENT_EQ(result, expectedResult);
}
@@ -937,15 +929,11 @@ TEST_F(InclusionProjectionExecutionTestWithoutFallBackToDefault,
auto inclusion = makeInclusionProjectionWithNoArrayRecursion(BSON("a.b" << true));
// {a: [1, {b: 2, c: 3}, [{b: 4, c: 5}], {d: 6}]} => {a: [{b: 2}, {}]}
- auto result = inclusion->applyTransformation(
- Document{{"a",
- vector<Value>{Value(1),
- Value(Document{{"b", 2}, {"c", 3}}),
- Value(vector<Value>{Value(Document{{"b", 4}, {"c", 5}})}),
- Value(Document{{"d", 6}})}}});
+ auto result = inclusion->applyTransformation(Document{
+ {"a",
+ {1, Document{{"b", 2}, {"c", 3}}, {Document{{"b", 4}, {"c", 5}}}, Document{{"d", 6}}}}});
- auto expectedResult = Document{
- {"a", vector<Value>{Value(), Value(Document{{"b", 2}}), Value(), Value(Document{})}}};
+ auto expectedResult = Document{{"a", {Value(), Document{{"b", 2}}, Value(), Document{}}}};
ASSERT_DOCUMENT_EQ(result, expectedResult);
}
@@ -955,12 +943,9 @@ TEST_F(InclusionProjectionExecutionTestWithoutFallBackToDefault,
auto inclusion = makeInclusionProjectionWithNoArrayRecursion(BSON("a" << true));
// {a: [1, {b: 2, c: 3}, [{b: 4, c: 5}], {d: 6}]} => [output doc identical to input]
- const auto inputDoc =
- Document{{"a",
- vector<Value>{Value(1),
- Value(Document{{"b", 2}, {"c", 3}}),
- Value(vector<Value>{Value(Document{{"b", 4}, {"c", 5}})}),
- Value(Document{{"d", 6}})}}};
+ const auto inputDoc = Document{
+ {"a",
+ {1, Document{{"b", 2}, {"c", 3}}, {Document{{"b", 4}, {"c", 5}}}, Document{{"d", 6}}}}};
auto result = inclusion->applyTransformation(inputDoc);
const auto& expectedResult = inputDoc;
@@ -973,22 +958,21 @@ TEST_F(InclusionProjectionExecutionTestWithFallBackToDefault,
auto inclusion =
makeInclusionProjectionWithDefaultPolicies(BSON("a.b" << wrapInLiteral("COMPUTED")));
- vector<Value> nestedValues = {Value(1),
- Value(Document{}),
- Value(Document{{"b", 1}}),
- Value(Document{{"b", 1}, {"c", 2}}),
- Value(vector<Value>{}),
- Value(vector<Value>{Value(1), Value(Document{{"c", 1}})})};
- vector<Value> expectedNestedValues = {
- Value(Document{{"b", "COMPUTED"_sd}}),
- Value(Document{{"b", "COMPUTED"_sd}}),
- Value(Document{{"b", "COMPUTED"_sd}}),
- Value(Document{{"b", "COMPUTED"_sd}}),
- Value(vector<Value>{}),
- Value(vector<Value>{Value(Document{{"b", "COMPUTED"_sd}}),
- Value(Document{{"b", "COMPUTED"_sd}})})};
- auto result = inclusion->applyTransformation(Document{{"a", nestedValues}});
- auto expectedResult = Document{{"a", expectedNestedValues}};
+ auto result = inclusion->applyTransformation(Document{{"a",
+ {1,
+ Document{},
+ Document{{"b", 1}},
+ Document{{"b", 1}, {"c", 2}},
+ vector<Value>{},
+ {1, Document{{"c", 1}}}}}});
+ auto expectedResult =
+ Document{{"a",
+ {Document{{"b", "COMPUTED"_sd}},
+ Document{{"b", "COMPUTED"_sd}},
+ Document{{"b", "COMPUTED"_sd}},
+ Document{{"b", "COMPUTED"_sd}},
+ vector<Value>{},
+ {Document{{"b", "COMPUTED"_sd}}, Document{{"b", "COMPUTED"_sd}}}}}};
ASSERT_DOCUMENT_EQ(result, expectedResult);
}
@@ -1002,22 +986,20 @@ TEST_F(InclusionProjectionExecutionTestWithFallBackToDefault,
// of the array and all nested arrays individually. With kDoNotRecurseNestedArrays, the
// nested arrays are replaced rather than being traversed, in exactly the same way as
// scalar values.
- vector<Value> nestedValues = {Value(1),
- Value(Document{}),
- Value(Document{{"b", 1}}),
- Value(Document{{"b", 1}, {"c", 2}}),
- Value(vector<Value>{}),
- Value(vector<Value>{Value(1), Value(Document{{"c", 1}})})};
-
- vector<Value> expectedNestedValues = {Value(Document{{"b", "COMPUTED"_sd}}),
- Value(Document{{"b", "COMPUTED"_sd}}),
- Value(Document{{"b", "COMPUTED"_sd}}),
- Value(Document{{"b", "COMPUTED"_sd}}),
- Value(Document{{"b", "COMPUTED"_sd}}),
- Value(Document{{"b", "COMPUTED"_sd}})};
-
- auto result = inclusion->applyTransformation(Document{{"a", nestedValues}});
- auto expectedResult = Document{{"a", expectedNestedValues}};
+ auto result = inclusion->applyTransformation(Document{{"a",
+ {1,
+ Document{},
+ Document{{"b", 1}},
+ Document{{"b", 1}, {"c", 2}},
+ vector<Value>{},
+ {1, Document{{"c", 1}}}}}});
+ auto expectedResult = Document{{"a",
+ {Document{{"b", "COMPUTED"_sd}},
+ Document{{"b", "COMPUTED"_sd}},
+ Document{{"b", "COMPUTED"_sd}},
+ Document{{"b", "COMPUTED"_sd}},
+ Document{{"b", "COMPUTED"_sd}},
+ Document{{"b", "COMPUTED"_sd}}}}};
ASSERT_DOCUMENT_EQ(result, expectedResult);
}
} // namespace
View Lorry Controller Status