summaryrefslogtreecommitdiff
path: root/jstests/core/geo_s2dupe_points.js
blob: 5b9a30e61c56c1836d9d7ff2235fb585dd82036a (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
// See: SERVER-9240, SERVER-9401.
// s2 rejects shapes with duplicate adjacent points as invalid, but they are
// valid in GeoJSON.  We store the duplicates, but internally remove them
// before indexing or querying.
t = db.geo_s2dupe_points;
t.drop();
t.ensureIndex({geo: "2dsphere"});

function testDuplicates(shapeName, shapeWithDupes, shapeWithoutDupes) {
    // insert a doc with dupes
    assert.writeOK(t.insert(shapeWithDupes));

    // duplicates are preserved when the document is fetched by _id
    assert.eq(shapeWithDupes, t.findOne({_id: shapeName}));
    assert.neq(shapeWithoutDupes, t.findOne({_id: shapeName}).geo);

    // can query with $geoIntersects inserted doc using both the duplicated and de-duplicated docs
    assert.eq(t.find({ geo: { $geoIntersects: { $geometry : shapeWithDupes.geo  } } } ).itcount(), 1);
    assert.eq(t.find({ geo: { $geoIntersects: { $geometry : shapeWithoutDupes  } } } ).itcount(), 1);

    // direct document equality in queries is preserved
    assert.eq(t.find({ geo: shapeWithoutDupes} ).itcount(), 0);
    assert.eq(t.find({ geo: shapeWithDupes.geo } ).itcount(), 1);
}

// LineString
var lineWithDupes = { _id: "line", geo: { type: "LineString",
        coordinates: [ [40,5], [40,5], [ 40, 5], [41, 6], [41,6] ]
    }
};
var lineWithoutDupes = { type: "LineString", coordinates: [ [40,5], [41,6] ] };

// Polygon
var polygonWithDupes = { _id: "poly", geo: { type: "Polygon",
    coordinates: [
        [ [-3.0, -3.0], [3.0, -3.0], [3.0, 3.0], [-3.0, 3.0], [-3.0, -3.0] ],
        [ [-2.0, -2.0], [2.0, -2.0], [2.0, 2.0], [-2.0, 2.0], [-2.0, -2.0], [-2.0, -2.0] ]
    ] }
};
var polygonWithoutDupes = { type: "Polygon",
    coordinates: [
        [ [-3.0, -3.0], [3.0, -3.0], [3.0, 3.0], [-3.0, 3.0], [-3.0, -3.0] ],
        [ [-2.0, -2.0], [2.0, -2.0], [2.0, 2.0], [-2.0, 2.0], [-2.0, -2.0] ]
    ]
};

// MultiPolygon
var multiPolygonWithDupes = { _id: "multi", geo: { type: "MultiPolygon", coordinates: [
        [
            [ [102.0, 2.0], [103.0, 2.0], [103.0, 2.0], [103.0, 3.0], [102.0, 3.0], [102.0, 2.0] ]
        ],
        [
            [ [100.0, 0.0], [101.0, 0.0], [101.0, 1.0], [101.0, 1.0], [100.0, 1.0], [100.0, 0.0] ],
            [ [100.2, 0.2], [100.8, 0.2], [100.8, 0.8], [100.8, 0.8], [100.8, 0.8], [100.2, 0.8], [100.2, 0.2] ]
        ]
    ]
} };
var multiPolygonWithoutDupes = { type: "MultiPolygon", coordinates: [
        [
            [ [102.0, 2.0], [103.0, 2.0], [103.0, 3.0], [102.0, 3.0], [102.0, 2.0] ]
        ],
        [
            [ [100.0, 0.0], [101.0, 0.0], [101.0, 1.0], [100.0, 1.0], [100.0, 0.0] ],
            [ [100.2, 0.2], [100.8, 0.2], [100.8, 0.8], [100.2, 0.8], [100.2, 0.2] ]
        ]
    ]
};

testDuplicates("line", lineWithDupes, lineWithoutDupes);
testDuplicates("poly", polygonWithDupes, polygonWithoutDupes);
testDuplicates("multi", multiPolygonWithDupes, multiPolygonWithoutDupes);