Add integration test

lambdaworks · Jun 24, 2023 · b901e6d · b901e6d
1 parent 62b3312
commit b901e6d
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 0 deletions.
diff --git a/modules/library/src/it/scala/zio/elasticsearch/HttpExecutorSpec.scala b/modules/library/src/it/scala/zio/elasticsearch/HttpExecutorSpec.scala
@@ -1706,6 +1706,41 @@ object HttpExecutorSpec extends IntegrationSpec {
             }
           } @@ after(Executor.execute(ElasticRequest.deleteIndex(geoDistanceIndex)).orDie)
         ),
+        suite("geo-polygon query")(
+          test("using geo-polygon query") {
+            checkOnce(genTestDocument) { document =>
+              val indexDefinition =
+                """
+                  |{
+                  |  "mappings": {
+                  |      "properties": {
+                  |        "locationField": {
+                  |          "type": "geo_point"
+                  |      }
+                  |    }
+                  |  }
+                  |}
+                  |""".stripMargin
+
+              for {
+                _ <- Executor.execute(ElasticRequest.createIndex(geoPolygonIndex, indexDefinition))
+                _ <- Executor.execute(ElasticRequest.deleteByQuery(geoPolygonIndex, matchAll))
+                _ <- Executor.execute(
+                       ElasticRequest.create[TestDocument](geoPolygonIndex, document).refreshTrue
+                     )
+                r1 <- Executor
+                        .execute(
+                          ElasticRequest.search(
+                            geoPolygonIndex,
+                            ElasticQuery
+                              .geoPolygon("locationField", List("0, 0", "0, 90", "90, 90", "90, 0"))
+                          )
+                        )
+                        .documentAs[TestDocument]
+              } yield assertTrue(r1 == Chunk(document))
+            }
+          } @@ after(Executor.execute(ElasticRequest.deleteIndex(geoPolygonIndex)).orDie)
+        ),
         suite("search for documents using FunctionScore query")(
           test("using randomScore function") {
             checkOnce(genTestDocument, genTestDocument) { (firstDocument, secondDocument) =>

diff --git a/modules/library/src/it/scala/zio/elasticsearch/IntegrationSpec.scala b/modules/library/src/it/scala/zio/elasticsearch/IntegrationSpec.scala
@@ -51,6 +51,8 @@ trait IntegrationSpec extends ZIOSpecDefault {
 
   val geoDistanceIndex: IndexName = IndexName("geo-distance-index")
 
+  val geoPolygonIndex: IndexName = IndexName("geo-polygon-index")
+
   val prepareElasticsearchIndexForTests: TestAspect[Nothing, Any, Throwable, Any] = beforeAll((for {
     _ <- Executor.execute(ElasticRequest.createIndex(index))
     _ <- Executor.execute(ElasticRequest.deleteByQuery(index, matchAll).refreshTrue)