ci: opt-in dist-upgrade and reboot for the currency lane

tests/infrastructure/scripts/prepare-test-environment.test.sh

@@ -0,0 +1,95 @@
+#!/usr/bin/env bash
+# Self-check for prepare-test-environment.sh's opt-in dist-upgrade branch.
+#
+# Stubs sshpass/curl/sleep on PATH so both paths run offline in ~0s, then
+# asserts on the commands the script actually issued.
+#
+# Run: bash tests/infrastructure/scripts/prepare-test-environment.test.sh
+
+set -euo pipefail
+
+SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+TARGET="$SCRIPT_DIR/prepare-test-environment.sh"
+
+TMP="$(mktemp -d)"
+trap 'rm -rf "$TMP"' EXIT
+
+mkdir -p "$TMP/bin"
+
+# Fake sshpass: log every invocation, emit a plausible dpkg-query result.
+cat > "$TMP/bin/sshpass" <<'STUB'
+#!/usr/bin/env bash
+echo "$*" >> "$STUB_LOG"
+case "$*" in
+    *dpkg-query*) echo -e "proxmox-kernel-6.14\t6.14.11-1\npve-manager\t9.2.1" ;;
+esac
+exit 0
+STUB
+
+# wait-for-api.sh greps curl output for "version"; sleep must not really sleep.
+cat > "$TMP/bin/curl" <<'STUB'
+#!/usr/bin/env bash
+echo '{"data":{"version":"9.2.1"}}'
+STUB
+cat > "$TMP/bin/sleep" <<'STUB'
+#!/usr/bin/env bash
+exit 0
+STUB
+
+chmod +x "$TMP/bin/"*
+export PATH="$TMP/bin:$PATH"
+
+fail=0
+check() {
+    local desc="$1" haystack="$2" needle="$3" want="$4"
+    if grep -q -- "$needle" "$haystack"; then found=yes; else found=no; fi
+    if [[ "$found" == "$want" ]]; then
+        echo "  ok: $desc"
+    else
+        echo "  FAIL: $desc (expected present=$want, got present=$found)"
+        fail=1
+    fi
+}
+
+echo "case 1: no dist-upgrade argument — lane 1 path must be untouched"
+export STUB_LOG="$TMP/log1"
+: > "$STUB_LOG"
+bash "$TARGET" 10.0.0.1 secret > "$TMP/out1" 2>&1
+check "no dist-upgrade issued"      "$STUB_LOG" "dist-upgrade"    no
+check "no reboot issued"            "$STUB_LOG" "systemctl reboot" no
+check "no package set recorded"     "$STUB_LOG" "dpkg-query"      no
+check "storage still configured"    "$STUB_LOG" "pvesm set local" yes
+
+echo "case 2: dist-upgrade requested"
+export STUB_LOG="$TMP/log2"
+: > "$STUB_LOG"
+bash "$TARGET" 10.0.0.1 secret 1 "$TMP/packages.txt" > "$TMP/out2" 2>&1
+check "dist-upgrade issued"         "$STUB_LOG" "dist-upgrade"    yes
+check "reboot issued"               "$STUB_LOG" "systemctl reboot" yes
+check "package set recorded"        "$STUB_LOG" "dpkg-query"      yes
+check "storage still configured"    "$STUB_LOG" "pvesm set local" yes
+
+# The reboot must be issued after the upgrade, or the node records a package
+# set it never booted.
+upgrade_line=$(grep -n "dist-upgrade" "$STUB_LOG" | head -1 | cut -d: -f1)
+reboot_line=$(grep -n "systemctl reboot" "$STUB_LOG" | head -1 | cut -d: -f1)
+if [[ "$reboot_line" -gt "$upgrade_line" ]]; then
+    echo "  ok: reboot ordered after dist-upgrade"
+else
+    echo "  FAIL: reboot ordered before dist-upgrade"
+    fail=1
+fi
+
+if [[ -s "$TMP/packages.txt" ]]; then
+    echo "  ok: package file non-empty"
+else
+    echo "  FAIL: package file empty or missing"
+    fail=1
+fi
+
+if [[ "$fail" -eq 0 ]]; then
+    echo "PASS"
+else
+    echo "FAILED"
+    exit 1
+fi