Allow logging to be enabled for ensurePaAndRcaEnabled

opendistro-for-elasticsearch · Jul 8, 2020 · 93e8b28 · 93e8b28
1 parent a270904
commit 93e8b28
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/...t/java/com/amazon/opendistro/elasticsearch/performanceanalyzer/PerformanceAnalyzerIT.java b/...t/java/com/amazon/opendistro/elasticsearch/performanceanalyzer/PerformanceAnalyzerIT.java
@@ -81,15 +81,17 @@ public static void ensurePaAndRcaEnabled() throws Exception {
             Response resp = client().performRequest(new Request("GET", "_opendistro/_performanceanalyzer/cluster/config"));
             Map<String, Object> respMap = mapper.readValue(EntityUtils.toString(resp.getEntity(), "UTF-8"),
                     new TypeReference<Map<String, Object>>(){});
-            if (respMap.get("currentPerformanceAnalyzerClusterState").equals(3)) {
+            if (respMap.get("currentPerformanceAnalyzerClusterState").equals(3) &&
+                    !respMap.get("currentPerformanceAnalyzerClusterState").equals(7)) {
                 break;
             }
             Thread.sleep(1000L);
         }
         Response resp = client().performRequest(new Request("GET", "_opendistro/_performanceanalyzer/cluster/config"));
         Map<String, Object> respMap = mapper.readValue(EntityUtils.toString(resp.getEntity(), "UTF-8"),
                 new TypeReference<Map<String, Object>>(){});
-        if (!respMap.get("currentPerformanceAnalyzerClusterState").equals(3)) {
+        if (!respMap.get("currentPerformanceAnalyzerClusterState").equals(3) &&
+                !respMap.get("currentPerformanceAnalyzerClusterState").equals(7)) {
             throw new Exception("PA and RCA are not enabled on the target cluster!");
         }
     }