accept past tense error message in getModel IT

Signed-off-by: HenryL27 <[email protected]>
HenryL27 · Dec 7, 2023 · a7df1df · a7df1df
1 parent c6d8141
commit a7df1df
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