From 4c1016d23958d8a87eeb831daed3fa441c521044 Mon Sep 17 00:00:00 2001
From: Farhantigadi <109067327+Farhantigadi@users.noreply.github.com>
Date: Thu, 8 Jan 2026 21:46:39 +0530
Subject: [PATCH] FINERACT-2434: Add input validation to split-tests.sh script
 (#5296)

Co-authored-by: FarhanAlstonair <Farhan@github123>
---
 scripts/split-tests.sh | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/scripts/split-tests.sh b/scripts/split-tests.sh
index 713466917..fc7fd6e14 100755
--- a/scripts/split-tests.sh
+++ b/scripts/split-tests.sh
@@ -28,6 +28,21 @@ if [[ -z "$TOTAL_SHARDS" || -z "$SHARD_INDEX" ]]; then
   exit 1
 fi
 
+if ! [[ "$TOTAL_SHARDS" =~ ^[1-9][0-9]*$ ]]; then
+  echo "ERROR: <total-shards> must be a positive integer."
+  exit 1
+fi
+
+if ! [[ "$SHARD_INDEX" =~ ^[1-9][0-9]*$ ]]; then
+  echo "ERROR: <shard-index> must be a positive integer."
+  exit 1
+fi
+
+if [[ "$SHARD_INDEX" -gt "$TOTAL_SHARDS" ]]; then
+  echo "ERROR: <shard-index> ($SHARD_INDEX) must be between 1 and <total-shards> ($TOTAL_SHARDS)."
+  exit 1
+fi
+
 echo "🔍 Searching for eligible JUnit test classes..."
 
 ALL_TESTS=$(find . -type f -path "*/src/test/java/*.java" \