[ML] Wait for ML indices in rolling upgrade tests (#30615)

elastic · May 16, 2018 · c2e0061 · c2e0061
1 parent b1eda60
commit c2e0061
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/...a/rolling-upgrade/src/test/resources/rest-api-spec/test/mixed_cluster/30_ml_jobs_crud.yml b/...a/rolling-upgrade/src/test/resources/rest-api-spec/test/mixed_cluster/30_ml_jobs_crud.yml
@@ -110,6 +110,13 @@ setup:
       xpack.ml.close_job:
         job_id: mixed-cluster-job
 
+# Wait for indices to be fully allocated before
+# killing the node
+  - do:
+     cluster.health:
+       index: [".ml-state", ".ml-anomalies-shared"]
+       wait_for_status: green
+
 ---
 "Test get job with rules":
 

diff --git a/.../qa/rolling-upgrade/src/test/resources/rest-api-spec/test/old_cluster/30_ml_jobs_crud.yml b/.../qa/rolling-upgrade/src/test/resources/rest-api-spec/test/old_cluster/30_ml_jobs_crud.yml
@@ -49,6 +49,13 @@
         job_id: old-cluster-job
   - match: { count: 1 }
 
+# Wait for indices to be fully allocated before
+# killing the node
+  - do:
+     cluster.health:
+       index: [".ml-state", ".ml-anomalies-shared"]
+       wait_for_status: green
+
 ---
 "Put job on the old cluster with the default model memory limit and post some data":
   - do:
@@ -96,6 +103,13 @@
         job_id: no-model-memory-limit-job
   - match: { count: 201 }
 
+# Wait for indices to be fully allocated before
+# killing the node
+  - do:
+     cluster.health:
+       index: [".ml-state", ".ml-anomalies-shared"]
+       wait_for_status: green
+
 ---
 "Put job with empty strings in the configuration":
   - do: