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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
// $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; } ) );
|
