Skip to content

Commit b0ee146

Browse files
committed
fix test
1 parent ff16ebf commit b0ee146

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

x-pack/plugin/esql/src/internalClusterTest/java/org/elasticsearch/xpack/esql/action/CrossClustersCancellationIT.java

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121
import org.elasticsearch.compute.operator.exchange.ExchangeService;
2222
import org.elasticsearch.core.TimeValue;
2323
import org.elasticsearch.plugins.Plugin;
24+
import org.elasticsearch.tasks.TaskCancelledException;
2425
import org.elasticsearch.tasks.TaskInfo;
2526
import org.elasticsearch.test.AbstractMultiClustersTestCase;
2627
import org.elasticsearch.transport.TransportService;
@@ -32,6 +33,7 @@
3233
import java.util.ArrayList;
3334
import java.util.Collection;
3435
import java.util.List;
36+
import java.util.Map;
3537
import java.util.concurrent.TimeUnit;
3638

3739
import 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

Comments
 (0)