1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
// $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, 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; } ) );
|
