Datafeed jobs it wait for task cancel (elastic#120177)

* Fix task cancellation in DatafeedJobsIT.

* fix

* unmute DatafeedJobsIT

Author: jan-elastic

muted-tests.yml

@@ -232,8 +232,6 @@ tests:
 - class: org.elasticsearch.xpack.ilm.actions.SearchableSnapshotActionIT
   method: testUpdatePolicyToAddPhasesYieldsInvalidActionsToBeSkipped
   issue: https://github.com/elastic/elasticsearch/issues/118406
-- class: org.elasticsearch.xpack.ml.integration.DatafeedJobsIT
-  issue: https://github.com/elastic/elasticsearch/issues/120088
 - class: org.elasticsearch.xpack.security.QueryableReservedRolesIT
   method: testConfiguredReservedRolesAfterClosingAndOpeningIndex
   issue: https://github.com/elastic/elasticsearch/issues/120127

x-pack/plugin/ml/qa/native-multi-node-tests/src/javaRestTest/java/org/elasticsearch/xpack/ml/integration/DatafeedJobsIT.java

@@ -83,9 +83,12 @@ public void cleanup() {
         updateClusterSettings(Settings.builder().putNull("logger.org.elasticsearch.xpack.ml.datafeed"));
         cleanUp();
         // Race conditions between closing and killing tasks in these tests,
-        // sometimes result in lingering persistent tasks (such as "_close"),
-        // which cause subsequent tests to fail.
-        client().execute(TransportCancelTasksAction.TYPE, new CancelTasksRequest());
+        // sometimes result in lingering persistent close tasks, which cause
+        // subsequent tests to fail. Therefore, they're explicitly cancelled.
+        CancelTasksRequest cancelTasksRequest = new CancelTasksRequest();
+        cancelTasksRequest.setActions("*close*");
+        cancelTasksRequest.setWaitForCompletion(true);
+        client().execute(TransportCancelTasksAction.TYPE, cancelTasksRequest).actionGet();
     }
 
     public void testLookbackOnly() throws Exception {