diff options
Diffstat (limited to 'src/mongo/scripting/bson_template_evaluator_test.cpp')
| -rw-r--r-- | src/mongo/scripting/bson_template_evaluator_test.cpp | 360 |
1 files changed, 360 insertions, 0 deletions
diff --git a/src/mongo/scripting/bson_template_evaluator_test.cpp b/src/mongo/scripting/bson_template_evaluator_test.cpp new file mode 100644 index 00000000000..5b334e0404f --- /dev/null +++ b/src/mongo/scripting/bson_template_evaluator_test.cpp @@ -0,0 +1,360 @@ +/** @file mongo/scripting/bson_template_evaluator_tests : tests related to bson template language */ + +/** + * Copyright (C) 2012 10gen Inc. + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU Affero General Public License, version 3, + * as published by the Free Software Foundation. + * + * 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 + * GNU Affero General Public License for more details. + * + * You should have received a copy of the GNU Affero General Public License + * along with this program. If not, see <http://www.gnu.org/licenses/>. + */ + +#include "mongo/db/jsobj.h" +#include "mongo/scripting/bson_template_evaluator.h" +#include "mongo/unittest/unittest.h" + +namespace mongo { + + namespace { + + TEST(BSONTemplateEvaluatorTest, RAND_INT) { + + BsonTemplateEvaluator *t = new BsonTemplateEvaluator(); + int randValue1, randValue2; + + // Test failure when the arguments to RAND_INT are not integers + BSONObjBuilder builder1; + BSONObj randObj = BSON( "#RAND_INT" << BSON_ARRAY("hello" << "world") ); + ASSERT_EQUALS( BsonTemplateEvaluator::StatusOpEvaluationError, + t->evaluate(BSON("randField" << randObj), builder1) ); + + // Test failure when operator does not exists + BSONObjBuilder builder2; + randObj = BSON( "#RAND_OP_NOT_EXISTS" << BSON_ARRAY( 5 << 0 ) ); + ASSERT_EQUALS( BsonTemplateEvaluator::StatusBadOperator, + t->evaluate(BSON("randField" << randObj), builder2) ); + + // Test failure when arguments to RAND_INT are not correct (max < min) + BSONObjBuilder builder3; + randObj = BSON( "#RAND_INT" << BSON_ARRAY( 5 << 0 ) ); + ASSERT_EQUALS( BsonTemplateEvaluator::StatusOpEvaluationError, + t->evaluate(BSON("randField" << randObj), builder3) ); + + // Test failure when operators are arbitrarily nested + // {id: { #RAND_INT: [ { #RAND_INT: [10, 20] }, 20] } + BSONObjBuilder builder4; + BSONObj innerRandObj = BSON( "#RAND_INT" << BSON_ARRAY( 0 << 5 ) ); + BSONObj outerRandObj = BSON( "#RAND_INT" << BSON_ARRAY( innerRandObj << 10 ) ); + ASSERT_EQUALS( BsonTemplateEvaluator::StatusOpEvaluationError, + t->evaluate(BSON("randField" << outerRandObj), builder4) ); + + // Test success with a single element + BSONObjBuilder builder5; + randObj = BSON( "#RAND_INT" << BSON_ARRAY( 0 << 5 ) ); + ASSERT_EQUALS( BsonTemplateEvaluator::StatusSuccess, + t->evaluate(BSON("randField" << randObj), builder5) ); + BSONObj obj5 = builder5.obj(); + ASSERT_EQUALS(obj5.nFields(), 1); + ASSERT_GREATER_THAN_OR_EQUALS(obj5.firstElement().numberInt(), 0); + ASSERT_LESS_THAN(obj5.firstElement().numberInt(), 5); + + // Test success with two #RAND_INT elements + BSONObjBuilder builder6; + randObj = BSON( "#RAND_INT" << BSON_ARRAY( 0 << 5 ) ); + ASSERT_EQUALS( BsonTemplateEvaluator::StatusSuccess, + t->evaluate(BSON("randField1" << randObj << + "randField2" << randObj), builder6) ); + BSONObj obj6 = builder6.obj(); + ASSERT_EQUALS(obj6.nFields(), 2); + BSONObjIterator iter6(obj6); + randValue1 = iter6.next().numberInt(); + randValue2 = iter6.next().numberInt(); + ASSERT_GREATER_THAN_OR_EQUALS(randValue1, 0); + ASSERT_LESS_THAN(randValue1, 5); + ASSERT_GREATER_THAN_OR_EQUALS(randValue2, 0); + ASSERT_LESS_THAN(randValue2, 5); + + // Test success with #RAND_INT as the last element + BSONObjBuilder builder7; + randObj = BSON( "#RAND_INT" << BSON_ARRAY( 0 << 5 ) ); + ASSERT_EQUALS( BsonTemplateEvaluator::StatusSuccess, + t->evaluate(BSON("id" << 1 << "hello" << "world" << + "randField" << randObj), builder7) ); + BSONObj obj7 = builder7.obj(); + ASSERT_EQUALS(obj7.nFields(), 3); + BSONObjIterator iter7(obj7); + iter7++; + randValue1 = iter7.next().numberInt(); + ASSERT_GREATER_THAN_OR_EQUALS(randValue1, 0); + ASSERT_LESS_THAN(randValue1, 5); + + // Test success with #RAND_INT as first element + BSONObjBuilder builder8; + randObj = BSON( "#RAND_INT" << BSON_ARRAY( 0 << 5 ) ); + ASSERT_EQUALS( BsonTemplateEvaluator::StatusSuccess, + t->evaluate(BSON("randField" << randObj << "hello" << "world" << + "id" << 1), builder8) ); + BSONObj obj8 = builder8.obj(); + ASSERT_EQUALS(obj8.nFields(), 3); + ASSERT_GREATER_THAN_OR_EQUALS(obj8.firstElement().numberInt(), 0); + ASSERT_LESS_THAN(obj8.firstElement().numberInt(), 5); + + // Test success with #RAND_INT as the middle element + BSONObjBuilder builder9; + randObj = BSON( "#RAND_INT" << BSON_ARRAY( 0 << 5 ) ); + ASSERT_EQUALS( BsonTemplateEvaluator::StatusSuccess, + t->evaluate(BSON("id" << 1 << "randField" << randObj << "hello" << + "world"), builder9) ); + BSONObj obj9 = builder9.obj(); + ASSERT_EQUALS(obj9.nFields(), 3); + BSONObjIterator iter9(obj9); + randValue1 = iter9.next().numberInt(); + ASSERT_GREATER_THAN_OR_EQUALS(randValue1, 0); + ASSERT_LESS_THAN(randValue1, 5); + + // Test success with #RAND_INT as the first and the last element + BSONObjBuilder builder10; + randObj = BSON( "#RAND_INT" << BSON_ARRAY( 0 << 5 ) ); + ASSERT_EQUALS( BsonTemplateEvaluator::StatusSuccess, + t->evaluate(BSON("randField1" << randObj << "hello" << + "world" << "randField2" << randObj), builder10) ); + BSONObj obj10 = builder10.obj(); + ASSERT_EQUALS(obj10.nFields(), 3); + + BSONObjIterator iter10(obj10); + randValue1 = (*iter10).numberInt(); + ASSERT_GREATER_THAN_OR_EQUALS(randValue1, 0); + ASSERT_LESS_THAN(randValue1, 5); + iter10++; + randValue2 = iter10.next().numberInt(); + ASSERT_GREATER_THAN_OR_EQUALS(randValue2, 0); + ASSERT_LESS_THAN(randValue2, 5); + + // Test success when one of the element is an array + BSONObjBuilder builder11; + randObj = BSON( "#RAND_INT" << BSON_ARRAY( 0 << 5 ) ); + ASSERT_EQUALS( BsonTemplateEvaluator::StatusSuccess, + t->evaluate(BSON("testArray" << BSON_ARRAY( 0 << 5 << 10 << 20 ) << + "hello" << "world" << + "randField" << randObj), builder11) ); + BSONObj obj11 = builder11.obj(); + ASSERT_EQUALS(obj11.nFields(), 3); + BSONObjIterator iter11(obj11); + iter11++; + randValue1 = iter11.next().numberInt(); + ASSERT_GREATER_THAN_OR_EQUALS(randValue1, 0); + ASSERT_LESS_THAN(randValue1, 5); + + // Test success with a 3rd argument to #RAND_INT + BSONObjBuilder builder12; + randObj = BSON( "#RAND_INT" << BSON_ARRAY( 0 << 5 << 4 ) ); + ASSERT_EQUALS( BsonTemplateEvaluator::StatusSuccess, + t->evaluate(BSON("id" << randObj), builder12) ); + BSONObj obj12 = builder12.obj(); + ASSERT_EQUALS(obj12.nFields(), 1); + ASSERT_GREATER_THAN_OR_EQUALS(obj12.firstElement().numberInt(), 0); + ASSERT_LESS_THAN_OR_EQUALS(obj12.firstElement().numberInt(), 16); + } + + TEST(BSONTemplateEvaluatorTest, RAND_STRING) { + + BsonTemplateEvaluator *t = new BsonTemplateEvaluator(); + + // Test failure when the arguments to RAND_STRING is not an integer + BSONObjBuilder builder1; + BSONObj randObj = BSON( "#RAND_STRING" << BSON_ARRAY("hello") ); + ASSERT_EQUALS( BsonTemplateEvaluator::StatusOpEvaluationError, + t->evaluate(BSON("randField" << randObj), builder1) ); + + // Test failure when there is more than 1 argument to RAND_STRING + BSONObjBuilder builder2; + randObj = BSON( "#RAND_STRING" << BSON_ARRAY( 2 << 8 ) ); + ASSERT_EQUALS( BsonTemplateEvaluator::StatusOpEvaluationError, + t->evaluate(BSON("randField" << randObj), builder2) ); + + // Test failure when length argument to RAND_STRING is 0 + BSONObjBuilder builder3; + randObj = BSON( "#RAND_STRING" << BSON_ARRAY(0 ) ); + ASSERT_EQUALS( BsonTemplateEvaluator::StatusOpEvaluationError, + t->evaluate(BSON("randField" << randObj), builder3) ); + + // Test success with a single element + BSONObjBuilder builder4; + randObj = BSON( "#RAND_STRING" << BSON_ARRAY(5) ); + ASSERT_EQUALS( BsonTemplateEvaluator::StatusSuccess, + t->evaluate(BSON("randField" << randObj), builder4) ); + BSONObj obj4 = builder4.obj(); + ASSERT_EQUALS(obj4.nFields(), 1); + ASSERT_EQUALS(obj4.firstElement().str().length(), 5U); + + // Test success with two #RAND_STRING elements + BSONObjBuilder builder5; + randObj = BSON( "#RAND_STRING" << BSON_ARRAY(5) ); + ASSERT_EQUALS( BsonTemplateEvaluator::StatusSuccess, + t->evaluate(BSON("randField1" << randObj << + "randField2" << randObj), builder5) ); + BSONObj obj5 = builder5.obj(); + ASSERT_EQUALS(obj5.nFields(), 2); + BSONObjIterator iter5(obj5); + ASSERT_EQUALS(iter5.next().str().length(), 5U); + ASSERT_EQUALS(iter5.next().str().length(), 5U); + + // Test success with #RAND_STRING as the last element + BSONObjBuilder builder6; + randObj = BSON( "#RAND_STRING" << BSON_ARRAY(5) ); + ASSERT_EQUALS( BsonTemplateEvaluator::StatusSuccess, + t->evaluate(BSON("id" << 1 << "hello" << "world" << + "randField" << randObj), builder6) ); + BSONObj obj6 = builder6.obj(); + ASSERT_EQUALS(obj6.nFields(), 3); + BSONObjIterator iter6(obj6); + iter6++; + ASSERT_EQUALS(iter6.next().str().length(), 5U); + + // Test success with #RAND_STRING as first element + BSONObjBuilder builder7; + randObj = BSON( "#RAND_STRING" << BSON_ARRAY(5) ); + ASSERT_EQUALS( BsonTemplateEvaluator::StatusSuccess, + t->evaluate(BSON("randField" << randObj << "hello" << "world" << + "id" << 1), builder7) ); + BSONObj obj7 = builder7.obj(); + ASSERT_EQUALS(obj7.nFields(), 3); + ASSERT_EQUALS(obj7.firstElement().str().length(), 5U); + + // Test success with #RAND_STRING as the middle element + BSONObjBuilder builder8; + randObj = BSON( "#RAND_STRING" << BSON_ARRAY(5) ); + ASSERT_EQUALS( BsonTemplateEvaluator::StatusSuccess, + t->evaluate(BSON("id" << 1 << "randField" << randObj << "hello" << + "world"), builder8) ); + BSONObj obj8 = builder8.obj(); + ASSERT_EQUALS(obj8.nFields(), 3); + BSONObjIterator iter8(obj8); + iter8++; + ASSERT_EQUALS((*iter8).str().length(), 5U); + + // Test success with #RAND_INT as the first and the last element + BSONObjBuilder builder10; + randObj = BSON( "#RAND_STRING" << BSON_ARRAY(5) ); + ASSERT_EQUALS( BsonTemplateEvaluator::StatusSuccess, + t->evaluate(BSON("randField1" << randObj << "hello" << + "world" << "randField2" << randObj), builder10) ); + BSONObj obj10 = builder10.obj(); + ASSERT_EQUALS(obj10.nFields(), 3); + + BSONObjIterator iter10(obj10); + ASSERT_EQUALS((*iter10).str().length(), 5U); + iter10++; + ASSERT_EQUALS(iter10.next().str().length(), 5U); + + // Test success when one of the element is an array + BSONObjBuilder builder11; + randObj = BSON( "#RAND_STRING" << BSON_ARRAY(5) ); + ASSERT_EQUALS( BsonTemplateEvaluator::StatusSuccess, + t->evaluate(BSON("testArray" << BSON_ARRAY( 0 << 5 << 10 << 20 ) << + "hello" << "world" << + "randField" << randObj), builder11) ); + BSONObj obj11 = builder11.obj(); + ASSERT_EQUALS(obj11.nFields(), 3); + BSONObjIterator iter11(obj11); + iter11++; + ASSERT_EQUALS(iter11.next().str().length(), 5U); + } + + TEST(BSONTemplateEvaluatorTest, CONCAT) { + + BsonTemplateEvaluator *t = new BsonTemplateEvaluator(); + + // Test failure when the arguments to #CONCAT has only one argument + BSONObjBuilder builder1; + BSONObj concatObj = BSON( "#CONCAT" << BSON_ARRAY("hello") ); + ASSERT_EQUALS( BsonTemplateEvaluator::StatusOpEvaluationError, + t->evaluate(BSON("concatField" << concatObj), builder1) ); + + // Test success when all arguments to #CONCAT are strings + BSONObjBuilder builder2; + concatObj = BSON( "#CONCAT" << BSON_ARRAY("hello" << " " << "world")); + ASSERT_EQUALS( BsonTemplateEvaluator::StatusSuccess, + t->evaluate(BSON("concatField" << concatObj), builder2) ); + BSONObj obj2 = builder2.obj(); + ASSERT_EQUALS(obj2.nFields(), 1); + BSONObj expectedObj = BSON("concatField" << "hello world"); + ASSERT_EQUALS(obj2.equal(expectedObj), true); + + // Test success when some arguments to #CONCAT are integers + BSONObjBuilder builder3; + concatObj = BSON( "#CONCAT" << BSON_ARRAY("F" << 1 << "racing")); + ASSERT_EQUALS( BsonTemplateEvaluator::StatusSuccess, + t->evaluate(BSON("concatField" << concatObj), builder3) ); + BSONObj obj3 = builder3.obj(); + ASSERT_EQUALS(obj3.nFields(), 1); + expectedObj = BSON("concatField" << "F1racing"); + ASSERT_EQUALS(obj3.equal(expectedObj), true); + + // Test success with #CONCAT as first element and last element + BSONObjBuilder builder4; + concatObj = BSON( "#CONCAT" << BSON_ARRAY("hello" << " " << "world")); + ASSERT_EQUALS( BsonTemplateEvaluator::StatusSuccess, + t->evaluate(BSON("concatField1" << concatObj << + "middleKey" << 1 << + "concatField2" << concatObj), builder4) ); + BSONObj obj4 = builder4.obj(); + ASSERT_EQUALS(obj4.nFields(), 3); + expectedObj = BSON("concatField1" << "hello world" << + "middleKey" << 1 << + "concatField2" << "hello world"); + ASSERT_EQUALS(obj4.equal(expectedObj), true); + + // Test success when one of the arguments to #CONCAT is an array + BSONObjBuilder builder5; + concatObj = BSON( "#CONCAT" << BSON_ARRAY("hello" << BSON_ARRAY(1 << 10) << "world")); + ASSERT_EQUALS( BsonTemplateEvaluator::StatusSuccess, + t->evaluate(BSON("concatField" << concatObj), builder5) ); + BSONObj obj5 = builder5.obj(); + ASSERT_EQUALS(obj5.nFields(), 1); + expectedObj = BSON("concatField" << "hello[ 1, 10 ]world"); + ASSERT_EQUALS(obj5.equal(expectedObj), true); + } + + TEST(BSONTemplateEvaluatorTest, COMBINED_OPERATORS) { + + BsonTemplateEvaluator *t = new BsonTemplateEvaluator(); + BSONObj randIntObj = BSON( "#RAND_INT" << BSON_ARRAY( 0 << 5 ) ); + BSONObj randStrObj = BSON( "#RAND_STRING" << BSON_ARRAY(5) ); + + // Test success when #RAND_INT, and #RAND_STRING are combined + BSONObjBuilder builder1; + ASSERT_EQUALS( BsonTemplateEvaluator::StatusSuccess, + t->evaluate(BSON("randInt" << randIntObj << + "randStr" << randStrObj), builder1) ); + BSONObj obj1 = builder1.obj(); + ASSERT_EQUALS(obj1.nFields(), 2); + BSONObjIterator iter1(obj1); + int randInt = iter1.next().numberInt(); + ASSERT_GREATER_THAN_OR_EQUALS(randInt, 0); + ASSERT_LESS_THAN(randInt, 5); + string randStr = iter1.next().str(); + ASSERT_EQUALS(randStr.length(), 5U); + + // Test success when the #CONCAT and #RAND_INT and #RAND_STRING are combined + BSONObjBuilder builder2; + BSONObj concatObj = BSON("#CONCAT" << BSON_ARRAY(randIntObj << " hello world " << + randStrObj)); + ASSERT_EQUALS( BsonTemplateEvaluator::StatusSuccess, + t->evaluate(BSON("concatField" << concatObj), builder2) ); + BSONObj obj2 = builder2.obj(); + ASSERT_EQUALS(obj2.nFields(), 1); + // check that the resulting string has a length of 19. + // randIntObj.length = 1, " hello world " = 13, randStrObj.length = 5 + // so total string length should 1 + 13 + 5 = 19 + ASSERT_EQUALS(obj2.firstElement().str().length(), 19U); + } + } // end anonymous namespace +} // end namespace mongo |