summaryrefslogtreecommitdiff
path: root/jstests/core/geo_group.js
blob: b19c35ec738d22d70cd320c07b5faaa7958f5506 (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
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 } } ]
);