fixing security integ test (opensearch-project#3646) (opensearch-project#3780)

Signed-off-by: Dhrubo Saha <[email protected]>
(cherry picked from commit f19614e)

Co-authored-by: Dhrubo Saha <[email protected]>

Author: opensearch-trigger-bot[bot]

plugin/src/test/java/org/opensearch/ml/rest/SecureMLRestIT.java

@@ -408,10 +408,31 @@ public void testTrainWithReadOnlyMLAccess() throws IOException {
     }
 
     public void testPredictWithReadOnlyMLAccess() throws IOException {
-        exceptionRule.expect(ResponseException.class);
-        exceptionRule.expectMessage("no permissions for [cluster:admin/opensearch/ml/predict]");
         KMeansParams kMeansParams = KMeansParams.builder().build();
-        predict(mlReadOnlyClient, FunctionName.KMEANS, "modelId", irisIndex, kMeansParams, searchSourceBuilder, null);
+        train(mlFullAccessClient, FunctionName.KMEANS, irisIndex, kMeansParams, searchSourceBuilder, trainResult -> {
+            String modelId = (String) trainResult.get("model_id");
+            assertNotNull(modelId);
+            String status = (String) trainResult.get("status");
+            assertEquals(MLTaskState.COMPLETED.name(), status);
+            try {
+                // Verify the model exists
+                getModel(mlFullAccessClient, modelId, model -> {
+                    String algorithm = (String) model.get("algorithm");
+                    assertEquals(FunctionName.KMEANS.name(), algorithm);
+                });
+
+                // Attempt prediction with read-only client and expect a permission error
+                ResponseException exception = assertThrows(ResponseException.class, () -> {
+                    predict(mlReadOnlyClient, FunctionName.KMEANS, modelId, irisIndex, kMeansParams, searchSourceBuilder, null);
+                });
+
+                // Verify the exception message
+                assertTrue(exception.getMessage().contains("no permissions for [cluster:admin/opensearch/ml/predict]"));
+
+            } catch (IOException e) {
+                fail("Unexpected IOException: " + e.getMessage());
+            }
+        }, false);
     }
 
     public void testTrainAndPredictWithFullAccess() throws IOException {