Add memory threshold settings in ML Model API tests.

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

Author: nathaliellenaa

Diff for: 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

Diff for: 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

Diff for: 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

Diff for: 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

Diff for: 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

Diff for: 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:
+        status: COMPLETED
     output:
       model_id: payload.model_id
   - path: /_plugins/_ml/models/{model_id}