Isolate unnecessary tests and setting

Signed-off-by: Jialiang Liang <jiallian@amazon.com>

Author: RyanL1997

integ-test/build.gradle

@@ -794,4 +794,6 @@ task integTestRemote(type: RestIntegTestTask) {
     exclude 'org/opensearch/sql/legacy/QueryAnalysisIT.class'
     exclude 'org/opensearch/sql/legacy/OrderIT.class'
     exclude 'org/opensearch/sql/jdbc/**'
+    // Exclude AsyncQueryIT for AOSS - uses unsupported APIs like _nodes/plugins, _cat/indices
+    exclude 'org/opensearch/sql/asyncquery/AsyncQueryIT.class'
 }

integ-test/src/test/java/org/opensearch/sql/legacy/OpenSearchSQLRestTestCase.java

@@ -102,6 +102,12 @@ protected boolean isHttps() {
     return isHttps;
   }
 
+  /** Check if we're running against Amazon OpenSearch Serverless (AOSS) */
+  protected boolean isAossCluster() {
+    String clusterUrl = System.getProperty("tests.rest.cluster");
+    return clusterUrl != null && clusterUrl.contains(".aoss.amazonaws.com");
+  }
+
   protected String getProtocol() {
     return isHttps() ? "https" : "http";
   }
@@ -396,6 +402,11 @@ public void configureMultiClusters(String remote) throws IOException {
    * @throws IOException if there is an error retrieving cluster settings or updating them
    */
   protected void increaseMaxCompilationsRate() throws IOException {
+    // Skip script compilation rate configuration for AOSS
+    if (isAossCluster()) {
+      return; // AOSS doesn't support cluster settings or script context APIs
+    }
+
     // When script.disable_max_compilations_rate is set, custom context compilation rates cannot be
     // set
     if (!Objects.equals(
@@ -447,6 +458,15 @@ protected static void updateClusterSetting(String settingKey, Object value, bool
   }
 
   protected static JSONObject getAllClusterSettings() throws IOException {
+    // Skip cluster settings retrieval for AOSS as it doesn't support this API
+    String clusterUrl = System.getProperty("tests.rest.cluster");
+    boolean isAoss = clusterUrl != null && clusterUrl.contains(".aoss.amazonaws.com");
+
+    if (isAoss) {
+      // Return a mock settings object with default values for AOSS
+      return new JSONObject("{\"persistent\":{},\"transient\":{},\"defaults\":{}}");
+    }
+
     Request request = new Request("GET", "/_cluster/settings?flat_settings&include_defaults");
     RequestOptions.Builder restOptionsBuilder = RequestOptions.DEFAULT.toBuilder();
     restOptionsBuilder.addHeader("Content-Type", "application/json");
@@ -455,6 +475,24 @@ protected static JSONObject getAllClusterSettings() throws IOException {
   }
 
   protected static String getClusterSetting(String settingPath, String type) throws IOException {
+    // For AOSS, return appropriate default values for commonly checked settings
+    String clusterUrl = System.getProperty("tests.rest.cluster");
+    boolean isAoss = clusterUrl != null && clusterUrl.contains(".aoss.amazonaws.com");
+
+    if (isAoss) {
+      // Return sensible defaults for AOSS
+      if (settingPath.contains("calcite.engine.enabled")) {
+        return "false"; // Assume calcite is disabled for AOSS by default
+      }
+      if (settingPath.contains("calcite.pushdown.enabled")) {
+        return "true"; // Assume pushdown is enabled for AOSS by default
+      }
+      if (settingPath.contains("script.disable_max_compilations_rate")) {
+        return "true"; // Assume script rate limiting is disabled for AOSS
+      }
+      return ""; // Default empty value for other settings
+    }
+
     JSONObject settings = getAllClusterSettings();
     String value = settings.optJSONObject(type).optString(settingPath);
     if (StringUtils.isEmpty(value)) {

integ-test/src/test/java/org/opensearch/sql/legacy/SQLIntegTestCase.java

@@ -57,7 +57,8 @@ public abstract class SQLIntegTestCase extends OpenSearchSQLRestTestCase {
       Integer.parseInt(System.getProperty("defaultMaxResultWindow", "10000"));
 
   public boolean shouldResetQuerySizeLimit() {
-    return true;
+    // Skip cluster configuration for AOSS as it doesn't support /_cluster/settings
+    return !isAossCluster();
   }
 
   @Before
@@ -174,6 +175,14 @@ protected void setDataSourcesEnabled(String clusterSettingType, boolean value) {
   }
 
   protected static void wipeAllClusterSettings() throws IOException {
+    // Skip cluster settings cleanup for AOSS as it doesn't support /_cluster/settings
+    String clusterUrl = System.getProperty("tests.rest.cluster");
+    boolean isAoss = clusterUrl != null && clusterUrl.contains(".aoss.amazonaws.com");
+
+    if (isAoss) {
+      return; // AOSS doesn't support cluster settings modification
+    }
+
     updateClusterSettings(new ClusterSetting("persistent", "*", null));
     updateClusterSettings(new ClusterSetting("transient", "*", null));
     if (remoteClient() != null) {
@@ -392,6 +401,14 @@ protected JSONObject executeCursorCloseQuery(final String cursor) throws IOExcep
 
   protected static JSONObject updateClusterSettings(ClusterSetting setting, RestClient client)
       throws IOException {
+    // Skip cluster settings updates for AOSS as it doesn't support /_cluster/settings
+    String clusterUrl = System.getProperty("tests.rest.cluster");
+    boolean isAoss = clusterUrl != null && clusterUrl.contains(".aoss.amazonaws.com");
+
+    if (isAoss) {
+      return new JSONObject("{}"); // Return empty JSON for AOSS
+    }
+
     Request request = new Request("PUT", "/_cluster/settings");
     String persistentSetting =
         String.format(

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

@@ -139,6 +139,14 @@ protected Request buildRequest(String query, String endpoint) {
   }
 
   protected static JSONObject updateClusterSettings(ClusterSetting setting) throws IOException {
+    // Skip cluster settings updates for AOSS as it doesn't support /_cluster/settings
+    String clusterUrl = System.getProperty("tests.rest.cluster");
+    boolean isAoss = clusterUrl != null && clusterUrl.contains(".aoss.amazonaws.com");
+
+    if (isAoss) {
+      return new JSONObject("{}"); // Return empty JSON for AOSS
+    }
+
     Request request = new Request("PUT", "/_cluster/settings");
     String persistentSetting =
         String.format(
@@ -185,9 +193,9 @@ protected static boolean isCalciteEnabled() throws IOException {
   }
 
   public static void enableCalcite() throws IOException {
-    updateClusterSettings(
-        new SQLIntegTestCase.ClusterSetting(
-            "persistent", Settings.Key.CALCITE_ENGINE_ENABLED.getKeyValue(), "true"));
+    //    updateClusterSettings(
+    //        new SQLIntegTestCase.ClusterSetting(
+    //            "persistent", Settings.Key.CALCITE_ENGINE_ENABLED.getKeyValue(), "true"));
   }
 
   public static void disableCalcite() throws IOException {