109 lines
2.9 KiB
Bash
Executable File
109 lines
2.9 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
TEST_ROOT="$(cd "$SCRIPT_DIR/.." && pwd)"
|
|
TMP_ROOT="$TEST_ROOT/tmp"
|
|
API_BASE="http://localhost:32300/api/v1/master"
|
|
NODE_ID="$(cat "$TMP_ROOT/node_id")"
|
|
AGENT_HOSTNAME="dev-e2euser-e2einst-pod-0"
|
|
NETWORK_NAME="tests_default"
|
|
NEW_AGENT_IP="172.28.0.200"
|
|
|
|
compose() {
|
|
if docker compose version >/dev/null 2>&1; then
|
|
docker compose "$@"
|
|
else
|
|
docker-compose "$@"
|
|
fi
|
|
}
|
|
|
|
before_file="$TMP_ROOT/before_restart.json"
|
|
curl -sS "$API_BASE/nodes/$NODE_ID" -o "$before_file"
|
|
prev_last_updated=$(python3 - "$before_file" <<'PY'
|
|
import json, sys
|
|
with open(sys.argv[1]) as handle:
|
|
node = json.load(handle)
|
|
print(node.get("last_updated", ""))
|
|
PY
|
|
)
|
|
prev_ip=$(python3 - "$before_file" <<'PY'
|
|
import json, sys
|
|
with open(sys.argv[1]) as handle:
|
|
node = json.load(handle)
|
|
print(node["meta_data"].get("ip", ""))
|
|
PY
|
|
)
|
|
initial_ip=$(cat "$TMP_ROOT/initial_ip")
|
|
if [[ "$prev_ip" != "$initial_ip" ]]; then
|
|
echo "[ERROR] Expected initial IP $initial_ip, got $prev_ip" >&2
|
|
exit 1
|
|
fi
|
|
|
|
pushd "$TEST_ROOT" >/dev/null
|
|
compose rm -sf agent
|
|
popd >/dev/null
|
|
|
|
docker container rm -f argus-agent-e2e >/dev/null 2>&1 || true
|
|
|
|
AGENT_DIR="$TEST_ROOT/private/argus/agent/$AGENT_HOSTNAME"
|
|
HEALTH_DIR="$TEST_ROOT/private/argus/agent/health/$AGENT_HOSTNAME"
|
|
|
|
# 先以 sleep 方式启动容器，确保我们掌握注册时的网络状态
|
|
if ! docker run -d \
|
|
--name argus-agent-e2e \
|
|
--hostname "$AGENT_HOSTNAME" \
|
|
--network "$NETWORK_NAME" \
|
|
--ip "$NEW_AGENT_IP" \
|
|
-v "$AGENT_DIR:/private/argus/agent/$AGENT_HOSTNAME" \
|
|
-v "$HEALTH_DIR:/private/argus/agent/health/$AGENT_HOSTNAME" \
|
|
argus-agent:dev \
|
|
sleep 300 >/dev/null; then
|
|
echo "[ERROR] Failed to start agent container with custom IP" >&2
|
|
exit 1
|
|
fi
|
|
|
|
# 在容器内启动真实 agent 进程
|
|
if ! docker exec -d argus-agent-e2e python -m app.main; then
|
|
echo "[ERROR] Failed to spawn agent process inside container" >&2
|
|
exit 1
|
|
fi
|
|
|
|
success=false
|
|
detail_file="$TMP_ROOT/post_restart.json"
|
|
for _ in {1..20}; do
|
|
sleep 3
|
|
if ! curl -sS "$API_BASE/nodes/$NODE_ID" -o "$detail_file"; then
|
|
continue
|
|
fi
|
|
if python3 - "$detail_file" "$prev_last_updated" "$NODE_ID" "$prev_ip" "$NEW_AGENT_IP" <<'PY'
|
|
import json, sys
|
|
with open(sys.argv[1]) as handle:
|
|
node = json.load(handle)
|
|
prev_last_updated = sys.argv[2]
|
|
expected_id = sys.argv[3]
|
|
old_ip = sys.argv[4]
|
|
expected_ip = sys.argv[5]
|
|
last_updated = node.get("last_updated")
|
|
current_ip = node["meta_data"].get("ip")
|
|
assert node["id"] == expected_id
|
|
if current_ip != expected_ip:
|
|
raise SystemExit(1)
|
|
if current_ip == old_ip:
|
|
raise SystemExit(1)
|
|
if not last_updated or last_updated == prev_last_updated:
|
|
raise SystemExit(1)
|
|
PY
|
|
then
|
|
success=true
|
|
break
|
|
fi
|
|
done
|
|
|
|
if [[ "$success" != true ]]; then
|
|
echo "[ERROR] Agent did not report expected new IP $NEW_AGENT_IP after restart" >&2
|
|
exit 1
|
|
fi
|
|
|
|
echo "[INFO] Agent restart produced successful re-registration with IP change"
|