add test mods
This commit is contained in:
Meliox
parent
88c8610373
commit
ab31aa6964
206
src/Scripts/test/test-mods.sh
Normal file
206
src/Scripts/test/test-mods.sh
Normal file
@ -0,0 +1,206 @@
|
|||||||
|
#!/usr/bin/env bash
|
||||||
|
# src/Scripts/test/test-mods.sh
|
||||||
|
#
|
||||||
|
# Exercises pve-mod's patch engine against the installed Proxmox files.
|
||||||
|
#
|
||||||
|
# Usage: test-mods.sh <mod> | all
|
||||||
|
#
|
||||||
|
# Mods are auto-discovered from the [modules] section of the main config, so
|
||||||
|
# adding a new mod (a new [modules] key plus a patches/<mod>/ directory) is
|
||||||
|
# picked up automatically - no edits to this script are required.
|
||||||
|
#
|
||||||
|
# For each target mod it:
|
||||||
|
# 1. enables only that mod in the main config
|
||||||
|
# 2. turns on every conditional flag referenced by the mod's patches.list,
|
||||||
|
# so all of the mod's patches are exercised (e.g. node_info's gpu_history)
|
||||||
|
# 3. runs apply-patches.sh and asserts it exits 0
|
||||||
|
# 4. runs revert-patches.sh and asserts it reports no unclean reversions
|
||||||
|
# 5. restores the pristine Proxmox files before moving to the next mod
|
||||||
|
#
|
||||||
|
# Failure detection relies on the apply exit code: the patch engine performs an
|
||||||
|
# atomic preflight dry-run and exits non-zero if any mod cannot apply cleanly.
|
||||||
|
#
|
||||||
|
# Must run as root (it edits /etc/pve-mod and the patched system files).
|
||||||
|
|
||||||
|
set -u
|
||||||
|
|
||||||
|
MAIN_CONF="${PVE_MOD_MAIN_CONF:-/etc/pve-mod/pve-mod.conf}"
|
||||||
|
CONFD_DIR="${PVE_MOD_CONFD_DIR:-/etc/pve-mod/conf.d}"
|
||||||
|
PATCHES_DIR="${PVE_MOD_PATCHES_DIR:-/usr/lib/pve-mod/patches}"
|
||||||
|
APPLY="${PVE_MOD_APPLY:-/usr/lib/pve-mod/apply-patches.sh}"
|
||||||
|
REVERT="${PVE_MOD_REVERT:-/usr/lib/pve-mod/revert-patches.sh}"
|
||||||
|
|
||||||
|
# Proxmox files touched by the patches; snapshotted before and restored after
|
||||||
|
# each mod so every mod is tested from a pristine baseline.
|
||||||
|
PROXMOX_FILES=(
|
||||||
|
/usr/share/perl5/PVE/API2/Nodes.pm
|
||||||
|
/usr/share/pve-manager/js/pvemanagerlib.js
|
||||||
|
/usr/share/javascript/proxmox-widget-toolkit/proxmoxlib.js
|
||||||
|
/usr/share/javascript/proxmox-widget-toolkit/proxmoxlib.min.js
|
||||||
|
)
|
||||||
|
|
||||||
|
SNAP_DIR="$(mktemp -d)"
|
||||||
|
trap 'rm -rf "$SNAP_DIR"' EXIT
|
||||||
|
|
||||||
|
info() { echo "[test] $*"; }
|
||||||
|
warn() { echo "[test] WARNING: $*" >&2; }
|
||||||
|
|
||||||
|
# List the keys of the [modules] section in the main config, one per line.
|
||||||
|
list_modules() {
|
||||||
|
[[ -f "$MAIN_CONF" ]] || return 0
|
||||||
|
awk -F= '
|
||||||
|
/^\[/ { in_sec = ($0 == "[modules]") }
|
||||||
|
in_sec && /^[^#=]+=/ {
|
||||||
|
gsub(/^[[:space:]]+|[[:space:]]+$/, "", $1)
|
||||||
|
print $1
|
||||||
|
}
|
||||||
|
' "$MAIN_CONF"
|
||||||
|
}
|
||||||
|
|
||||||
|
# set_conf <file> <section> <key> <value>: set an existing key within a section.
|
||||||
|
set_conf() {
|
||||||
|
local file="$1" section="$2" key="$3" value="$4"
|
||||||
|
[[ -f "$file" ]] || { warn "config not found: $file"; return 1; }
|
||||||
|
awk -v sec="[$section]" -v k="$key" -v v="$value" '
|
||||||
|
/^\[/ { in_sec = ($0 == sec) }
|
||||||
|
{
|
||||||
|
if (in_sec && $0 ~ "^[[:space:]]*"k"[[:space:]]*=") {
|
||||||
|
print k"="v
|
||||||
|
} else {
|
||||||
|
print
|
||||||
|
}
|
||||||
|
}
|
||||||
|
' "$file" > "$file.tmp" && mv "$file.tmp" "$file"
|
||||||
|
}
|
||||||
|
|
||||||
|
# Disable every discovered mod in the main config.
|
||||||
|
disable_all_modules() {
|
||||||
|
local m
|
||||||
|
for m in $(list_modules); do
|
||||||
|
set_conf "$MAIN_CONF" modules "$m" 0
|
||||||
|
done
|
||||||
|
}
|
||||||
|
|
||||||
|
# Turn on every conditional flag a mod's patches.list references, so all of its
|
||||||
|
# patches become active. Conditions look like: <patch-file> section.key=value
|
||||||
|
enable_conditions() {
|
||||||
|
local mod="$1"
|
||||||
|
local manifest="$PATCHES_DIR/$mod/patches.list"
|
||||||
|
local mod_conf="$CONFD_DIR/$mod.conf"
|
||||||
|
[[ -f "$manifest" ]] || return 0
|
||||||
|
local line cond local_key want sect ckey
|
||||||
|
while IFS= read -r line; do
|
||||||
|
line="${line%%#*}"
|
||||||
|
line="$(echo "$line" | sed -E 's/^[[:space:]]+|[[:space:]]+$//g')"
|
||||||
|
[[ -z "$line" ]] && continue
|
||||||
|
# Skip lines without a condition (no whitespace after the patch name).
|
||||||
|
[[ "$line" == *[[:space:]]* ]] || continue
|
||||||
|
cond="$(echo "${line#*[[:space:]]}" | sed -E 's/^[[:space:]]+//')"
|
||||||
|
[[ "$cond" == *=* ]] || continue
|
||||||
|
local_key="${cond%%=*}"
|
||||||
|
want="${cond#*=}"
|
||||||
|
sect="${local_key%%.*}"
|
||||||
|
ckey="${local_key#*.}"
|
||||||
|
set_conf "$mod_conf" "$sect" "$ckey" "$want" \
|
||||||
|
&& info " enabled condition $local_key=$want in $(basename "$mod_conf")"
|
||||||
|
done < "$manifest"
|
||||||
|
}
|
||||||
|
|
||||||
|
snapshot_files() {
|
||||||
|
rm -rf "$SNAP_DIR"/*; mkdir -p "$SNAP_DIR"
|
||||||
|
local i=0 f
|
||||||
|
for f in "${PROXMOX_FILES[@]}"; do
|
||||||
|
if [[ -e "$f" || -L "$f" ]]; then
|
||||||
|
cp -a "$f" "$SNAP_DIR/$i"
|
||||||
|
fi
|
||||||
|
i=$((i+1))
|
||||||
|
done
|
||||||
|
}
|
||||||
|
|
||||||
|
restore_files() {
|
||||||
|
local i=0 f
|
||||||
|
for f in "${PROXMOX_FILES[@]}"; do
|
||||||
|
if [[ -e "$SNAP_DIR/$i" || -L "$SNAP_DIR/$i" ]]; then
|
||||||
|
rm -f "$f"
|
||||||
|
cp -a "$SNAP_DIR/$i" "$f"
|
||||||
|
fi
|
||||||
|
i=$((i+1))
|
||||||
|
done
|
||||||
|
}
|
||||||
|
|
||||||
|
# Test a single mod end to end. Returns 0 on success, 1 on failure.
|
||||||
|
test_one_mod() {
|
||||||
|
local mod="$1"
|
||||||
|
echo "::group::Testing mod: $mod"
|
||||||
|
local ok=0
|
||||||
|
|
||||||
|
if [[ ! -f "$PATCHES_DIR/$mod/patches.list" ]]; then
|
||||||
|
warn "no patch manifest for mod '$mod' ($PATCHES_DIR/$mod/patches.list)"
|
||||||
|
echo "::endgroup::"
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
disable_all_modules
|
||||||
|
set_conf "$MAIN_CONF" modules "$mod" 1
|
||||||
|
enable_conditions "$mod"
|
||||||
|
snapshot_files
|
||||||
|
|
||||||
|
info "Applying patches for '$mod'..."
|
||||||
|
local apply_log apply_rc
|
||||||
|
apply_log="$("$APPLY" 2>&1)"; apply_rc=$?
|
||||||
|
echo "$apply_log"
|
||||||
|
if [[ $apply_rc -ne 0 ]]; then
|
||||||
|
warn "apply-patches.sh failed for mod '$mod' (exit $apply_rc)"
|
||||||
|
ok=1
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [[ $ok -eq 0 ]]; then
|
||||||
|
info "Reverting patches for '$mod'..."
|
||||||
|
local revert_log
|
||||||
|
revert_log="$("$REVERT" 2>&1)"
|
||||||
|
echo "$revert_log"
|
||||||
|
if echo "$revert_log" | grep -qE "could not be reverted cleanly|reported an error"; then
|
||||||
|
warn "revert-patches.sh reported an unclean revert for mod '$mod'"
|
||||||
|
ok=1
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
|
||||||
|
restore_files
|
||||||
|
|
||||||
|
if [[ $ok -eq 0 ]]; then
|
||||||
|
info "PASS: $mod"
|
||||||
|
else
|
||||||
|
warn "FAIL: $mod"
|
||||||
|
fi
|
||||||
|
echo "::endgroup::"
|
||||||
|
return $ok
|
||||||
|
}
|
||||||
|
|
||||||
|
# ── main ──────────────────────────────────────────────────────────────────────
|
||||||
|
[[ $# -eq 1 ]] || { echo "Usage: $0 <mod>|all" >&2; exit 2; }
|
||||||
|
|
||||||
|
if ! command -v patch >/dev/null 2>&1; then
|
||||||
|
warn "'patch' command not found; install the 'patch' package."
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
targets=()
|
||||||
|
if [[ "$1" == "all" ]]; then
|
||||||
|
mapfile -t targets < <(list_modules)
|
||||||
|
[[ ${#targets[@]} -gt 0 ]] || { warn "no mods found in [modules] of $MAIN_CONF"; exit 1; }
|
||||||
|
else
|
||||||
|
targets=("$1")
|
||||||
|
fi
|
||||||
|
|
||||||
|
failed=()
|
||||||
|
for mod in "${targets[@]}"; do
|
||||||
|
test_one_mod "$mod" || failed+=("$mod")
|
||||||
|
done
|
||||||
|
|
||||||
|
echo ""
|
||||||
|
if [[ ${#failed[@]} -gt 0 ]]; then
|
||||||
|
warn "Failed mods: ${failed[*]}"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
info "All tested mods passed: ${targets[*]}"
|
||||||
|
exit 0
|
||||||
Loading…
Reference in New Issue
Block a user