summaryrefslogtreecommitdiff
path: root/jstests/core/geo3.js
blob: da3d86410493d213bfe9d447afdeb69a6d3f751a (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

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

n = 1;
arr = [];
for (var x = -100; x < 100; x += 2) {
    for (var y = -100; y < 100; y += 2) {
        arr.push({_id: n++, loc: [x, y], a: Math.abs(x) % 5, b: Math.abs(y) % 5});
    }
}
t.insert(arr);
assert.eq(t.count(), 100 * 100);
assert.eq(t.count(), n - 1);

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

fast = db.runCommand({geoNear: t.getName(), near: [50, 50], num: 10});

// test filter

filtered1 = db.runCommand({geoNear: t.getName(), near: [50, 50], num: 10, query: {a: 2}});
assert.eq(10, filtered1.results.length, "B1");
filtered1.results.forEach(function(z) {
    assert.eq(2, z.obj.a, "B2: " + tojson(z));
});
// printjson( filtered1.stats );

function avgA(q, len) {
    if (!len)
        len = 10;
    var realq = {
        loc: {$near: [50, 50]}
    };
    if (q)
        Object.extend(realq, q);
    var as = t.find(realq).limit(len).map(function(z) {
        return z.a;
    });
    assert.eq(len, as.length, "length in avgA");
    return Array.avg(as);
}

function testFiltering(msg) {
    assert.gt(2, avgA({}), msg + " testFiltering 1 ");
    assert.eq(2, avgA({a: 2}), msg + " testFiltering 2 ");
    assert.eq(4, avgA({a: 4}), msg + " testFiltering 3 ");
}

testFiltering("just loc");

t.dropIndex({loc: "2d"});
assert.eq(1, t.getIndexKeys().length, "setup 3a");
t.ensureIndex({loc: "2d", a: 1});
assert.eq(2, t.getIndexKeys().length, "setup 3b");

filtered2 = db.runCommand({geoNear: t.getName(), near: [50, 50], num: 10, query: {a: 2}});
assert.eq(10, filtered2.results.length, "B3");
filtered2.results.forEach(function(z) {
    assert.eq(2, z.obj.a, "B4: " + tojson(z));
});

assert.eq(filtered1.stats.avgDistance, filtered2.stats.avgDistance, "C1");
assert.gt(filtered1.stats.objectsLoaded, filtered2.stats.objectsLoaded, "C3");

testFiltering("loc and a");

t.dropIndex({loc: "2d", a: 1});
assert.eq(1, t.getIndexKeys().length, "setup 4a");
t.ensureIndex({loc: "2d", b: 1});
assert.eq(2, t.getIndexKeys().length, "setup 4b");

testFiltering("loc and b");

q = {
    loc: {$near: [50, 50]}
};
assert.eq(100, t.find(q).limit(100).itcount(), "D1");
assert.eq(100, t.find(q).limit(100).size(), "D2");

assert.eq(20, t.find(q).limit(20).itcount(), "D3");
assert.eq(20, t.find(q).limit(20).size(), "D4");

// SERVER-14039 Wrong limit after skip with $nearSphere, 2d index
assert.eq(10, t.find(q).skip(10).limit(10).itcount(), "D5");
assert.eq(10, t.find(q).skip(10).limit(10).size(), "D6");