diff options
Diffstat (limited to 'src/mongo/db/pipeline/document_source_add_fields_test.cpp')
| -rw-r--r-- | src/mongo/db/pipeline/document_source_add_fields_test.cpp | 137 |
1 files changed, 137 insertions, 0 deletions
diff --git a/src/mongo/db/pipeline/document_source_add_fields_test.cpp b/src/mongo/db/pipeline/document_source_add_fields_test.cpp new file mode 100644 index 00000000000..63896f31e27 --- /dev/null +++ b/src/mongo/db/pipeline/document_source_add_fields_test.cpp @@ -0,0 +1,137 @@ +/** + * Copyright (C) 2016 MongoDB, 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/>. + * + * 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 GNU Affero General 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/platform/basic.h" + +#include <vector> + +#include "mongo/db/pipeline/aggregation_context_fixture.h" +#include "mongo/db/pipeline/document.h" +#include "mongo/db/pipeline/document_source.h" +#include "mongo/db/pipeline/document_value_test_util.h" +#include "mongo/unittest/unittest.h" +#include "mongo/util/assert_util.h" + +namespace mongo { +namespace { + +using std::vector; + +// +// DocumentSourceAddFields delegates much of its responsibilities to the ParsedAddFields, which +// derives from ParsedAggregationProjection. +// Most of the functional tests are testing ParsedAddFields directly. These are meant as +// simpler integration tests. +// + +// This provides access to getExpCtx(), but we'll use a different name for this test suite. +using AddFieldsTest = AggregationContextFixture; + +TEST_F(AddFieldsTest, ShouldKeepUnspecifiedFieldsReplaceExistingFieldsAndAddNewFields) { + auto addFields = + DocumentSourceAddFields::create(BSON("e" << 2 << "b" << BSON("c" << 3)), getExpCtx()); + auto mock = + DocumentSourceMock::create({Document{{"a", 1}, {"b", Document{{"c", 1}}}, {"d", 1}}}); + addFields->setSource(mock.get()); + + auto next = addFields->getNext(); + ASSERT_TRUE(next.isAdvanced()); + Document expected = Document{{"a", 1}, {"b", Document{{"c", 3}}}, {"d", 1}, {"e", 2}}; + ASSERT_DOCUMENT_EQ(next.releaseDocument(), expected); + + ASSERT_TRUE(addFields->getNext().isEOF()); + ASSERT_TRUE(addFields->getNext().isEOF()); + ASSERT_TRUE(addFields->getNext().isEOF()); +} + +TEST_F(AddFieldsTest, ShouldOptimizeInnerExpressions) { + auto addFields = DocumentSourceAddFields::create( + BSON("a" << BSON("$and" << BSON_ARRAY(BSON("$const" << true)))), getExpCtx()); + addFields->optimize(); + // The $and should have been replaced with its only argument. + vector<Value> serializedArray; + addFields->serializeToArray(serializedArray); + ASSERT_BSONOBJ_EQ(serializedArray[0].getDocument().toBson(), + fromjson("{$addFields: {a: {$const: true}}}")); +} + +TEST_F(AddFieldsTest, ShouldErrorOnNonObjectSpec) { + BSONObj spec = BSON("$addFields" + << "foo"); + BSONElement specElement = spec.firstElement(); + ASSERT_THROWS_CODE( + DocumentSourceAddFields::createFromBson(specElement, getExpCtx()), UserException, 40272); +} + +TEST_F(AddFieldsTest, ShouldBeAbleToProcessMultipleDocuments) { + auto addFields = DocumentSourceAddFields::create(BSON("a" << 10), getExpCtx()); + auto mock = + DocumentSourceMock::create({Document{{"a", 1}, {"b", 2}}, Document{{"c", 3}, {"d", 4}}}); + addFields->setSource(mock.get()); + + auto next = addFields->getNext(); + ASSERT_TRUE(next.isAdvanced()); + Document expected = Document{{"a", 10}, {"b", 2}}; + ASSERT_DOCUMENT_EQ(next.releaseDocument(), expected); + + next = addFields->getNext(); + ASSERT_TRUE(next.isAdvanced()); + expected = Document{{"c", 3}, {"d", 4}, {"a", 10}}; + ASSERT_DOCUMENT_EQ(next.releaseDocument(), expected); + + ASSERT_TRUE(addFields->getNext().isEOF()); + ASSERT_TRUE(addFields->getNext().isEOF()); + ASSERT_TRUE(addFields->getNext().isEOF()); +} + +TEST_F(AddFieldsTest, ShouldAddReferencedFieldsToDependencies) { + auto addFields = DocumentSourceAddFields::create( + fromjson("{a: true, x: '$b', y: {$and: ['$c','$d']}, z: {$meta: 'textScore'}}"), + getExpCtx()); + DepsTracker dependencies(DepsTracker::MetadataAvailable::kTextScore); + ASSERT_EQUALS(DocumentSource::SEE_NEXT, addFields->getDependencies(&dependencies)); + ASSERT_EQUALS(3U, dependencies.fields.size()); + + // No implicit _id dependency. + ASSERT_EQUALS(0U, dependencies.fields.count("_id")); + + // Replaced field is not dependent. + ASSERT_EQUALS(0U, dependencies.fields.count("a")); + + // Field path expression dependency. + ASSERT_EQUALS(1U, dependencies.fields.count("b")); + + // Nested expression dependencies. + ASSERT_EQUALS(1U, dependencies.fields.count("c")); + ASSERT_EQUALS(1U, dependencies.fields.count("d")); + ASSERT_EQUALS(false, dependencies.needWholeDocument); + ASSERT_EQUALS(true, dependencies.getNeedTextScore()); +} + +} // namespace +} // namespace mongo |