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
|
GeoNearRandomTest = function(name, dbToUse) {
this.name = name;
this.db = (dbToUse || db);
this.t = this.db[name];
this.nPts = 0;
// Reset state
this.t.drop();
Random.srand(1234);
print("Starting getNear test: " + name);
};
GeoNearRandomTest.prototype.mkPt = function mkPt(scale, indexBounds) {
if (!indexBounds) {
scale = scale || 1; // scale is good for staying away from edges
return [((Random.rand() * 359.8) - 179.9) * scale, ((Random.rand() * 180) - 90) * scale];
} else {
var range = indexBounds.max - indexBounds.min;
var eps = Math.pow(2, -40);
// Go very close to the borders but not quite there.
return [
(Random.rand() * (range - eps) + eps) + indexBounds.min,
(Random.rand() * (range - eps) + eps) + indexBounds.min
];
}
};
GeoNearRandomTest.prototype.insertPts = function(nPts, indexBounds, scale) {
assert.eq(this.nPts, 0, "insertPoints already called");
this.nPts = nPts;
var bulk = this.t.initializeUnorderedBulkOp();
for (var i = 0; i < nPts; i++) {
bulk.insert({_id: i, loc: this.mkPt(scale, indexBounds)});
}
assert.commandWorked(bulk.execute());
if (!indexBounds)
this.t.ensureIndex({loc: '2d'});
else
this.t.ensureIndex({loc: '2d'}, indexBounds);
};
GeoNearRandomTest.prototype.assertIsPrefix = function(short, long, errmsg) {
for (var i = 0; i < short.length; i++) {
var xS = short[i] ? short[i].loc[0] : short[i].loc[0];
var yS = short[i] ? short[i].loc[1] : short[i].loc[1];
var dS = short[i] ? short[i].dis : 1;
var xL = long[i] ? long[i].loc[0] : long[i].loc[0];
var yL = long[i] ? long[i].loc[1] : long[i].loc[1];
var dL = long[i] ? long[i].dis : 1;
assert.eq([xS, yS, dS], [xL, yL, dL], errmsg);
}
};
GeoNearRandomTest.prototype.testPt = function(pt, opts) {
assert.neq(this.nPts, 0, "insertPoints not yet called");
opts = opts || {};
opts['sphere'] = opts['sphere'] || 0;
opts['nToTest'] = opts['nToTest'] || this.nPts; // be careful, test is O( N^2 )
print("testing point: " + tojson(pt) + " opts: " + tojson(opts));
let query = {loc: {}};
query.loc[opts.sphere ? '$nearSphere' : '$near'] = pt;
const proj = {dis: {$meta: "geoNearDistance"}};
const runQuery = (limit) => this.t.find(query, proj).limit(opts.nToTest).toArray();
let last = runQuery(1);
for (var i = 2; i <= opts.nToTest; i++) {
let ret = runQuery(i);
this.assertIsPrefix(last, ret, `Unexpected result when comparing ${i - 1} and ${i}`);
// Make sure distances are in increasing order.
assert.gte(ret[ret.length - 1].dis, last[last.length - 1].dis);
last = ret;
}
// Test that a query using $near or $nearSphere returns the same points in order as the $geoNear
// aggregation stage.
const queryResults = runQuery(opts.nToTest);
const aggResults = this.t
.aggregate([
{$geoNear: {near: pt, distanceField: "dis", spherical: opts.sphere}},
{$limit: opts.nToTest}
])
.toArray();
assert.eq(queryResults, aggResults);
};
|
