random tests

Author: dnhatn

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

@@ -150,6 +150,9 @@ protected final EsqlQueryResponse run(String esqlCommands, QueryPragmas pragmas)
     protected EsqlQueryResponse run(String esqlCommands, QueryPragmas pragmas, QueryBuilder filter) {
         EsqlQueryRequest request = EsqlQueryRequest.syncEsqlQueryRequest();
         request.query(esqlCommands);
+        if (randomBoolean()) {
+            request.allowPartialResults(randomBoolean());
+        }
         if (pragmas != null) {
             request.pragmas(pragmas);
         }

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

@@ -79,6 +79,7 @@ protected EsqlQueryResponse run(EsqlQueryRequest request) {
     }
 
     private EsqlQueryResponse runQueryWithDisruption(EsqlQueryRequest request) {
+        request.allowPartialResults(false);
         final ServiceDisruptionScheme disruptionScheme = addRandomDisruptionScheme();
         logger.info("--> start disruption scheme [{}]", disruptionScheme);
         disruptionScheme.startDisrupting();
@@ -113,6 +114,7 @@ private EsqlQueryResponse runQueryWithDisruption(EsqlQueryRequest request) {
             ensureBlocksReleased();
             logger.info("--> failed to execute esql query with disruption; retrying...", e);
             EsqlTestUtils.assertEsqlFailure(e);
+            request.allowPartialResults(false);
             return client().execute(EsqlQueryAction.INSTANCE, request).actionGet(2, TimeUnit.MINUTES);
         }
     }