From 79bdd4b8f225efea2d601fc5696343c88614607d Mon Sep 17 00:00:00 2001
From: yew1eb <yew1eb@gmail.com>
Date: Fri, 23 Jan 2026 22:42:28 +0800
Subject: [PATCH] [AURON #1951] Add skip-unit-tests switch for CI workflows

---
 .github/workflows/celeborn.yml       | 1 +
 .github/workflows/tpcds-reusable.yml | 9 +++++++--
 .github/workflows/uniffle.yml        | 1 +
 3 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/celeborn.yml b/.github/workflows/celeborn.yml
index d8b4c4b0f..f3555b3c4 100644
--- a/.github/workflows/celeborn.yml
+++ b/.github/workflows/celeborn.yml
@@ -44,6 +44,7 @@ jobs:
     name: Test Celeborn ${{ matrix.celebornver }}
     with:
       celebornver: ${{ matrix.celebornver }}
+      skip-unit-tests: true
       sparkver: "spark-3.5"
       hadoop-profile: 'hadoop3'
       scalaver: "2.12"
diff --git a/.github/workflows/tpcds-reusable.yml b/.github/workflows/tpcds-reusable.yml
index 9ed97a15b..48df7def1 100644
--- a/.github/workflows/tpcds-reusable.yml
+++ b/.github/workflows/tpcds-reusable.yml
@@ -44,6 +44,11 @@ on:
         required: false
         type: string
         default: '2.12'
+      skip-unit-tests:
+        description: 'Whether to skip unit tests (true/false)'
+        required: false
+        type: string
+        default: 'false'
       celebornver:
         description: 'Optional Celeborn version'
         required: false
@@ -146,7 +151,7 @@ jobs:
             UNIFFLE_NUMBER="${UNIFFLE_NUMBER#uniffle-}"
           fi
           
-          CMD="./auron-build.sh --pre --sparkver $SPARK_NUMBER --scalaver ${{ inputs.scalaver }} --skiptests false"
+          CMD="./auron-build.sh --pre --sparkver $SPARK_NUMBER --scalaver ${{ inputs.scalaver }} --skiptests ${{ inputs.skip-unit-tests }}"
           if [ -n "${{ inputs.celebornver }}" ]; then
             CMD="$CMD --celeborn $CELEBORN_NUMBER"
           fi
@@ -155,7 +160,7 @@ jobs:
           fi
 
           SPARK_TESTS="${{ inputs.sparktests }}"
-          if [[ "$SPARK_TESTS" == "true" ]]; then
+          if [[ "${{ inputs.skip-unit-tests }}" != "true" && "$SPARK_TESTS" == "true" ]]; then
             CMD="$CMD --sparktests true"
           fi
 
diff --git a/.github/workflows/uniffle.yml b/.github/workflows/uniffle.yml
index ee7941f07..c2dee077d 100644
--- a/.github/workflows/uniffle.yml
+++ b/.github/workflows/uniffle.yml
@@ -48,6 +48,7 @@ jobs:
       scalaver: "2.12"
       hadoopver: ${{ matrix.hadoopver }}
       unifflever: ${{ matrix.unifflever }}
+      skip-unit-tests: true
       extrasparkconf: >-
         --conf spark.shuffle.manager=org.apache.spark.sql.execution.auron.shuffle.uniffle.AuronUniffleShuffleManager
         --conf spark.serializer=org.apache.spark.serializer.KryoSerializer