ADD Integration tests

elastic · Apr 25, 2022 · 6f11033 · 6f11033
1 parent e93bb3c
commit 6f11033
Showing 1 changed file with 87 additions and 0 deletions.
diff --git a/...ramework/src/main/java/org/elasticsearch/search/geo/GeoBoundingBoxQueryIntegTestCase.java b/...ramework/src/main/java/org/elasticsearch/search/geo/GeoBoundingBoxQueryIntegTestCase.java
@@ -120,6 +120,34 @@ public void testSimpleBoundingBoxTest() throws Exception {
         for (SearchHit hit : searchResponse.getHits()) {
             assertThat(hit.getId(), anyOf(equalTo("1"), equalTo("3"), equalTo("5")));
         }
+
+        searchResponse = client().prepareSearch() // top == bottom && left == right
+            .setQuery(geoBoundingBoxQuery("location").setCorners(40.7143528, -74.0059731, 40.7143528, -74.0059731))
+            .get();
+        assertThat(searchResponse.getHits().getTotalHits().value, equalTo(1L));
+        assertThat(searchResponse.getHits().getHits().length, equalTo(1));
+        for (SearchHit hit : searchResponse.getHits()) {
+            assertThat(hit.getId(), equalTo("1"));
+        }
+
+        searchResponse = client().prepareSearch() // top == bottom
+            .setQuery(geoBoundingBoxQuery("location").setCorners(40.759011, -74.00009, 40.759011, -73.0059731))
+            .get();
+        assertThat(searchResponse.getHits().getTotalHits().value, equalTo(1L));
+        assertThat(searchResponse.getHits().getHits().length, equalTo(1));
+        for (SearchHit hit : searchResponse.getHits()) {
+            assertThat(hit.getId(), equalTo("2"));
+        }
+
+        searchResponse = client().prepareSearch() // left == right
+            .setQuery(geoBoundingBoxQuery("location").setCorners(41.8, -73.9844722, 40.7, -73.9844722))
+            .get();
+        assertThat(searchResponse.getHits().getTotalHits().value, equalTo(1L));
+        assertThat(searchResponse.getHits().getHits().length, equalTo(1));
+        for (SearchHit hit : searchResponse.getHits()) {
+            assertThat(hit.getId(), equalTo("2"));
+        }
+
         // Distance query
         searchResponse = client().prepareSearch() // from NY
             .setQuery(geoDistanceQuery("location").point(40.5, -73.9).distance(25, DistanceUnit.KILOMETERS))
@@ -191,6 +219,54 @@ public void testLimit2BoundingBox() throws Exception {
             .get();
         assertThat(searchResponse.getHits().getTotalHits().value, equalTo(1L));
 
+        // top == bottom && left == right
+        searchResponse = client().prepareSearch()
+            .setQuery(
+                boolQuery().must(termQuery("userid", 880))
+                    .filter(geoBoundingBoxQuery("location").setCorners(18.036842, 59.328355000000002, 18.036842, 59.328355000000002))
+            )
+            .get();
+        assertThat(searchResponse.getHits().getTotalHits().value, equalTo(1L));
+        searchResponse = client().prepareSearch()
+            .setQuery(
+                boolQuery().must(termQuery("userid", 534))
+                    .filter(geoBoundingBoxQuery("location").setCorners(45.509526999999999, -73.570986000000005, 45.509526999999999, -73.570986000000005))
+            )
+            .get();
+        assertThat(searchResponse.getHits().getTotalHits().value, equalTo(1L));
+
+        // top == bottom
+        searchResponse = client().prepareSearch()
+            .setQuery(
+                boolQuery().must(termQuery("userid", 880))
+                    .filter(geoBoundingBoxQuery("location").setCorners(18.036842, 143.5, 18.036842, 113.96875))
+            )
+            .get();
+        assertThat(searchResponse.getHits().getTotalHits().value, equalTo(1L));
+        searchResponse = client().prepareSearch()
+            .setQuery(
+                boolQuery().must(termQuery("userid", 534))
+                    .filter(geoBoundingBoxQuery("location").setCorners(45.509526999999999, 143.5, 45.509526999999999, 113.96875))
+            )
+            .get();
+        assertThat(searchResponse.getHits().getTotalHits().value, equalTo(1L));
+
+        // left == right
+        searchResponse = client().prepareSearch()
+            .setQuery(
+                boolQuery().must(termQuery("userid", 880))
+                    .filter(geoBoundingBoxQuery("location").setCorners(74.579421999999994, 59.328355000000002, -66.668903999999998, 59.328355000000002))
+            )
+            .get();
+        assertThat(searchResponse.getHits().getTotalHits().value, equalTo(1L));
+        searchResponse = client().prepareSearch()
+            .setQuery(
+                boolQuery().must(termQuery("userid", 534))
+                    .filter(geoBoundingBoxQuery("location").setCorners(74.579421999999994, -73.570986000000005, -66.668903999999998, -73.570986000000005))
+            )
+            .get();
+        assertThat(searchResponse.getHits().getTotalHits().value, equalTo(1L));
+
         // Distance query
         searchResponse = client().prepareSearch()
             .setQuery(
@@ -273,6 +349,17 @@ public void testCompleteLonRange() throws Exception {
             .get();
         assertThat(searchResponse.getHits().getTotalHits().value, equalTo(2L));
 
+        // top == bottom
+        searchResponse = client().prepareSearch()
+            .setQuery(geoBoundingBoxQuery("location").setValidationMethod(GeoValidationMethod.COERCE).setCorners(59.328355000000002, 0, 59.328355000000002, 360))
+            .get();
+        assertThat(searchResponse.getHits().getTotalHits().value, equalTo(1L));
+        searchResponse = client().prepareSearch()
+            .setQuery(geoBoundingBoxQuery("location").setValidationMethod(GeoValidationMethod.COERCE).setCorners(59.328355000000002, -180, 59.328355000000002, 180))
+            .get();
+        assertThat(searchResponse.getHits().getTotalHits().value, equalTo(1L));
+
+
         // Distance query
         searchResponse = client().prepareSearch()
             .setQuery(geoDistanceQuery("location").point(60.0, -20.0).distance(1800, DistanceUnit.MILES))