fix TransformRobustnessIT intermittent test failures (#51420)

ensure the cluster is not in some intermediate state when cleaning up. fixes #51347
elastic · Jan 24, 2020 · ada0616 · ada0616
1 parent 03df107
commit ada0616
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/...ts/src/test/java/org/elasticsearch/xpack/transform/integration/TransformRestTestCase.java b/...ts/src/test/java/org/elasticsearch/xpack/transform/integration/TransformRestTestCase.java
@@ -363,9 +363,12 @@ protected static void deleteTransform(String transformId) throws IOException {
     }
 
     @After
-    public void waitForDataFrame() throws Exception {
-        wipeTransforms();
-        waitForPendingDataFrameTasks();
+    public void waitForTransform() throws Exception {
+        if (preserveClusterUponCompletion() == false) {
+            ensureNoInitializingShards();
+            wipeTransforms();
+            waitForPendingTransformTasks();
+        }
     }
 
     @AfterClass
@@ -416,7 +419,7 @@ public void wipeTransforms() throws IOException {
         }
     }
 
-    protected static void waitForPendingDataFrameTasks() throws Exception {
+    protected static void waitForPendingTransformTasks() throws Exception {
         waitForPendingTasks(adminClient(), taskName -> taskName.startsWith(TransformField.TASK_NAME) == false);
     }