From 2ce4399316ebd1ae19832f08f3e11e060885b84e Mon Sep 17 00:00:00 2001
From: Xiang Fu <xiangfu@startree.ai>
Date: Mon, 9 Mar 2026 13:08:33 -0700
Subject: [PATCH] Improve integration CI surefire fork diagnostics and heap
 sizing

---
 .../pr-tests/.pinot_tests_integration.sh      | 41 ++++++++++++++++++-
 1 file changed, 39 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/scripts/pr-tests/.pinot_tests_integration.sh b/.github/workflows/scripts/pr-tests/.pinot_tests_integration.sh
index 6ffea61c13d0..227b295d7734 100755
--- a/.github/workflows/scripts/pr-tests/.pinot_tests_integration.sh
+++ b/.github/workflows/scripts/pr-tests/.pinot_tests_integration.sh
@@ -25,13 +25,50 @@ java -version
 ifconfig
 netstat -i
 
+print_surefire_dumps() {
+  local reports_dir="target/surefire-reports"
+  local dump_files
+
+  if [ ! -d "${reports_dir}" ]; then
+    echo "Surefire reports directory not found: ${reports_dir}"
+    return
+  fi
+
+  dump_files="$(find "${reports_dir}" -maxdepth 1 -type f \
+    \( -name "*.dump" -o -name "*.dumpstream" -o -name "*jvmRun*" \) | sort)"
+  if [ -z "${dump_files}" ]; then
+    echo "No Surefire dump files found under ${reports_dir}"
+    return
+  fi
+
+  echo "Printing Surefire dump files from ${reports_dir}"
+  while IFS= read -r dump_file; do
+    [ -z "${dump_file}" ] && continue
+    echo "===== BEGIN ${dump_file} ====="
+    cat "${dump_file}"
+    echo "===== END ${dump_file} ====="
+  done <<< "${dump_files}"
+}
+
 # Integration Tests
 cd pinot-integration-tests || exit 1
 if [ "$RUN_TEST_SET" == "1" ]; then
   mvn test \
-      -P github-actions,codecoverage,integration-tests-set-1 && exit 0 || exit 1
+      -P github-actions,codecoverage,integration-tests-set-1 || {
+    print_surefire_dumps
+    exit 1
+  }
+  exit 0
 fi
 if [ "$RUN_TEST_SET" == "2" ]; then
   mvn test \
-      -P github-actions,codecoverage,integration-tests-set-2 && exit 0 || exit 1
+      -DargLine="-Xms1g -Xmx2g -Dlog4j2.configurationFile=log4j2.xml" \
+      -P github-actions,codecoverage,integration-tests-set-2 || {
+    print_surefire_dumps
+    exit 1
+  }
+  exit 0
 fi
+
+echo "Unsupported RUN_TEST_SET value: ${RUN_TEST_SET}"
+exit 1