63 lines
2.1 KiB
JavaScript
63 lines
2.1 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.
|
|
|
|
t = db.jstests_aggregation_cond;
|
|
t.drop();
|
|
|
|
t.save( {} );
|
|
|
|
function assertError( expectedErrorCode, condSpec ) {
|
|
assert.eq( expectedErrorCode,
|
|
t.aggregate( { $project:{ a:{ $cond:condSpec } } } ).code );
|
|
}
|
|
|
|
function assertResult( expectedResult, arg0, arg1, arg2 ) {
|
|
assert.eq( expectedResult,
|
|
t.aggregate( { $project:{ a:{ $cond:[ arg0, arg1, arg2 ] } } } ).result[ 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] );
|
|
|
|
// Literal expressions.
|
|
assertResult( 1, true, 1, 2 );
|
|
assertResult( 2, false, 1, 2 );
|
|
|
|
// 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' ] } ] } } } ).result
|
|
.map( function( x ) { return x.a; } ) );
|