diff options
Diffstat (limited to 'src/mongo/db/pipeline/monotonic_expression_test.cpp')
| -rw-r--r-- | src/mongo/db/pipeline/monotonic_expression_test.cpp | 149 |
1 files changed, 149 insertions, 0 deletions
diff --git a/src/mongo/db/pipeline/monotonic_expression_test.cpp b/src/mongo/db/pipeline/monotonic_expression_test.cpp new file mode 100644 index 00000000000..0c667108201 --- /dev/null +++ b/src/mongo/db/pipeline/monotonic_expression_test.cpp @@ -0,0 +1,149 @@ +/** + * Copyright (C) 2022-present MongoDB, Inc. + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the Server Side Public License, version 1, + * as published by MongoDB, Inc. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * Server Side Public License for more details. + * + * You should have received a copy of the Server Side Public License + * along with this program. If not, see + * <http://www.mongodb.com/licensing/server-side-public-license>. + * + * As a special exception, the copyright holders give permission to link the + * code of portions of this program with the OpenSSL library under certain + * conditions as described in each individual source file and distribute + * linked combinations including the program with the OpenSSL library. You + * must comply with the Server Side Public License in all respects for + * all of the code used other than as permitted herein. If you modify file(s) + * with this exception, you may extend this exception to your version of the + * file(s), but you are not obligated to do so. If you do not wish to do so, + * delete this exception statement from your version. If you delete this + * exception statement from all source files in the program, then also delete + * it in the license file. + */ + +#include "mongo/db/pipeline/expression_context_for_test.h" +#include "mongo/db/pipeline/monotonic_expression.h" +#include "mongo/db/service_context_d_test_fixture.h" + +namespace mongo { +namespace { + +class MonotonicExpressionFixture : public ServiceContextMongoDTest { +public: + bool isMonotonicExpression(BSONObj expressionSpec, const FieldPath& monotonicField) { + auto expression = + Expression::parseExpression(&_expCtx, expressionSpec, _expCtx.variablesParseState); + return expression->isMonotonic(monotonicField); + } + +private: + ExpressionContextForTest _expCtx; +}; + +} // namespace + +TEST_F(MonotonicExpressionFixture, ConstIsMonotonic) { + ASSERT_TRUE(isMonotonicExpression(BSON("$const" << 1), "a")); +} + +TEST_F(MonotonicExpressionFixture, MonotonicFieldIsMonotonic) { + ASSERT_TRUE(isMonotonicExpression(BSON("$add" << BSON_ARRAY("$a")), "a")); +} + +TEST_F(MonotonicExpressionFixture, NonMonotonicFieldIsNonMonotonic) { + ASSERT_FALSE(isMonotonicExpression(BSON("$add" << BSON_ARRAY("$b")), "a")); +} + +TEST_F(MonotonicExpressionFixture, MonotonicWithOppositeSignIsStillMonotonic) { + ASSERT_TRUE(isMonotonicExpression(BSON("$subtract" << BSON_ARRAY(1 << "$a")), "a")); +} + +TEST_F(MonotonicExpressionFixture, NestedOppositesAreProcessedCorrectlyForMonotonic) { + // Test expression is: dateDiff(1 - (1 - (1 - a))), a) + const auto& startDate = fromjson("{$subtract: [1, {$subtract: [1, {$subtract: [1, '$a']}]}]}"); + const auto& expression = BSON("$dateDiff" << BSON("startDate" << startDate << "endDate" + << "$a" + << "unit" + << "hours")); + ASSERT_TRUE(isMonotonicExpression(expression, "a")); +} + +TEST_F(MonotonicExpressionFixture, NestedOppositesAreProcessedCorrectlyForNonMonotonic) { + // LHS expression (1 - (1 - floor(a)) - 1 is increasing assuming $a is increasing. + const auto& lhs = + fromjson("{$subtract: [{$subtract: [1, {$subtract: [1, {$floor: '$a'}]}]}, 1]}"); + // RHS expression 1 - (1 - (1 - ceil(a))) is decreasing assuming $a is decreasing. + const auto& rhs = + fromjson("{$subtract: [1, {$subtract: [1, {$subtract: [1, {$ceil: '$a'}]}]}]}"); + // Both LHS and RHS are monotonic + ASSERT_TRUE(isMonotonicExpression(lhs, "a")); + ASSERT_TRUE(isMonotonicExpression(rhs, "a")); + // Because they are monotonic in different directions, their sum is non monotonic, but their + // difference is monotonic. + ASSERT_FALSE(isMonotonicExpression(BSON("$add" << BSON_ARRAY(lhs << rhs)), "a")); + ASSERT_TRUE(isMonotonicExpression(BSON("$subtract" << BSON_ARRAY(lhs << rhs)), "a")); +} + +TEST_F(MonotonicExpressionFixture, MonotonicAndNonMonotonicFieldIsNonMonotonic) { + ASSERT_FALSE(isMonotonicExpression(BSON("$add" << BSON_ARRAY("$a" + << "$b")), + "a")); +} + +TEST_F(MonotonicExpressionFixture, MonotonicAndConstantIsMonotonic) { + ASSERT_TRUE(isMonotonicExpression(BSON("$add" << BSON_ARRAY("$a" + << "1")), + "a")); +} + +TEST_F(MonotonicExpressionFixture, ConstantAndConstantIsMonotonic) { + ASSERT_TRUE(isMonotonicExpression(BSON("$add" << BSON_ARRAY("1" + << "2")), + "a")); +} + +TEST_F(MonotonicExpressionFixture, MonotonicAndOppositeMonotonicIsNonMonotonic) { + ASSERT_FALSE(isMonotonicExpression(BSON("$subtract" << BSON_ARRAY("$a" + << "$a")), + "a")); +} + +TEST_F(MonotonicExpressionFixture, MonotonicAndMonotonicIsMonotonic) { + const auto& dateTrunc = BSON("$dateTrunc" << BSON("date" + << "$time" + << "unit" + << "hour" + << "timezone" + << "America/New_York")); + ASSERT_TRUE(isMonotonicExpression(BSON("$add" << BSON_ARRAY(dateTrunc << "$time")), "time")); + ASSERT_TRUE(isMonotonicExpression(BSON("$add" << BSON_ARRAY("$time" + << "$time")), + "time")); +} +TEST_F(MonotonicExpressionFixture, FunctionWithConstantNonMonotonicChildrenIsMonotonic) { + ASSERT_TRUE(isMonotonicExpression(BSON("$dateTrunc" << BSON("date" + << "$time" + << "unit" + << "hour" + << "timezone" + << "America/New_York")), + "time")); +} + +TEST_F(MonotonicExpressionFixture, FunctionWithNonConstantNonMonotonicChildrenIsNonMonotonic) { + ASSERT_FALSE(isMonotonicExpression(BSON("$dateTrunc" << BSON("date" + << "$time" + << "unit" + << "hour" + << "binSize" + << "$time")), + "time")); +} + +} // namespace mongo |