Wait for ML templates after in TooManyJobsIT (#54793)

davidkyle · web-flow · commit a92c1ddf5ccd · 2020-04-06T12:48:16.000+01:00
diff --git a/x-pack/plugin/ml/src/test/java/org/elasticsearch/xpack/ml/integration/TooManyJobsIT.java b/x-pack/plugin/ml/src/test/java/org/elasticsearch/xpack/ml/integration/TooManyJobsIT.java
@@ -74,6 +74,7 @@ public void testLazyNodeValidation() throws Exception {
         }
         logger.info("Started [{}] nodes", numNodes);
         ensureStableCluster(numNodes);
+        ensureTemplatesArePresent();
         logger.info("[{}] is [{}]", MachineLearning.MAX_LAZY_ML_NODES.getKey(), maxNumberOfLazyNodes);
         // Set our lazy node number
         assertTrue(client().admin()
@@ -208,6 +209,7 @@ private void startMlCluster(int numNodes, int maxNumberOfWorkersPerNode) throws
         }
         logger.info("Started [{}] nodes", numNodes);
         ensureStableCluster(numNodes);
+        ensureTemplatesArePresent();
     }
 
     private long calculateMaxMlMemory() {

Original file line number	Diff line number	Diff line change
`@@ -74,6 +74,7 @@ public void testLazyNodeValidation() throws Exception {`
`74`	`74`	`}`
`75`	`75`	`logger.info("Started [{}] nodes", numNodes);`
`76`	`76`	`ensureStableCluster(numNodes);`
	`77`	`+ ensureTemplatesArePresent();`
`77`	`78`	`logger.info("[{}] is [{}]", MachineLearning.MAX_LAZY_ML_NODES.getKey(), maxNumberOfLazyNodes);`
`78`	`79`	`// Set our lazy node number`
`79`	`80`	`assertTrue(client().admin()`
`@@ -208,6 +209,7 @@ private void startMlCluster(int numNodes, int maxNumberOfWorkersPerNode) throws`
`208`	`209`	`}`
`209`	`210`	`logger.info("Started [{}] nodes", numNodes);`
`210`	`211`	`ensureStableCluster(numNodes);`
	`212`	`+ ensureTemplatesArePresent();`
`211`	`213`	`}`
`212`	`214`
`213`	`215`	`private long calculateMaxMlMemory() {`