diff options
Diffstat (limited to 'jstests/core/index/geo/geo_s2within_line_polygon_sphere.js')
| -rw-r--r-- | jstests/core/index/geo/geo_s2within_line_polygon_sphere.js | 241 |
1 files changed, 241 insertions, 0 deletions
diff --git a/jstests/core/index/geo/geo_s2within_line_polygon_sphere.js b/jstests/core/index/geo/geo_s2within_line_polygon_sphere.js new file mode 100644 index 00000000000..8cafa602ec4 --- /dev/null +++ b/jstests/core/index/geo/geo_s2within_line_polygon_sphere.js @@ -0,0 +1,241 @@ +// Tests for $geowithin $centerSphere operator with LineString and Polygon. +(function() { + +function testGeoWithinCenterSphereLinePolygon(coll) { + coll.drop(); + + // Convenient test function for $geowithin $centerSphere. + function testGeoWithinCenterSphere(centerSphere, expected) { + let result = coll.find({geoField: {$geoWithin: {$centerSphere: centerSphere}}}, + {"name": 1, "_id": 0}) + .sort({"name": 1}) + .toArray(); + assert.eq(result, expected); + } + + // Basic tests. + assert.commandWorked( + coll.insert({name: "Point1", geoField: {type: "Point", coordinates: [1, 1]}})); + assert.commandWorked(coll.insert( + {name: "LineString1", geoField: {type: "LineString", coordinates: [[1, 1], [2, 2]]}})); + assert.commandWorked(coll.insert({ + name: "Polygon1", + geoField: {type: "Polygon", coordinates: [[[1, 1], [2, 2], [2, 1], [1, 1]]]} + })); + + // The second parameter of $centerSphere is in radian and the angle between [1, 1] and [2,2] + // is about 0.0246 radian, much less than 1. + testGeoWithinCenterSphere([[1, 1], 1], + [{name: 'LineString1'}, {name: 'Point1'}, {name: 'Polygon1'}]); + + let geoDoc = { + "name": "LineString2", + "geoField": { + "type": "LineString", + "coordinates": + [[151.0997772216797, -33.86157820443923], [151.21719360351562, -33.8952122494965]] + } + }; + assert.commandWorked(coll.insert(geoDoc)); + + // Test for a LineString within a geowithin sphere. + testGeoWithinCenterSphere([[151.16789425018004, -33.8508357122312], 0.0011167360027064348], + [{name: "LineString2"}]); + + // Test for a LineString intersecting with geowithin sphere (should not return a match). + testGeoWithinCenterSphere([[151.09822404831158, -33.85109290503663], 0.0013568277575574095], + []); + + geoDoc = { + "name": "LineString3", + "geoField": { + "type": "LineString", + "coordinates": [ + [174.72896575927734, -36.86698689106876], + [174.72965240478516, -36.90707799098374], + [174.7808074951172, -36.9062544131224], + [174.77840423583982, -36.88154294352893], + [174.72827911376953, -36.88373984256185] + ] + } + }; + assert.commandWorked(coll.insert(geoDoc)); + + // Test for a LineString forming a closed loop rectangle within a geowithin sphere. + testGeoWithinCenterSphere([[174.75211152791763, -36.88962755605813], 0.000550933650273084], + [{name: "LineString3"}]); + + // Test for a LineString intersecting with geowithin sphere (should not return a match). + testGeoWithinCenterSphere([[174.75689891704758, -36.8998373317427], 0.0005315628331256537], []); + + // Test for a LineString outside of geowithin sphere (should not return a match). + testGeoWithinCenterSphere([[174.8099591465865, -36.89409450096385], 0.00027296698925637807], + []); + + // Test for a Polygon within a geowithin sphere. + geoDoc = { + "name": "Polygon2", + "city": "Wellington", + "geoField": { + "type": "Polygon", + "coordinates": [[ + [174.72930908203125, -41.281676559981676], + [174.76261138916013, -41.34820622928743], + [174.84329223632812, -41.32861539747227], + [174.8312759399414, -41.280902559820895], + [174.72930908203125, -41.281676559981676] + ]] + } + }; + assert.commandWorked(coll.insert(geoDoc)); + + // Test for a Polygon within a geowithin sphere. + testGeoWithinCenterSphere([[174.78536621904806, -41.30510816038769], 0.0009483659386360411], + [{name: "Polygon2"}]); + + // Test for an empty query cap (radius 0) inside of a polygon that covers the centerSphere + // (should not return a match). + testGeoWithinCenterSphere([[174.79144274337722, -41.307682001033385], 0], []); + + // Test for a Polygon intersecting with geowithin sphere (should not return a match). + testGeoWithinCenterSphere([[174.7599527533759, -41.27137819591382], 0.0011247013153526434], []); + + // Test for a Polygon outside of geowithin sphere (should not return a match). + testGeoWithinCenterSphere([[174.80008799649448, -41.201484845543426], 0.0007748581633291528], + []); + + geoDoc = { + "name": "MultiPolygon1", + "city": "Sydney", + "geoField": { + "type": "MultiPolygon", + "coordinates": [ + [[ + [151.21032714843747, -33.85074408022877], + [151.23367309570312, -33.84333046657819], + [151.20929718017578, -33.81680727566872], + [151.1876678466797, -33.829927301798676], + [151.21032714843747, -33.85074408022877] + ]], + [[ + [151.20140075683594, -33.856446422184305], + [151.17565155029297, -33.88979749364442], + [151.2044906616211, -33.9151583833889], + [151.23058319091797, -33.87041555094182], + [151.20140075683594, -33.856446422184305] + ]] + ] + } + }; + + assert.commandWorked(coll.insert(geoDoc)); + + // Test for a MultiPolygon (two seperate polygons) within a geowithin sphere. + testGeoWithinCenterSphere([[151.20821632978107, -33.865139891361636], 0.000981007241416606], + [{name: "MultiPolygon1"}]); + + // Verify that only one of the polygons of a MultiPolygon in the $centerSphere does not + // match + testGeoWithinCenterSphere([[151.20438542915883, -33.89006380099829], 0.0006390286437185907], + []); + + geoDoc = { + "name": "MultiPolygon2", + "city": "Sydney", + "geoField": { + "type": "MultiPolygon", + "coordinates": [[ + [ + [151.203031539917, -33.87116383262648], + [151.20401859283447, -33.88270791866475], + [151.21891021728516, -33.88256540860479], + [151.2138032913208, -33.86817066653049], + [151.203031539917, -33.87116383262648] + ], + [ + [151.21041297912598, -33.86980979429744], + [151.20938301086426, -33.8767579211837], + [151.2121295928955, -33.87722110953139], + [151.21315956115723, -33.86995232565932], + [151.21041297912598, -33.86980979429744] + ] + ]] + } + }; + assert.commandWorked(coll.insert(geoDoc)); + + // Test for a MultiPolygon (with a hole) within a geowithin sphere. + testGeoWithinCenterSphere([[151.20936119647115, -33.875266834633265], 0.00020277354002627845], + [{name: "MultiPolygon2"}]); + + // Test for centerSphere as big as earth radius (should return all). + testGeoWithinCenterSphere([[151.20936119647115, -33.875266834633265], 3.14159265358979323846], [ + {name: "LineString1"}, + {name: "LineString2"}, + {name: "LineString3"}, + {name: "MultiPolygon1"}, + {name: "MultiPolygon2"}, + {name: "Point1"}, + {name: "Polygon1"}, + {name: "Polygon2"} + ]); + + // Test for a MultiPolygon with holes intersecting with geowithin sphere (should not return + // a match). + testGeoWithinCenterSphere([[151.21028000820485, -33.87067923462358], 0.00013138775245714733], + []); + + // Test for a MultiPolygon with holes with geowithin sphere inside the hole (should not + // return a match). + testGeoWithinCenterSphere([[151.21093787887645, -33.87533330567804], 0.000016565456776516003], + []); + + coll.drop(); + + // Test for a large query cap containing both of line vertices but not the line itself. + // (should not return a match). + geoDoc = { + "name": "HorizontalLongLine", + "geoField": { + "type": "LineString", + "coordinates": [[96.328125, 5.61598581915534], [153.984375, -6.315298538330033]] + } + }; + assert.commandWorked(coll.insert(geoDoc)); + + // Test for a large query cap containing both of line vertices but not the line itself. + // (should not return a match). + testGeoWithinCenterSphere([[-59.80246852929814, -2.3633072488322853], 2.768403272464979], []); + + coll.drop(); + + // Test for a large query cap containing all polygon vertices but not the whole polygon. + // (should not return a match). + geoDoc = { + "name": "LargeRegion", + "geoField": { + "type": "Polygon", + "coordinates": [[ + [98.96484375, -11.350796722383672], + [135.35156249999997, -11.350796722383672], + [135.35156249999997, 0.8788717828324276], + [98.96484375, 0.8788717828324276], + [98.96484375, -11.350796722383672] + ]] + } + }; + assert.commandWorked(coll.insert(geoDoc)); + + // Test for a large query cap containing both of line vertices but not the line itself. + // (should not return a match). + testGeoWithinCenterSphere([[-61.52266094410311, 17.79937981451866], 2.9592242752161573], []); +} + +// Test $geowithin $centerSphere for LineString and Polygon without index. +let coll = db.geo_s2within_line_polygon_sphere; +testGeoWithinCenterSphereLinePolygon(coll); + +// Test $geowithin $centerSphere for LineString and Polygon with 2dsphere index. +assert.commandWorked(coll.createIndex({geoField: "2dsphere"})); +testGeoWithinCenterSphereLinePolygon(coll); +})(); |