update

2026-07-25 19:35:17 -05:00
parent f2da5b3062
commit 7102550f4e
12 changed files with 430 additions and 30 deletions
@@ -0,0 +1,70 @@
+#!/usr/bin/env bash
+set -u -o pipefail
+
+TEST_ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
+source "${TEST_ROOT}/tests/testlib.sh"
+source "${TEST_ROOT}/inc/evacuate-proxmox-node.sh"
+
+set_routing_fixture() {
+	MIGRATION_NODES=(node2 node3)
+	NODE_STORAGE_CACHE=()
+	NODE_STORAGE_CACHE[node2]='shared-a,shared-b'
+	NODE_STORAGE_CACHE[node3]='shared-a,shared-c'
+	PREFERRED_TARGET=node2
+}
+
+set_routing_fixture
+assert_success "preferred destination is eligible" \
+	choose_destination shared-a '' 0
+assert_equal node2 "$CHOSEN_DESTINATION" "preferred destination selected"
+
+set_routing_fixture
+assert_success "affinity fallback is available" \
+	choose_destination shared-a node3 1
+assert_equal node3 "$CHOSEN_DESTINATION" "affinity destination selected"
+assert_equal "routed to satisfy HA node-affinity preference" "$CHOSEN_NOTE" \
+	"affinity routing explanation"
+
+set_routing_fixture
+assert_success "storage fallback is available" \
+	choose_destination shared-c '' 0
+assert_equal node3 "$CHOSEN_DESTINATION" "storage-compatible destination selected"
+
+set_routing_fixture
+assert_failure "strict affinity blocks two noncompliant destinations" \
+	choose_destination shared-a node4 1
+
+set_routing_fixture
+MIGRATION_NODES=(node2)
+assert_success "sole eligible node overrides affinity" \
+	choose_destination shared-a node4 1
+assert_equal node2 "$CHOSEN_DESTINATION" "sole eligible destination selected"
+assert_equal "only eligible node; HA node-affinity preference overridden" \
+	"$CHOSEN_NOTE" "sole-node override explanation"
+
+TEST_RULES_FILE="$(mktemp /tmp/tapm-ha-rules.XXXXXX)"
+cleanup_evacuation_tests() {
+	if [[ "$TEST_RULES_FILE" == /tmp/tapm-ha-rules.* ]]; then
+		rm -f -- "$TEST_RULES_FILE"
+	fi
+}
+trap cleanup_evacuation_tests EXIT
+
+printf '%s\n' \
+	'node-affinity: preferred-nodes' \
+	'    resources vm:210,ct:211' \
+	'    nodes node3:20,node2:10' \
+	'    strict 1' >"$TEST_RULES_FILE"
+HA_RULES_FILE="$TEST_RULES_FILE"
+
+assert_equal $'preferred-nodes\x1f1\x1fnode3,node2' \
+	"$(get_guest_node_affinity qemu 210)" \
+	"QEMU affinity parsing"
+assert_equal $'preferred-nodes\x1f1\x1fnode3,node2' \
+	"$(get_guest_node_affinity lxc 211)" \
+	"LXC affinity parsing"
+assert_equal $'none\x1f0\x1f' \
+	"$(get_guest_node_affinity qemu 999)" \
+	"guest without affinity"
+
+finish_tests