83 lines
2.5 KiB
JavaScript
83 lines
2.5 KiB
JavaScript
// $cond returns the evaluated second argument if the first evaluates to true but the evaluated
|
|
// third argument if the first evaluates to false.
|
|
load('jstests/aggregation/extras/utils.js');
|
|
|
|
t = db.jstests_aggregation_cond;
|
|
t.drop();
|
|
|
|
t.save({});
|
|
|
|
function assertError(expectedErrorCode, condSpec) {
|
|
assertErrorCode(t, {$project: {a: {$cond: condSpec}}}, expectedErrorCode);
|
|
}
|
|
|
|
function assertResult(expectedResult, arg) {
|
|
assert.eq(expectedResult, t.aggregate({$project: {a: {$cond: arg}}}).toArray()[0].a);
|
|
}
|
|
|
|
// Wrong number of args.
|
|
assertError(16020, []);
|
|
assertError(16020, [1]);
|
|
assertError(16020, [false]);
|
|
assertError(16020, [1, 1]);
|
|
assertError(16020, [1, 1, null, 1]);
|
|
assertError(16020, [1, 1, 1, undefined]);
|
|
|
|
// Bad object cases
|
|
assertError(17080, {"else": 1, then: 1});
|
|
assertError(17081, {"if": 1, "else": 1});
|
|
assertError(17082, {"if": 1, then: 1});
|
|
assertError(17083, {asdf: 1, then: 1});
|
|
|
|
// Literal expressions.
|
|
assertResult(1, [true, 1, 2]);
|
|
assertResult(2, [false, 1, 2]);
|
|
|
|
// Order independence for object case
|
|
assertResult(1, {"if": true, "then": 1, "else": 2});
|
|
assertResult(1, {"if": true, "else": 2, "then": 1});
|
|
assertResult(1, {"then": 1, "if": true, "else": 2});
|
|
assertResult(1, {"then": 1, "else": 2, "if": true});
|
|
assertResult(1, {"else": 2, "then": 1, "if": true});
|
|
assertResult(1, {"else": 2, "if": true, "then": 1});
|
|
|
|
// Computed expressions.
|
|
assertResult(1, [{$and: []}, {$add: [1]}, {$add: [1, 1]}]);
|
|
assertResult(2, [{$or: []}, {$add: [1]}, {$add: [1, 1]}]);
|
|
|
|
t.drop();
|
|
t.save({t: true, f: false, x: 'foo', y: 'bar'});
|
|
|
|
// Field path expressions.
|
|
assertResult('foo', ['$t', '$x', '$y']);
|
|
assertResult('bar', ['$f', '$x', '$y']);
|
|
|
|
t.drop();
|
|
t.save({});
|
|
|
|
// Coerce to bool.
|
|
assertResult('a', [1, 'a', 'b']);
|
|
assertResult('a', ['', 'a', 'b']);
|
|
assertResult('b', [0, 'a', 'b']);
|
|
|
|
// Nested.
|
|
t.drop();
|
|
t.save({noonSense: 'am', mealCombined: 'no'});
|
|
t.save({noonSense: 'am', mealCombined: 'yes'});
|
|
t.save({noonSense: 'pm', mealCombined: 'yes'});
|
|
t.save({noonSense: 'pm', mealCombined: 'no'});
|
|
assert.eq(['breakfast', 'brunch', 'linner', 'dinner'],
|
|
t.aggregate({
|
|
$project: {
|
|
a: {
|
|
$cond: [
|
|
{$eq: ['$noonSense', 'am']},
|
|
{$cond: [{$eq: ['$mealCombined', 'yes']}, 'brunch', 'breakfast']},
|
|
{$cond: [{$eq: ['$mealCombined', 'yes']}, 'linner', 'dinner']}
|
|
]
|
|
}
|
|
}
|
|
}).map(function(x) {
|
|
return x.a;
|
|
}));
|