Fix search alert IT failure (#329)

* Fix search alert IT failure Signed-off-by: zane-neo <[email protected]> * format code Signed-off-by: zane-neo <[email protected]> --------- Signed-off-by: zane-neo <[email protected]> (cherry picked from commit 51e58c5) Signed-off-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
opensearch-project · Jun 14, 2024 · f0c202b · f0c202b
1 parent 6021976
commit f0c202b
Showing 1 changed file with 15 additions and 5 deletions.
diff --git a/src/test/java/org/opensearch/integTest/BaseAgentToolsIT.java b/src/test/java/org/opensearch/integTest/BaseAgentToolsIT.java
@@ -8,6 +8,7 @@
 import java.io.IOException;
 import java.util.Collections;
 import java.util.List;
+import java.util.Locale;
 import java.util.Map;
 import java.util.Optional;
 import java.util.function.Predicate;
@@ -235,10 +236,13 @@ protected void deleteModel(String modelId) {
     }
 
     protected void createIndexWithConfiguration(String indexName, String indexConfiguration) throws Exception {
-        Response response = makeRequest(client(), "PUT", indexName, null, indexConfiguration, null);
-        Map<String, Object> responseInMap = parseResponseToMap(response);
-        assertEquals("true", responseInMap.get("acknowledged").toString());
-        assertEquals(indexName, responseInMap.get("index").toString());
+        boolean indexExists = indexExists(indexName);
+        if (!indexExists) {
+            Response response = makeRequest(client(), "PUT", indexName, null, indexConfiguration, null);
+            Map<String, Object> responseInMap = parseResponseToMap(response);
+            assertEquals("true", responseInMap.get("acknowledged").toString());
+            assertEquals(indexName, responseInMap.get("index").toString());
+        }
     }
 
     protected void createIngestPipelineWithConfiguration(String pipelineName, String body) throws Exception {
@@ -274,7 +278,13 @@ protected void deleteSystemIndices() throws IOException {
                 .collect(Collectors.toList());
 
             for (final String indexName : externalIndices) {
-                adminClient().performRequest(new Request("DELETE", "/" + indexName));
+                Response deleteResponse = adminClient().performRequest(new Request("DELETE", "/" + indexName));
+                Map<String, Object> responseInMap = parseResponseToMap(deleteResponse);
+                assertEquals(
+                    String.format(Locale.ROOT, "delete index %s failed with response: %s", indexName, gson.toJson(responseInMap)),
+                    "true",
+                    responseInMap.get("acknowledged").toString()
+                );
             }
         }
     }