Fix flaky tests in ML Model APIs (#782)

* Add memory threshold settings in ML Model API tests.

Signed-off-by: Nathalie Jonathan <[email protected]>

* Fixed response payload status property to response payload state property in upload.yaml, updated CHANGELOG.md.

Signed-off-by: Nathalie Jonathan <[email protected]>

---------

Signed-off-by: Nathalie Jonathan <[email protected]>

Author: nathaliellenaa

CHANGELOG.md

@@ -96,6 +96,7 @@ Inspired from [Keep a Changelog](https://keepachangelog.com/en/1.0.0/)
 - Fixed ingest API schemas ([#766](https://github.com/opensearch-project/opensearch-api-specification/pull/766))
 - Fixed CAT API schemas ([#769](https://github.com/opensearch-project/opensearch-api-specification/pull/769))
 - Fixed core API schemas ([#777](https://github.com/opensearch-project/opensearch-api-specification/pull/777))
+- Fixed Memory Circuit Breaker flaky test in ML Model APIs ([#782](https://github.com/opensearch-project/opensearch-api-specification/pull/782))
 
 ### Changed
 - Changed `tasks._common:TaskInfo` and `tasks._common:TaskGroup` to be composed of a `tasks._common:TaskInfoBase` ([#683](https://github.com/opensearch-project/opensearch-api-specification/pull/683))

tests/plugins/ml/ml/models/load.yaml

@@ -2,6 +2,12 @@ $schema: ../../../../../json_schemas/test_story.schema.yaml
 
 description: Test the deployment of a model using deprecated load API.
 prologues:
+  - path: /_cluster/settings
+    method: PUT
+    request:
+      payload:
+        persistent:
+          plugins.ml_commons.native_memory_threshold: 100
   - path: /_plugins/_ml/models/_register
     id: register_model
     method: POST

tests/plugins/ml/ml/models/predict.yaml

@@ -10,7 +10,7 @@ prologues:
         persistent:
           plugins.ml_commons.only_run_on_ml_node: false
           plugins.ml_commons.model_access_control_enabled: true
-          plugins.ml_commons.native_memory_threshold: 99
+          plugins.ml_commons.native_memory_threshold: 100
   - path: /_plugins/_ml/model_groups/_register
     id: create_model_group
     method: POST

tests/plugins/ml/ml/models/search.yaml

@@ -3,6 +3,12 @@ $schema: ../../../../../json_schemas/test_story.schema.yaml
 description: Test the search of models.
 version: '>= 2.7'
 prologues:
+  - path: /_cluster/settings
+    method: PUT
+    request:
+      payload:
+        persistent:
+          plugins.ml_commons.native_memory_threshold: 100
   - path: /_plugins/_ml/models/_register
     id: register_model
     method: POST

tests/plugins/ml/ml/models/undeploy.yaml

@@ -9,6 +9,7 @@ prologues:
       payload:
         persistent:
           plugins.ml_commons.allow_custom_deployment_plan: true
+          plugins.ml_commons.native_memory_threshold: 100
   - path: /_plugins/_ml/models/_register
     id: register_model
     method: POST

tests/plugins/ml/ml/models/unload.yaml

@@ -8,6 +8,7 @@ prologues:
       payload:
         persistent:
           plugins.ml_commons.allow_custom_deployment_plan: true
+          plugins.ml_commons.native_memory_threshold: 100
   - path: /_plugins/_ml/models/_register
     id: register_model
     method: POST

tests/plugins/ml/ml/models/upload.yaml

@@ -1,6 +1,13 @@
 $schema: ../../../../../json_schemas/test_story.schema.yaml
 
 description: Test the creation of models using upload API (deprecated).
+prologues:
+  - path: /_cluster/settings
+    method: PUT
+    request:
+      payload:
+        persistent:
+          plugins.ml_commons.native_memory_threshold: 100
 epilogues:
   - path: /_plugins/_ml/tasks/{task_id}
     id: get_completed_task
@@ -10,6 +17,10 @@ epilogues:
     retry:
       count: 3
       wait: 10000
+    response:
+      status: 200
+      payload:
+        state: COMPLETED
     output:
       model_id: payload.model_id
   - path: /_plugins/_ml/models/{model_id}