summaryrefslogtreecommitdiff
path: root/jstests/core/geo_group.js
blob: a571131e11460cd1b53f773b93493372b0ceaa26 (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
// @tags: [
//     # Cannot implicitly shard accessed collections because group is not supported on sharded
//     # collections.
//     assumes_unsharded_collection,
//     requires_fastcount,
//
//     # reduce uses javascript
//     requires_scripting,
// ]

t = db.geo_group;
t.drop();

n = 1;
var bulk = t.initializeUnorderedBulkOp();
for (var x = -100; x < 100; x += 2) {
    for (var y = -100; y < 100; y += 2) {
        bulk.insert({_id: n++, loc: [x, y]});
    }
}
assert.writeOK(bulk.execute());

t.ensureIndex({loc: "2d"});

// Test basic count with $near
assert.eq(t.find().count(), 10000);
assert.eq(t.find({loc: {$within: {$center: [[56, 8], 10]}}}).count(), 81);
assert.eq(t.find({loc: {$near: [56, 8, 10]}}).count(), 81);

// Test basic group that effectively does a count
assert.eq(t.group({
    reduce: function(obj, prev) {
        prev.sums = {count: prev.sums.count + 1};
    },
    initial: {sums: {count: 0}}
}),
          [{"sums": {"count": 10000}}]);

// Test basic group + $near that does a count
assert.eq(t.group({
    reduce: function(obj, prev) {
        prev.sums = {count: prev.sums.count + 1};
    },
    initial: {sums: {count: 0}},
    cond: {loc: {$near: [56, 8, 10]}}
}),
          [{"sums": {"count": 81}}]);