Add profile logging to non-csv tests

Author: ivancea

x-pack/plugin/esql/qa/server/src/main/java/org/elasticsearch/xpack/esql/qa/rest/ProfileLogger.java

@@ -25,7 +25,8 @@
  *     public ProfileLogger profileLogger = new ProfileLogger();
  *
  *     public void test() {
- *         var response = RestEsqlTestCase.runEsqlSync(..., profileLogger);
+ *         var response = RestEsqlTestCase.runEsql(requestObject, assertWarnings, profileLogger, mode);
+ *         // Or any of the other runEsql methods
  *     }
  * }
  * </pre>

x-pack/plugin/esql/qa/server/src/main/java/org/elasticsearch/xpack/esql/qa/rest/RestEsqlTestCase.java

@@ -36,6 +36,7 @@
 import org.junit.After;
 import org.junit.Assert;
 import org.junit.Before;
+import org.junit.Rule;
 
 import java.io.ByteArrayOutputStream;
 import java.io.IOException;
@@ -76,6 +77,9 @@
 
 public abstract class RestEsqlTestCase extends ESRestTestCase {
 
+    @Rule(order = Integer.MIN_VALUE)
+    public ProfileLogger profileLogger = new ProfileLogger();
+
     // Test runner will run multiple suites in parallel, with some of them requiring preserving state between
     // tests (like EsqlSpecTestCase), so test data (like index name) needs not collide and cleanup must be done locally.
     private static final String TEST_INDEX_NAME = "rest-esql-test";
@@ -455,7 +459,7 @@ public void testOutOfRangeComparisons() throws IOException {
                         "Line 1:29: java.lang.IllegalArgumentException: single-value function encountered multi-value"
                     )
                 );
-                var result = runEsql(query, assertWarnings, mode);
+                var result = runEsql(query, assertWarnings, profileLogger, mode);
 
                 var values = as(result.get("values"), ArrayList.class);
                 assertThat(
@@ -525,7 +529,7 @@ public void testInternalRange() throws IOException {
 
         for (String p : predicates) {
             var query = requestObjectBuilder().query(format(null, "from {} | where {}", testIndexName(), p));
-            var result = runEsql(query, new AssertWarnings.NoWarnings(), mode);
+            var result = runEsql(query, new AssertWarnings.NoWarnings(), profileLogger, mode);
             var values = as(result.get("values"), ArrayList.class);
             assertThat(
                 format(null, "Comparison [{}] should return all rows with single values.", p),
@@ -1240,7 +1244,7 @@ private static String expectedTextBody(String format, int count, @Nullable Chara
     }
 
     public Map<String, Object> runEsql(RequestObjectBuilder requestObject) throws IOException {
-        return runEsql(requestObject, new AssertWarnings.NoWarnings(), mode);
+        return runEsql(requestObject, new AssertWarnings.NoWarnings(), profileLogger, mode);
     }
 
     public static Map<String, Object> runEsqlSync(RequestObjectBuilder requestObject) throws IOException {