[ML] Wait for _infer to work after restart in full cluster restart tests (#93327) (#93566)

davidkyle · web-flow · commit ef118b668335 · 2023-02-07T12:11:51.000-05:00
Backport of #93327 Closes #93529
diff --git a/x-pack/qa/full-cluster-restart/src/javaRestTest/java/org/elasticsearch/xpack/restart/MLModelDeploymentFullClusterRestartIT.java b/x-pack/qa/full-cluster-restart/src/javaRestTest/java/org/elasticsearch/xpack/restart/MLModelDeploymentFullClusterRestartIT.java
@@ -104,8 +104,10 @@ public void testDeploymentSurvivesRestart() throws Exception {
                 request.addParameter("timeout", "70s");
             }));
             waitForDeploymentStarted(modelId);
-            assertInfer(modelId);
-            assertNewInfer(modelId);
+            assertBusy(() -> {
+                assertInfer(modelId);
+                assertNewInfer(modelId);
+            }, 90, TimeUnit.SECONDS);
             stopDeployment(modelId);
         }
     }

Original file line number	Diff line number	Diff line change
`@@ -104,8 +104,10 @@ public void testDeploymentSurvivesRestart() throws Exception {`
`104`	`104`	`request.addParameter("timeout", "70s");`
`105`	`105`	`}));`
`106`	`106`	`waitForDeploymentStarted(modelId);`
`107`		`- assertInfer(modelId);`
`108`		`- assertNewInfer(modelId);`
	`107`	`+ assertBusy(() -> {`
	`108`	`+ assertInfer(modelId);`
	`109`	`+ assertNewInfer(modelId);`
	`110`	`+ }, 90, TimeUnit.SECONDS);`
`109`	`111`	`stopDeployment(modelId);`
`110`	`112`	`}`
`111`	`113`	`}`