From 5e2cd27b8af88c1628e166fe6e8c0b79acf65a65 Mon Sep 17 00:00:00 2001
From: "opensearch-trigger-bot[bot]"
 <98922864+opensearch-trigger-bot[bot]@users.noreply.github.com>
Date: Wed, 31 Jul 2024 11:10:38 -0400
Subject: [PATCH] Fix test
 RestStatusTests.testStatusReturnsFailureStatusWhenFailuresExist (#15011)
 (#15027)

This test has a reproducible failure when the highest "failure" status
is 100 level. This happens because RestStatus.status treats these as OK.


(cherry picked from commit f977f196c48570f3b16f305b39becae6838e7271)

Signed-off-by: Marc Handalian <marc.handalian@gmail.com>
Signed-off-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
---
 .../src/test/java/org/opensearch/core/RestStatusTests.java  | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/server/src/test/java/org/opensearch/core/RestStatusTests.java b/server/src/test/java/org/opensearch/core/RestStatusTests.java
index f8dba99aa8b60..fbd238bd035d0 100644
--- a/server/src/test/java/org/opensearch/core/RestStatusTests.java
+++ b/server/src/test/java/org/opensearch/core/RestStatusTests.java
@@ -55,7 +55,11 @@ public void testStatusReturnsFailureStatusWhenFailuresExist() {
             heapOfFailures.add(failure);
         }
 
-        assertEquals(heapOfFailures.peek().status(), RestStatus.status(successfulShards, totalShards, failures));
+        final RestStatus status = heapOfFailures.peek().status();
+        // RestStatus.status will return RestStatus.OK when the highest failure code is 100 level.
+        final RestStatus expected = status.getStatusFamilyCode() == 1 ? RestStatus.OK : status;
+
+        assertEquals(expected, RestStatus.status(successfulShards, totalShards, failures));
     }
 
     public void testSerialization() throws IOException {