Add wheel count validation for release workflows

Author: cjdsellers

.github/workflows/build.yml

@@ -386,6 +386,7 @@ jobs:
       CLOUDFLARE_R2_URL: ${{ secrets.CLOUDFLARE_R2_URL }}
       CLOUDFLARE_R2_REGION: "auto"
       CLOUDFLARE_R2_BUCKET_NAME: "packages"
+      EXPECTED_WHEEL_COUNT: 9  # 3 Python versions × 3 platforms (ARM skipped on develop)
     steps:
       # https://github.com/step-security/harden-runner
       - uses: step-security/harden-runner@f4a75cfd619ee5ce8d5b864b0d183aff3c69b55a # v2.13.1
@@ -405,9 +406,12 @@ jobs:
         uses: actions/download-artifact@634f93cb2916e3fdff6788551b99b062d0335ce0 # v5.0.0
         with:
           path: dist
-          pattern: "*.whl"
+          pattern: "nautilus_trader-*.whl"
           merge-multiple: true
 
+      - name: Validate wheel count
+        run: bash ./scripts/ci/validate-wheel-count.sh "$EXPECTED_WHEEL_COUNT"
+
       # https://github.com/actions/attest-build-provenance
       - name: Attest wheel provenance
         uses: actions/attest-build-provenance@977bb373ede98d70efdf65b84cb5f73e068dcc2a # v3.0.0
@@ -446,6 +450,7 @@ jobs:
       CLOUDFLARE_R2_URL: ${{ secrets.CLOUDFLARE_R2_URL }}
       CLOUDFLARE_R2_REGION: "auto"
       CLOUDFLARE_R2_BUCKET_NAME: "packages"
+      EXPECTED_WHEEL_COUNT: 12  # 3 Python versions × 4 platforms
     steps:
       # https://github.com/step-security/harden-runner
       - uses: step-security/harden-runner@f4a75cfd619ee5ce8d5b864b0d183aff3c69b55a # v2.13.1
@@ -465,9 +470,12 @@ jobs:
         uses: actions/download-artifact@634f93cb2916e3fdff6788551b99b062d0335ce0 # v5.0.0
         with:
           path: dist
-          pattern: "*.whl"
+          pattern: "nautilus_trader-*.whl"
           merge-multiple: true
 
+      - name: Validate wheel count
+        run: bash ./scripts/ci/validate-wheel-count.sh "$EXPECTED_WHEEL_COUNT"
+
       # https://github.com/actions/attest-build-provenance
       - name: Attest wheel provenance
         uses: actions/attest-build-provenance@977bb373ede98d70efdf65b84cb5f73e068dcc2a # v3.0.0
@@ -506,6 +514,7 @@ jobs:
       CLOUDFLARE_R2_URL: ${{ secrets.CLOUDFLARE_R2_URL }}
       CLOUDFLARE_R2_REGION: "auto"
       CLOUDFLARE_R2_BUCKET_NAME: "packages"
+      EXPECTED_WHEEL_COUNT: 12  # 3 Python versions × 4 platforms
     steps:
       # https://github.com/step-security/harden-runner
       - uses: step-security/harden-runner@f4a75cfd619ee5ce8d5b864b0d183aff3c69b55a # v2.13.1
@@ -538,9 +547,12 @@ jobs:
         uses: actions/download-artifact@634f93cb2916e3fdff6788551b99b062d0335ce0 # v5.0.0
         with:
           path: dist
-          pattern: "*.whl"
+          pattern: "nautilus_trader-*.whl"
           merge-multiple: true
 
+      - name: Validate wheel count
+        run: bash ./scripts/ci/validate-wheel-count.sh "$EXPECTED_WHEEL_COUNT"
+
       # https://github.com/actions/attest-build-provenance
       - name: Attest wheel provenance
         uses: actions/attest-build-provenance@977bb373ede98d70efdf65b84cb5f73e068dcc2a # v3.0.0

scripts/ci/validate-wheel-count.sh

@@ -0,0 +1,30 @@
+#!/usr/bin/env bash
+set -euo pipefail
+
+# Validate wheel count matches expected count
+# Usage: validate-wheel-count.sh <expected_count>
+
+if [ $# -ne 1 ]; then
+  echo "Usage: $0 <expected_count>" >&2
+  exit 1
+fi
+
+expected_count=$1
+
+if ! [[ "$expected_count" =~ ^[0-9]+$ ]]; then
+  echo "ERROR: expected_count must be a positive integer, got: $expected_count" >&2
+  exit 1
+fi
+
+echo "Validating wheel count in dist/ directory..."
+
+wheel_count=$(find dist/ -name "*.whl" -type f | wc -l)
+
+if [ "$wheel_count" -ne "$expected_count" ]; then
+  echo "ERROR: Expected $expected_count wheels, found $wheel_count" >&2
+  echo "Downloaded wheels:" >&2
+  find dist/ -name "*.whl" -type f -ls >&2
+  exit 1
+fi
+
+echo "✓ Validated: Found all $expected_count wheels"