accept past tense error message in getModel IT

Signed-off-by: HenryL27 <[email protected]>
HenryL27 · Nov 29, 2023 · 9e0b80e · 9e0b80e
1 parent 87707ff
commit 9e0b80e
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/plugin/src/test/java/org/opensearch/ml/rest/RestMLGetModelActionIT.java b/plugin/src/test/java/org/opensearch/ml/rest/RestMLGetModelActionIT.java
@@ -22,9 +22,12 @@ public class RestMLGetModelActionIT extends MLCommonsRestTestCase {
     public ExpectedException exceptionRule = ExpectedException.none();
 
     public void testGetModelAPI_EmptyResources() throws IOException {
-        exceptionRule.expect(ResponseException.class);
-        exceptionRule.expectMessage("Fail to find model");
-        TestHelper.makeRequest(client(), "GET", "/_plugins/_ml/models/111222333", null, "", null);
+        ResponseException e = assertThrows(
+            ResponseException.class,
+            () -> TestHelper.makeRequest(client(), "GET", "/_plugins/_ml/models/111222333", null, "", null)
+        );
+        String msg = e.getMessage();
+        assert (msg.contains("Fail to find model") || msg.contains("Failed to find model"));
     }
 
     @Ignore