2121import org .elasticsearch .compute .operator .exchange .ExchangeService ;
2222import org .elasticsearch .core .TimeValue ;
2323import org .elasticsearch .plugins .Plugin ;
24+ import org .elasticsearch .tasks .TaskCancelledException ;
2425import org .elasticsearch .tasks .TaskInfo ;
2526import org .elasticsearch .test .AbstractMultiClustersTestCase ;
2627import org .elasticsearch .transport .TransportService ;
3233import java .util .ArrayList ;
3334import java .util .Collection ;
3435import java .util .List ;
36+ import java .util .Map ;
3537import java .util .concurrent .TimeUnit ;
3638
3739import static org .elasticsearch .xpack .esql .EsqlTestUtils .getValuesList ;
@@ -250,7 +252,7 @@ public void testCancelSkipUnavailable() throws Exception {
250252 request .includeCCSMetadata (true );
251253 PlainActionFuture <EsqlQueryResponse > requestFuture = new PlainActionFuture <>();
252254 client ().execute (EsqlQueryAction .INSTANCE , request , requestFuture );
253- assertTrue (PauseFieldPlugin .startEmitting .await (30 , TimeUnit .SECONDS ));
255+ assertTrue (SimplePauseFieldPlugin .startEmitting .await (30 , TimeUnit .SECONDS ));
254256 List <TaskInfo > rootTasks = new ArrayList <>();
255257 assertBusy (() -> {
256258 List <TaskInfo > tasks = client (REMOTE_CLUSTER ).admin ()
@@ -278,7 +280,7 @@ public void testCancelSkipUnavailable() throws Exception {
278280 }
279281 });
280282 } finally {
281- PauseFieldPlugin .allowEmitting .countDown ();
283+ SimplePauseFieldPlugin .allowEmitting .countDown ();
282284 }
283285 var resp = requestFuture .actionGet ();
284286 EsqlExecutionInfo executionInfo = resp .getExecutionInfo ();
@@ -305,8 +307,8 @@ public void testCloseSkipUnavailable() throws Exception {
305307 """ );
306308 request .pragmas (randomPragmas ());
307309 var requestFuture = client ().execute (EsqlQueryAction .INSTANCE , request );
308- assertTrue (PauseFieldPlugin .startEmitting .await (30 , TimeUnit .SECONDS ));
309- PauseFieldPlugin .allowEmitting .countDown ();
310+ assertTrue (SimplePauseFieldPlugin .startEmitting .await (30 , TimeUnit .SECONDS ));
311+ SimplePauseFieldPlugin .allowEmitting .countDown ();
310312 cluster (REMOTE_CLUSTER ).close ();
311313 try (var resp = requestFuture .actionGet ()) {
312314 EsqlExecutionInfo executionInfo = resp .getExecutionInfo ();
0 commit comments