add log to locate root cause of flaky test

Signed-off-by: zane-neo <[email protected]>
zane-neo · Aug 6, 2024 · fb2859f · fb2859f
1 parent 920685d
commit fb2859f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/plugin/src/test/java/org/opensearch/ml/rest/MLCommonsRestTestCase.java b/plugin/src/test/java/org/opensearch/ml/rest/MLCommonsRestTestCase.java
@@ -862,6 +862,7 @@ public static Map parseResponseToMap(Response response) throws IOException {
         HttpEntity entity = response.getEntity();
         assertNotNull(response);
         String entityString = TestHelper.httpEntityToString(entity);
+        assertEquals(String.format("response status is not success, raw response is: %s", entityString), 200, response.getStatusLine().getStatusCode());
         return StringUtils.gson.fromJson(entityString, Map.class);
     }