File: geo_s2cursorlimitskip.js

package info (click to toggle)
mongodb 1%3A2.4.10-5
  • links: PTS, VCS
  • area: main
  • in suites: jessie-kfreebsd
  • size: 82,464 kB
  • sloc: cpp: 740,225; ansic: 152,098; sh: 13,820; python: 11,864; makefile: 1,012; perl: 922; pascal: 617; java: 452; lisp: 222; asm: 174
file content (69 lines) | stat: -rw-r--r-- 2,520 bytes parent folder | download | duplicates (3)
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
// Test various cursor behaviors
var t = db.geo_s2getmmm
t.drop();
t.ensureIndex({geo: "2dsphere"});

Random.setRandomSeed();
var random = Random.rand;

/*
 * To test that getmore is working within 2dsphere index.
 * We insert a bunch of points, get a cursor, and fetch some
 * of the points.  Then we insert a bunch more points, and 
 * finally fetch a bunch more.
 * If the final fetches work successfully, then getmore should
 * be working
 */
function sign() { return random() > 0.5 ? 1 : -1; }
function insertRandomPoints(num, minDist, maxDist){
    for(var i = 0; i < num; i++){
        var lat = sign() * (minDist + random() * (maxDist - minDist));
        var lng = sign() * (minDist + random() * (maxDist - minDist));
        var point = { geo: { type: "Point", coordinates: [lng, lat] } };
        t.insert(point);
        assert(!db.getLastError());
    }
}

var initialPointCount = 200
var smallBit = 10
var secondPointCount = 100

// Insert points between 0.01 and 1.0 away.
insertRandomPoints(initialPointCount, 0.01, 1.0);

var cursor = t.find({geo: {$geoNear : {$geometry: {type: "Point", coordinates: [0.0, 0.0]}}}}).batchSize(4);
assert.eq(cursor.count(), initialPointCount);

for(var j = 0; j < smallBit; j++){
    assert(cursor.hasNext());
    cursor.next();
}
// We looked at (initialPointCount - smallBit) points, should be more.
assert(cursor.hasNext())

// Insert points outside of the shell we've tested thus far
insertRandomPoints(secondPointCount, 2.01, 3.0);
assert.eq(cursor.count(), initialPointCount + secondPointCount)

for(var k = 0; k < initialPointCount + secondPointCount - smallBit; k++){
    assert(cursor.hasNext())
    var tmpPoint = cursor.next();
}
// Shouldn't be any more points to look at now.
assert(!cursor.hasNext())

var someLimit = 23;
// Make sure limit does something.
cursor = t.find({geo: {$geoNear : {$geometry: {type: "Point", coordinates: [0.0, 0.0]}}}}).limit(someLimit)
// Count doesn't work here -- ignores limit/skip, so we use itcount.
assert.eq(cursor.itcount(), someLimit)
// Make sure skip works by skipping some stuff ourselves.
var someSkip = 3;
cursor = t.find({geo: {$geoNear : {$geometry: {type: "Point", coordinates: [0.0, 0.0]}}}}).limit(someLimit + someSkip)
for (var i = 0; i < someSkip; ++i) { cursor.next(); }
var cursor2 = t.find({geo: {$geoNear : {$geometry: {type: "Point", coordinates: [0.0, 0.0]}}}}).skip(someSkip).limit(someLimit)
while (cursor.hasNext()) {
    assert(cursor2.hasNext());
    assert.eq(cursor.next(), cursor2.next());
}