CI: Geo-Spatial Plugin integration test (#3244)

Signed-off-by: Andy Kwok <[email protected]>
Signed-off-by: xinyual <[email protected]>

Author: andy-k-improving

build.gradle

@@ -6,7 +6,7 @@
 
 buildscript {
     ext {
-        opensearch_version = System.getProperty("opensearch.version", "2.19.3-SNAPSHOT")
+        opensearch_version = System.getProperty("opensearch.version", "2.19.0-SNAPSHOT")
         isSnapshot = "true" == System.getProperty("build.snapshot", "true")
         buildVersionQualifier = System.getProperty("build.version_qualifier", "")
         version_tokens = opensearch_version.tokenize('-')

integ-test/build.gradle

@@ -195,6 +195,9 @@ dependencies {
     testCompileOnly 'org.apiguardian:apiguardian-api:1.1.2'
 
     zipArchive group: 'org.opensearch.plugin', name:'opensearch-job-scheduler', version: "${opensearch_build}"
+
+    // For GeoIP PPL functions
+    zipArchive group: 'org.opensearch.plugin', name:'geospatial', version: "${opensearch_build}"
 }
 
 dependencyLicenses.enabled = false
@@ -233,16 +236,27 @@ def getJobSchedulerPlugin() {
     })
 }
 
+def getGeoSpatialPlugin() {
+    provider { (RegularFile) (() ->
+            configurations.zipArchive.asFileTree.matching {
+                include '**/geospatia*'
+            }.singleFile )
+    }
+}
+
+
 testClusters {
     integTest {
         testDistribution = 'archive'
         plugin(getJobSchedulerPlugin())
+        plugin(getGeoSpatialPlugin())
         plugin ":opensearch-sql-plugin"
         setting "plugins.query.datasources.encryption.masterkey", "1234567812345678"
     }
     remoteCluster {
         testDistribution = 'archive'
         plugin(getJobSchedulerPlugin())
+        plugin(getGeoSpatialPlugin())
         plugin ":opensearch-sql-plugin"
     }
     integTestWithSecurity {

integ-test/src/test/java/org/opensearch/sql/ppl/GeoIpFunctionsIT.java

@@ -0,0 +1,48 @@
+/*
+ *
+ * Copyright OpenSearch Contributors
+ * SPDX-License-Identifier: Apache-2.0
+ *
+ */
+
+package org.opensearch.sql.ppl;
+
+import static org.opensearch.sql.legacy.TestUtils.getResponseBody;
+
+import java.io.IOException;
+import lombok.SneakyThrows;
+import org.junit.Assert;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.opensearch.client.Request;
+import org.opensearch.client.RequestOptions;
+import org.opensearch.client.Response;
+
+/** IP enrichment PPL request with OpenSearch Geo-sptial plugin */
+public class GeoIpFunctionsIT extends PPLIntegTestCase {
+
+  private static boolean initialized = false;
+
+  private static String PLUGIN_NAME = "opensearch-geospatial";
+
+  @SneakyThrows
+  @BeforeEach
+  public void initialize() {
+    if (!initialized) {
+      setUpIndices();
+      initialized = true;
+    }
+  }
+
+  @Test
+  public void testGeoPluginInstallation() throws IOException {
+
+    Request request = new Request("GET", "/_cat/plugins?v");
+    RequestOptions.Builder restOptionsBuilder = RequestOptions.DEFAULT.toBuilder();
+    restOptionsBuilder.addHeader("Content-Type", "application/json");
+    request.setOptions(restOptionsBuilder);
+    Response response = client().performRequest(request);
+    Assert.assertEquals(200, response.getStatusLine().getStatusCode());
+    Assert.assertTrue(getResponseBody(response, true).contains(PLUGIN_NAME));
+  }
+}