diff options
Diffstat (limited to 'jstests/aggregation')
| -rw-r--r-- | jstests/aggregation/expressions/set.js | 193 |
1 files changed, 121 insertions, 72 deletions
diff --git a/jstests/aggregation/expressions/set.js b/jstests/aggregation/expressions/set.js index 4dbc19b32df..d90ddc20919 100644 --- a/jstests/aggregation/expressions/set.js +++ b/jstests/aggregation/expressions/set.js @@ -3,7 +3,8 @@ */ (function() { "use strict"; -load("jstests/aggregation/extras/utils.js"); +load("jstests/aggregation/extras/utils.js"); // For assertErrorCode. +load('jstests/libs/sbe_assert_error_override.js'); // Override error-code-checking APIs. const coll = db.expression_set; coll.drop(); @@ -19,84 +20,132 @@ assert.commandWorked(coll.insert([ {_id: 7, arr1: [1, 2, 3], arr2: [1, 1, 2, 2, 3, 3]}, ])); -const result = coll.aggregate([ - {$sort: {_id: 1}}, - { - $project: { - union: {$setUnion: ["$arr1", "$arr2"]}, - intersection: {$setIntersection: ["$arr1", "$arr2"]}, - difference: {$setDifference: ["$arr1", "$arr2"]}, - isSubset: {$setIsSubset: ["$arr1", "$arr2"]}, - equals: {$setEquals: ["$arr1", "$arr2"]}, - } - } - ]) - .toArray(); - // The order of the output array elements is undefined for $setUnion, $setDifference and // $setIntersection expressions. Hence we do a sort operation to get a consistent order. -const sortSetFields = document => Object.assign(document, { - union: document.union.sort(), - intersection: document.intersection.sort(), - difference: document.difference.sort(), -}); +const sortSetFields = doc => { + let result = {}; + for (const key in doc) { + if (doc.hasOwnProperty(key)) { + const value = doc[key]; + result[key] = Array.isArray(value) ? value.sort() : value; + } + } + return result; +}; + +const runTest = function(pipeline, expectedResult) { + pipeline.push({$sort: {_id: 1}}); + const result = coll.aggregate(pipeline).toArray(); + assert.eq(expectedResult, result.map(sortSetFields)); +}; + +runTest( + [{$project: {union: {$setUnion: ["$arr1", "$arr2"]}}}], + [ + {_id: 0, union: [1, 2, 3, 4]}, + {_id: 1, union: [1, 2, 3, 4, 5, 6]}, + {_id: 2, union: [1, 2, 3]}, + {_id: 3, union: [4, 5, 6]}, + {_id: 4, union: [1, 2, 3]}, + {_id: 5, union: [2, 3, 4]}, + {_id: 6, union: [1, 2, 3]}, + {_id: 7, union: [1, 2, 3]}, + ], +); + +runTest( + [{$project: {intersection: {$setIntersection: ["$arr1", "$arr2"]}}}], + [ + {_id: 0, intersection: [2, 3]}, + {_id: 1, intersection: []}, + {_id: 2, intersection: []}, + {_id: 3, intersection: []}, + {_id: 4, intersection: [2, 3]}, + {_id: 5, intersection: [2, 3]}, + {_id: 6, intersection: [1, 2, 3]}, + {_id: 7, intersection: [1, 2, 3]}, + ], +); -assert.eq(result.map(sortSetFields), [ - { - _id: 0, - union: [1, 2, 3, 4], - intersection: [2, 3], - difference: [1], - isSubset: false, - equals: false - }, - { - _id: 1, - union: [1, 2, 3, 4, 5, 6], - intersection: [], - difference: [1, 2, 3], - isSubset: false, - equals: false - }, - { - _id: 2, - union: [1, 2, 3], - intersection: [], - difference: [1, 2, 3], - isSubset: false, - equals: false - }, - {_id: 3, union: [4, 5, 6], intersection: [], difference: [], isSubset: true, equals: false}, - { - _id: 4, - union: [1, 2, 3], - intersection: [2, 3], - difference: [1], - isSubset: false, - equals: false - }, - {_id: 5, union: [2, 3, 4], intersection: [2, 3], difference: [], isSubset: true, equals: false}, - { - _id: 6, - union: [1, 2, 3], - intersection: [1, 2, 3], - difference: [], - isSubset: true, - equals: true - }, - { - _id: 7, - union: [1, 2, 3], - intersection: [1, 2, 3], - difference: [], - isSubset: true, - equals: true - }, -]); +runTest( + [{$project: {difference: {$setDifference: ["$arr1", "$arr2"]}}}], + [ + {_id: 0, difference: [1]}, + {_id: 1, difference: [1, 2, 3]}, + {_id: 2, difference: [1, 2, 3]}, + {_id: 3, difference: []}, + {_id: 4, difference: [1]}, + {_id: 5, difference: []}, + {_id: 6, difference: []}, + {_id: 7, difference: []}, + ], +); + +runTest( + [{$project: {difference: {$setDifference: ["$arr2", "$arr1"]}}}], + [ + {_id: 0, difference: [4]}, + {_id: 1, difference: [4, 5, 6]}, + {_id: 2, difference: []}, + {_id: 3, difference: [4, 5, 6]}, + {_id: 4, difference: []}, + {_id: 5, difference: [4]}, + {_id: 6, difference: []}, + {_id: 7, difference: []}, + ], +); + +runTest( + [{$project: {equals: {$setEquals: ["$arr1", "$arr2"]}}}], + [ + {_id: 0, equals: false}, + {_id: 1, equals: false}, + {_id: 2, equals: false}, + {_id: 3, equals: false}, + {_id: 4, equals: false}, + {_id: 5, equals: false}, + {_id: 6, equals: true}, + {_id: 7, equals: true}, + ], +); + +runTest( + [{$project: {isSubset: {$setIsSubset: ["$arr1", "$arr2"]}}}], + [ + {_id: 0, isSubset: false}, + {_id: 1, isSubset: false}, + {_id: 2, isSubset: false}, + {_id: 3, isSubset: true}, + {_id: 4, isSubset: false}, + {_id: 5, isSubset: true}, + {_id: 6, isSubset: true}, + {_id: 7, isSubset: true}, + ], +); // No sets to union should produce an empty set for all records so we only check the first one. assert.eq(coll.aggregate([{$project: {x: {$setUnion: []}}}]).toArray()[0]['x'], []); // No sets to intersect should produce an empty set for all records so we only check the first one. assert.eq(coll.aggregate([{$project: {x: {$setIntersection: []}}}]).toArray()[0]['x'], []); + +const operators = [ + ["$setUnion", 17043], + ["$setIntersection", 17047], + ["$setDifference", [17048, 17049]], + ["$setEquals", 17044], + ["$setIsSubset", [17042, 17046]] +]; +const badDocuments = [ + {arr1: "123", arr2: [1, 2, 3]}, + {arr1: [1, 2, 3], arr2: "123"}, + {arr1: "123", arr2: "123"}, +]; +for (const [operator, errorCodes] of operators) { + for (const badDocument of badDocuments) { + assert(coll.drop()); + assert.commandWorked(coll.insertOne(badDocument)); + assertErrorCode(coll, [{$project: {output: {[operator]: ["$arr1", "$arr2"]}}}], errorCodes); + } +} }()); |