Skip to content

Commit 9098695

Browse files
committed
add async
1 parent 6cbfe19 commit 9098695

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

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

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727
import java.util.Map;
2828
import java.util.Set;
2929

30+
import static org.elasticsearch.core.TimeValue.timeValueSeconds;
3031
import static org.elasticsearch.test.hamcrest.ElasticsearchAssertions.assertAcked;
3132
import static org.elasticsearch.xpack.esql.EsqlTestUtils.getValuesList;
3233
import static org.hamcrest.Matchers.containsString;
@@ -90,7 +91,7 @@ protected void assertClusterMetadataSkipped(EsqlExecutionInfo.Cluster clusterMet
9091
}
9192

9293
protected EsqlQueryResponse runQuery(String query, Boolean ccsMetadataInResponse, QueryBuilder filter) {
93-
EsqlQueryRequest request = EsqlQueryRequest.syncEsqlQueryRequest();
94+
EsqlQueryRequest request = randomBoolean() ? EsqlQueryRequest.asyncEsqlQueryRequest() : EsqlQueryRequest.syncEsqlQueryRequest();
9495
request.query(query);
9596
request.pragmas(AbstractEsqlIntegTestCase.randomPragmas());
9697
request.profile(randomInt(5) == 2);
@@ -101,6 +102,7 @@ protected EsqlQueryResponse runQuery(String query, Boolean ccsMetadataInResponse
101102
if (filter != null) {
102103
request.filter(filter);
103104
}
105+
request.waitForCompletionTimeout(timeValueSeconds(30));
104106
return runQuery(request);
105107
}
106108

0 commit comments

Comments
 (0)