/* * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one * or more contributor license agreements. Licensed under the "Elastic License * 2.0", the "GNU Affero General Public License v3.0 only", and the "Server Side * Public License v 1"; you may not use this file except in compliance with, at * your election, the "Elastic License 2.0", the "GNU Affero General Public * License v3.0 only", or the "Server Side Public License, v 1". */ package org.elasticsearch.common.geo; import org.elasticsearch.ElasticsearchParseException; import org.elasticsearch.common.Strings; import org.elasticsearch.geometry.Geometry; import org.elasticsearch.geometry.GeometryCollection; import org.elasticsearch.geometry.Line; import org.elasticsearch.geometry.LinearRing; import org.elasticsearch.geometry.Point; import org.elasticsearch.geometry.Polygon; import org.elasticsearch.geometry.utils.StandardValidator; import org.elasticsearch.test.ESTestCase; import org.elasticsearch.xcontent.ToXContent; import org.elasticsearch.xcontent.XContentBuilder; import org.elasticsearch.xcontent.XContentFactory; import org.elasticsearch.xcontent.XContentParseException; import org.elasticsearch.xcontent.XContentParser; import java.util.List; import java.util.Map; /** * Tests for {@link GeometryParser} */ public class GeometryParserTests extends ESTestCase { public void testGeoJsonParsing() throws Exception { XContentBuilder pointGeoJson = XContentFactory.jsonBuilder() .startObject() .field("type", "Point") .startArray("coordinates") .value(100.0) .value(0.0) .endArray() .endObject(); try (XContentParser parser = createParser(pointGeoJson)) { parser.nextToken(); GeometryParserFormat format = GeometryParserFormat.geometryFormat(parser); assertEquals( new Point(100, 0), format.fromXContent(StandardValidator.instance(true), randomBoolean(), randomBoolean(), parser) ); XContentBuilder newGeoJson = XContentFactory.jsonBuilder(); format.toXContent(new Point(100, 10), newGeoJson, ToXContent.EMPTY_PARAMS); assertEquals("{\"type\":\"Point\",\"coordinates\":[100.0,10.0]}", Strings.toString(newGeoJson)); } XContentBuilder pointGeoJsonWithZ = XContentFactory.jsonBuilder() .startObject() .field("type", "Point") .startArray("coordinates") .value(100.0) .value(0.0) .value(10.0) .endArray() .endObject(); try (XContentParser parser = createParser(pointGeoJsonWithZ)) { parser.nextToken(); assertEquals(new Point(100, 0, 10.0), new GeometryParser(true, randomBoolean(), true).parse(parser)); } try (XContentParser parser = createParser(pointGeoJsonWithZ)) { parser.nextToken(); expectThrows(IllegalArgumentException.class, () -> new GeometryParser(true, randomBoolean(), false).parse(parser)); } XContentBuilder polygonGeoJson = XContentFactory.jsonBuilder() .startObject() .field("type", "Polygon") .startArray("coordinates") .startArray() .startArray() .value(100.0) .value(1.0) .endArray() .startArray() .value(101.0) .value(1.0) .endArray() .startArray() .value(101.0) .value(0.0) .endArray() .startArray() .value(100.0) .value(0.0) .endArray() .endArray() .endArray() .endObject(); Polygon p = new Polygon(new LinearRing(new double[] { 100d, 101d, 101d, 100d, 100d }, new double[] { 1d, 1d, 0d, 0d, 1d })); try (XContentParser parser = createParser(polygonGeoJson)) { parser.nextToken(); // Coerce should automatically close the polygon assertEquals(p, new GeometryParser(true, true, randomBoolean()).parse(parser)); } try (XContentParser parser = createParser(polygonGeoJson)) { parser.nextToken(); // No coerce - the polygon parsing should fail expectThrows(XContentParseException.class, () -> new GeometryParser(true, false, randomBoolean()).parse(parser)); } } public void testWKTParsing() throws Exception { XContentBuilder pointGeoJson = XContentFactory.jsonBuilder().startObject().field("foo", "Point (100 0)").endObject(); try (XContentParser parser = createParser(pointGeoJson)) { parser.nextToken(); // Start object parser.nextToken(); // Field Name parser.nextToken(); // Field Value GeometryParserFormat format = GeometryParserFormat.geometryFormat(parser); assertEquals( new Point(100, 0), format.fromXContent(StandardValidator.instance(true), randomBoolean(), randomBoolean(), parser) ); XContentBuilder newGeoJson = XContentFactory.jsonBuilder().startObject().field("val"); format.toXContent(new Point(100, 10), newGeoJson, ToXContent.EMPTY_PARAMS); newGeoJson.endObject(); assertEquals("{\"val\":\"POINT (100.0 10.0)\"}", Strings.toString(newGeoJson)); } // Make sure we can parse values outside the normal lat lon boundaries XContentBuilder lineGeoJson = XContentFactory.jsonBuilder().startObject().field("foo", "LINESTRING (100 0, 200 10)").endObject(); try (XContentParser parser = createParser(lineGeoJson)) { parser.nextToken(); // Start object parser.nextToken(); // Field Name parser.nextToken(); // Field Value assertEquals( new Line(new double[] { 100, 200 }, new double[] { 0, 10 }), new GeometryParser(true, randomBoolean(), randomBoolean()).parse(parser) ); } } public void testNullParsing() throws Exception { XContentBuilder pointGeoJson = XContentFactory.jsonBuilder().startObject().nullField("foo").endObject(); try (XContentParser parser = createParser(pointGeoJson)) { parser.nextToken(); // Start object parser.nextToken(); // Field Name parser.nextToken(); // Field Value GeometryParserFormat format = GeometryParserFormat.geometryFormat(parser); assertNull(format.fromXContent(StandardValidator.instance(true), randomBoolean(), randomBoolean(), parser)); XContentBuilder newGeoJson = XContentFactory.jsonBuilder().startObject().field("val"); // if we serialize non-null value - it should be serialized as geojson format.toXContent(new Point(100, 10), newGeoJson, ToXContent.EMPTY_PARAMS); newGeoJson.endObject(); assertEquals(""" {"val":{"type":"Point","coordinates":[100.0,10.0]}}""", Strings.toString(newGeoJson)); newGeoJson = XContentFactory.jsonBuilder().startObject().field("val"); format.toXContent(null, newGeoJson, ToXContent.EMPTY_PARAMS); newGeoJson.endObject(); assertEquals("{\"val\":null}", Strings.toString(newGeoJson)); } } public void testUnsupportedValueParsing() throws Exception { XContentBuilder pointGeoJson = XContentFactory.jsonBuilder().startObject().field("foo", 42).endObject(); try (XContentParser parser = createParser(pointGeoJson)) { parser.nextToken(); // Start object parser.nextToken(); // Field Name parser.nextToken(); // Field Value ElasticsearchParseException ex = expectThrows( ElasticsearchParseException.class, () -> new GeometryParser(true, randomBoolean(), randomBoolean()).parse(parser) ); assertEquals("shape must be an object consisting of type and coordinates", ex.getMessage()); } } public void testBasics() { GeometryParser parser = new GeometryParser(true, randomBoolean(), randomBoolean()); // point Point expectedPoint = new Point(-122.084110, 37.386637); testBasics(parser, Map.of("lat", 37.386637, "lon", -122.084110), expectedPoint); testBasics(parser, "37.386637, -122.084110", expectedPoint); testBasics(parser, "POINT (-122.084110 37.386637)", expectedPoint); testBasics(parser, Map.of("type", "Point", "coordinates", List.of(-122.084110, 37.386637)), expectedPoint); testBasics(parser, List.of(-122.084110, 37.386637), expectedPoint); // line Line expectedLine = new Line(new double[] { 0, 1 }, new double[] { 0, 1 }); testBasics(parser, "LINESTRING(0 0, 1 1)", expectedLine); testBasics(parser, Map.of("type", "LineString", "coordinates", List.of(List.of(0, 0), List.of(1, 1))), expectedLine); // polygon Polygon expectedPolygon = new Polygon(new LinearRing(new double[] { 0, 1, 1, 0, 0 }, new double[] { 0, 0, 1, 1, 0 })); testBasics(parser, "POLYGON((0 0, 1 0, 1 1, 0 1, 0 0))", expectedPolygon); testBasics( parser, Map.of( "type", "Polygon", "coordinates", List.of(List.of(List.of(0, 0), List.of(1, 0), List.of(1, 1), List.of(0, 1), List.of(0, 0))) ), expectedPolygon ); // geometry collection testBasics( parser, List.of( List.of(-122.084110, 37.386637), "37.386637, -122.084110", "POINT (-122.084110 37.386637)", Map.of("type", "Point", "coordinates", List.of(-122.084110, 37.386637)), Map.of("type", "LineString", "coordinates", List.of(List.of(0, 0), List.of(1, 1))), "POLYGON((0 0, 1 0, 1 1, 0 1, 0 0))" ), new GeometryCollection<>(List.of(expectedPoint, expectedPoint, expectedPoint, expectedPoint, expectedLine, expectedPolygon)) ); expectThrows(ElasticsearchParseException.class, () -> testBasics(parser, "not a geometry", null)); } private void testBasics(GeometryParser parser, Object value, Geometry expected) { Geometry geometry = parser.parseGeometry(value); assertEquals(expected, geometry); } }