summaryrefslogtreecommitdiff
path: root/jstests/core/projection.js
blob: 99addb31feeef1aa32bd11f14d63e7db23f69536 (plain)
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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
/**
 * Tests for FieldMatcher (later renamed to Projection).
 */

(function() {
'use strict';
const collNamePrefix = 'jstests_projection_';
let collCount = 0;

let res;

//
// Test cases originally in fm1.js.
//

let t = db.getCollection(collNamePrefix + collCount++);
t.drop();

let doc = {_id: 0, foo: {bar: 1}};
assert.commandWorked(t.insert(doc));

res = t.find({}, {foo: 1}).toArray();
assert.eq(1, res.length, tojson(res));
assert.docEq(doc, res[0], tojson(res));

res = t.find({}, {'foo.bar': 1}).toArray();
assert.eq(1, res.length, tojson(res));
assert.docEq(doc, res[0], tojson(res));

res = t.find({}, {'baz': 1}).toArray();
assert.eq(1, res.length, tojson(res));
assert.docEq({_id: 0}, res[0], tojson(res));

res = t.find({}, {'baz.qux': 1}).toArray();
assert.eq(1, res.length, tojson(res));
assert.docEq({_id: 0}, res[0], tojson(res));

res = t.find({}, {'foo.qux': 1}).toArray();
assert.eq(1, res.length, tojson(res));
assert.docEq({_id: 0, foo: {}}, res[0], tojson(res));

//
// Test cases originally in fm2.js
//

t = db.getCollection(collNamePrefix + collCount++);
t.drop();

doc = {
    _id: 1,
    one: {two: {three: 'four'}}
};
assert.commandWorked(t.insert(doc));

res = t.find({}, {'one.two': 1}).toArray();
assert.eq(1, res.length, tojson(res));
assert.docEq(doc, res[0], tojson(res));
assert.eq(1, Object.keySet(res[0].one).length, tojson(res));

//
// Test cases originally in fm3.js.
//

t = db.getCollection(collNamePrefix + collCount++);
t.drop();

assert.commandWorked(t.insert({a: [{c: {e: 1, f: 1}}, {d: 2}, 'z'], b: 1}));

res = t.findOne({}, {a: 1});
assert.eq(res.a, [{c: {e: 1, f: 1}}, {d: 2}, 'z'], "one a");
assert.eq(res.b, undefined, "one b");

res = t.findOne({}, {a: 0});
assert.eq(res.a, undefined, "two a");
assert.eq(res.b, 1, "two b");

res = t.findOne({}, {'a.d': 1});
assert.eq(res.a, [{}, {d: 2}], "three a");
assert.eq(res.b, undefined, "three b");

res = t.findOne({}, {'a.d': 0});
assert.eq(res.a, [{c: {e: 1, f: 1}}, {}, 'z'], "four a");
assert.eq(res.b, 1, "four b");

res = t.findOne({}, {'a.c': 1});
assert.eq(res.a, [{c: {e: 1, f: 1}}, {}], "five a");
assert.eq(res.b, undefined, "five b");

res = t.findOne({}, {'a.c': 0});
assert.eq(res.a, [{}, {d: 2}, 'z'], "six a");
assert.eq(res.b, 1, "six b");

res = t.findOne({}, {'a.c.e': 1});
assert.eq(res.a, [{c: {e: 1}}, {}], "seven a");
assert.eq(res.b, undefined, "seven b");

res = t.findOne({}, {'a.c.e': 0});
assert.eq(res.a, [{c: {f: 1}}, {d: 2}, 'z'], "eight a");
assert.eq(res.b, 1, "eight b");

//
// Test cases originally in fm4.js
//

t = db.getCollection(collNamePrefix + collCount++);
t.drop();

assert.commandWorked(t.insert({_id: 3, a: 1, b: 1}));

assert.docEq(t.findOne({}, {_id: 1}), {_id: 3}, "1");
assert.docEq(t.findOne({}, {_id: 0}), {a: 1, b: 1}, "2");

assert.docEq(t.findOne({}, {_id: 1, a: 1}), {_id: 3, a: 1}, "3");
assert.docEq(t.findOne({}, {_id: 0, a: 1}), {a: 1}, "4");

assert.docEq(t.findOne({}, {_id: 0, a: 0}), {b: 1}, "6");
assert.docEq(t.findOne({}, {a: 0}), {_id: 3, b: 1}, "5");

assert.docEq(t.findOne({}, {_id: 1, a: 0}), {_id: 3, b: 1}, "7");
})();