mirror of
https://github.com/dredx/prole.git
synced 2026-09-24 18:24:32 +00:00
283 lines
9.4 KiB
Bash
283 lines
9.4 KiB
Bash
#!/usr/bin/env bash
|
|
# Unit test for etc/init_cnpg_backup.sh
|
|
# Verifies we wait for CNPG webhook endpoints before applying a Backup.
|
|
|
|
set -euo pipefail
|
|
|
|
SCRIPT_DIR=$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)
|
|
KNOE_HOME_REPO=$(cd "$SCRIPT_DIR/../.." && pwd)
|
|
ETC_DIR="$KNOE_HOME_REPO/etc"
|
|
SCRIPT_UNDER_TEST="$ETC_DIR/init_cnpg_backup.sh"
|
|
|
|
TMP_DIR=$(mktemp -d)
|
|
trap 'rm -rf "$TMP_DIR"' EXIT
|
|
|
|
BIN_DIR="$TMP_DIR/bin"
|
|
mkdir -p "$BIN_DIR"
|
|
|
|
cat <<'K_EOF' > "$BIN_DIR/kubectl"
|
|
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
|
|
_log_file="${TMP_DIR}/mock_calls.log"
|
|
echo "Mocked kubectl called with $*" >> "${_log_file}"
|
|
|
|
args="$*"
|
|
|
|
# API server readiness probes
|
|
if [[ "${args}" == *"get --raw=/readyz"* || "${args}" == *"get --raw='/readyz'"* || "${args}" == *"get --raw=\"/readyz\""* ]]; then
|
|
echo "ok"
|
|
exit 0
|
|
fi
|
|
if [[ "${args}" == *"version --short"* ]]; then
|
|
echo "Client Version: v0.0.0"
|
|
echo "Server Version: v0.0.0"
|
|
exit 0
|
|
fi
|
|
|
|
# Namespace existence/creation
|
|
if [[ "${args}" == "get namespace test-ns" ]]; then
|
|
exit 0
|
|
fi
|
|
if [[ "${args}" == "create namespace test-ns" ]]; then
|
|
exit 0
|
|
fi
|
|
|
|
# Cluster existence
|
|
if [[ "${args}" == "get cluster knoe-db -n test-ns" || "${args}" == "get cluster knoe-db -n test-ns"* ]]; then
|
|
exit 0
|
|
fi
|
|
|
|
# Plugin config present (so the script doesn't try to patch the cluster in this unit test)
|
|
if [[ "${args}" == *"-n test-ns"*"get cluster knoe-db"*"-o jsonpath="*".spec.plugins[*].name"* ]]; then
|
|
echo "barman-cloud.cloudnative-pg.io"
|
|
exit 0
|
|
fi
|
|
if [[ "${args}" == *"-n test-ns"*"get cluster knoe-db"*"-o jsonpath="*".spec.plugins"*"barmanObjectName"* ]]; then
|
|
echo "knoe-db-barman-objectstore"
|
|
exit 0
|
|
fi
|
|
if [[ "${args}" == *"get cluster knoe-db"*"-o jsonpath={range .spec.plugins"*"parameters.barmanObjectName"* ]]; then
|
|
echo "knoe-db-barman-objectstore"
|
|
exit 0
|
|
fi
|
|
if [[ "${args}" == *"-n test-ns"*"get cluster knoe-db"*"-o jsonpath="*".spec.plugins"*".enabled"* ]]; then
|
|
echo "true"
|
|
exit 0
|
|
fi
|
|
if [[ "${args}" == *"get cluster knoe-db"*"-o jsonpath={range .spec.plugins"*".enabled"* ]]; then
|
|
echo "true"
|
|
exit 0
|
|
fi
|
|
if [[ "${args}" == *"-n test-ns"*"get cluster knoe-db"*"-o jsonpath="*".spec.plugins"*".isWALArchiver"* ]]; then
|
|
echo "true"
|
|
exit 0
|
|
fi
|
|
if [[ "${args}" == *"get cluster knoe-db"*"-o jsonpath={range .spec.plugins"*".isWALArchiver"* ]]; then
|
|
echo "true"
|
|
exit 0
|
|
fi
|
|
if [[ "${args}" == *"-n test-ns"*"get cluster knoe-db"*"-o jsonpath="*".spec.plugins"* ]]; then
|
|
echo '[{"enabled":true,"isWALArchiver":true,"name":"barman-cloud.cloudnative-pg.io","parameters":{"barmanObjectName":"knoe-db-barman-objectstore"}}]'
|
|
exit 0
|
|
fi
|
|
if [[ "${args}" == *"get cluster knoe-db"*"-o json" ]]; then
|
|
echo '{"spec":{"plugins":[{"enabled":true,"isWALArchiver":true,"name":"barman-cloud.cloudnative-pg.io","parameters":{"barmanObjectName":"knoe-db-barman-objectstore"}}]}}'
|
|
exit 0
|
|
fi
|
|
if [[ "${args}" == *"get cluster knoe-db"*"status.conditions"*"ContinuousArchiving"* ]]; then
|
|
printf 'True\tContinuousArchivingReady\twal archive plugin available\n'
|
|
exit 0
|
|
fi
|
|
|
|
# Barman-cloud deployment ready (required by wait_for_plugin_ready)
|
|
if [[ "${args}" == "get deployment -A -l app.kubernetes.io/name=barman-cloud -o jsonpath="* ]]; then
|
|
printf 'cnpg-system\t1\t1\n'
|
|
exit 0
|
|
fi
|
|
|
|
# Barman plugin service registration (used by wait_for_barman_plugin_infra_ready)
|
|
if [[ "${args}" == *"-n cnpg-system"*"get svc"*"-l cnpg.io/pluginName=barman-cloud.cloudnative-pg.io"*"-o jsonpath="*"metadata.name"* ]]; then
|
|
echo "barman-cloud"
|
|
exit 0
|
|
fi
|
|
if [[ "${args}" == *"-n cnpg-system"*"get svc barman-cloud"*"-o jsonpath="*"metadata.name"* ]]; then
|
|
echo "barman-cloud"
|
|
exit 0
|
|
fi
|
|
if [[ "${args}" == *"-n cnpg-system"*"get svc barman-cloud"*"pluginClientSecret"* ]]; then
|
|
echo "barman-cloud-client-tls"
|
|
exit 0
|
|
fi
|
|
if [[ "${args}" == *"-n cnpg-system"*"get svc barman-cloud"*"pluginServerSecret"* ]]; then
|
|
echo "barman-cloud-server-tls"
|
|
exit 0
|
|
fi
|
|
if [[ "${args}" == *"-n cnpg-system"*"get svc barman-cloud"*"pluginPort"* ]]; then
|
|
echo "9090"
|
|
exit 0
|
|
fi
|
|
if [[ "${args}" == *"-n cnpg-system"*"get endpoints barman-cloud"* ]]; then
|
|
echo "10.42.0.11"
|
|
exit 0
|
|
fi
|
|
|
|
# CNPG cluster pod selection for socket check
|
|
if [[ "${args}" == *"-n test-ns"*"get pods"*"-l cnpg.io/cluster=knoe-db,cnpg.io/instanceRole=primary"*"-o jsonpath={.items[0].metadata.name"* ]]; then
|
|
echo "knoe-db-1"
|
|
exit 0
|
|
fi
|
|
if [[ "${args}" == *"-n test-ns"*"get pods"*"-l cnpg.io/cluster=knoe-db"*"-o jsonpath={.items[0].metadata.name"* ]]; then
|
|
echo "knoe-db-1"
|
|
exit 0
|
|
fi
|
|
|
|
# Plugin socket check succeeds immediately
|
|
if [[ "${args}" == *"-n test-ns"*"exec knoe-db-1 -c postgres -- test -S /plugins/barman-cloud.cloudnative-pg.io"* ]]; then
|
|
exit 0
|
|
fi
|
|
if [[ "${args}" == *"-n test-ns"*"exec knoe-db-1 -c manager -- test -S /plugins/barman-cloud.cloudnative-pg.io"* ]]; then
|
|
exit 0
|
|
fi
|
|
|
|
# CNPG webhook endpoints: return empty twice, then return an IP.
|
|
if [[ "${args}" == *"-n cnpg-system"*"get endpoints"*"cnpg-webhook-service"* ]]; then
|
|
cfile="${TMP_DIR}/endpoints_calls"
|
|
c=0
|
|
[[ -f "$cfile" ]] && c=$(cat "$cfile")
|
|
c=$((c + 1))
|
|
echo "$c" > "$cfile"
|
|
if (( c >= 3 )); then
|
|
echo "10.42.0.10"
|
|
fi
|
|
exit 0
|
|
fi
|
|
|
|
# Capture applied Backup manifest
|
|
if [[ "${args}" == apply*"-n"*"test-ns"*"-f"*"-"* ]]; then
|
|
cat > "${TMP_DIR}/applied.yaml"
|
|
echo "applied"
|
|
exit 0
|
|
fi
|
|
|
|
# ObjectStore readiness checks
|
|
if [[ "${args}" == *"-n test-ns"*"get objectstore"*"knoe-db-barman-objectstore"*"-o jsonpath="*"conditions"*"Ready"* ]]; then
|
|
echo "True"
|
|
exit 0
|
|
fi
|
|
if [[ "${args}" == *"-n test-ns"*"get objectstore"*"knoe-db-barman-objectstore"*"-o jsonpath={.status.phase"* ]]; then
|
|
echo "Ready"
|
|
exit 0
|
|
fi
|
|
if [[ "${args}" == *"get objectstore"*"knoe-db-barman-objectstore"*"-o jsonpath={.spec.configuration.destinationPath}"* ]]; then
|
|
echo "s3://knoe-db-backups/"
|
|
exit 0
|
|
fi
|
|
if [[ "${args}" == *"get objectstore"*"knoe-db-barman-objectstore"*"-o jsonpath={.spec.configuration.endpointURL}"* ]]; then
|
|
echo "http://garage.test-system.svc.cluster.local:3900"
|
|
exit 0
|
|
fi
|
|
if [[ "${args}" == *"get objectstore"*"knoe-db-barman-objectstore"*"-o jsonpath={.spec.retentionPolicy}"* ]]; then
|
|
echo "30d"
|
|
exit 0
|
|
fi
|
|
if [[ "${args}" == *"get objectstore"*"knoe-db-barman-objectstore"*"-o jsonpath={.spec.configuration.s3Credentials.accessKeyId.name}"* ]]; then
|
|
echo "knoe-db-barman-s3"
|
|
exit 0
|
|
fi
|
|
if [[ "${args}" == *"get objectstore"*"knoe-db-barman-objectstore"*"-o jsonpath={.spec.configuration.s3Credentials.accessKeyId.key}"* ]]; then
|
|
echo "ACCESS_KEY_ID"
|
|
exit 0
|
|
fi
|
|
if [[ "${args}" == *"get objectstore"*"knoe-db-barman-objectstore"*"-o jsonpath={.spec.configuration.s3Credentials.secretAccessKey.name}"* ]]; then
|
|
echo "knoe-db-barman-s3"
|
|
exit 0
|
|
fi
|
|
if [[ "${args}" == *"get objectstore"*"knoe-db-barman-objectstore"*"-o jsonpath={.spec.configuration.s3Credentials.secretAccessKey.key}"* ]]; then
|
|
echo "SECRET_ACCESS_KEY"
|
|
exit 0
|
|
fi
|
|
if [[ "${args}" == *"get objectstore"*"knoe-db-barman-objectstore"*"-o jsonpath={.spec.configuration.s3Credentials.region.name}"* ]]; then
|
|
echo "knoe-db-barman-s3"
|
|
exit 0
|
|
fi
|
|
if [[ "${args}" == *"get objectstore"*"knoe-db-barman-objectstore"*"-o jsonpath={.spec.configuration.s3Credentials.region.key}"* ]]; then
|
|
echo "REGION"
|
|
exit 0
|
|
fi
|
|
if [[ "${args}" == *"get objectstore"*"knoe-db-barman-objectstore"*"-o jsonpath={.metadata.generation}"* ]]; then
|
|
echo "1"
|
|
exit 0
|
|
fi
|
|
if [[ "${args}" == *"get objectstore"*"knoe-db-barman-objectstore"*"-o jsonpath={.status.observedGeneration}"* ]]; then
|
|
echo "1"
|
|
exit 0
|
|
fi
|
|
|
|
exit 0
|
|
K_EOF
|
|
chmod +x "$BIN_DIR/kubectl"
|
|
|
|
export PATH="$BIN_DIR:$PATH"
|
|
export TMP_DIR
|
|
|
|
# knoe_cfg.sh sources $KNOE_HOME/env.sh if present; provide a minimal env.sh that points to our temp config.
|
|
CONF_DIR="$TMP_DIR/conf"
|
|
ENV_HOME="$TMP_DIR/env-home"
|
|
mkdir -p "$CONF_DIR" "$ENV_HOME"
|
|
|
|
cat <<C_EOF > "$CONF_DIR/knoe.cfg"
|
|
[User]
|
|
NAMESPACE = test-ns
|
|
SERVICE_NAMESPACE = test-system
|
|
KNOE_HOME = $KNOE_HOME_REPO
|
|
KNOE_SERVICE = $TMP_DIR/service
|
|
|
|
[Global]
|
|
DEPLOYMENT_MODE = k3s
|
|
C_EOF
|
|
|
|
cat <<E_EOF > "$ENV_HOME/env.sh"
|
|
#!/usr/bin/env bash
|
|
export KNOE_HOME="$KNOE_HOME_REPO"
|
|
export KNOE_CONF="$CONF_DIR"
|
|
E_EOF
|
|
chmod +x "$ENV_HOME/env.sh"
|
|
|
|
set +e
|
|
KNOE_HOME="$ENV_HOME" PROLE_NAMESPACE="test-ns" SERVICE_NAMESPACE="test-system" GARAGE_NAMESPACE="test-system" GARAGE_S3_ENDPOINT="http://garage.test-system.svc.cluster.local:3900" bash "$SCRIPT_UNDER_TEST" backup full >"$TMP_DIR/stdout" 2>"$TMP_DIR/stderr"
|
|
RC=$?
|
|
set -e
|
|
|
|
if [[ $RC -ne 0 ]]; then
|
|
echo "FAILURE: init_cnpg_backup.sh returned rc=$RC"
|
|
sed -n '1,200p' "$TMP_DIR/mock_calls.log" || true
|
|
sed -n '1,200p' "$TMP_DIR/stderr" || true
|
|
exit 1
|
|
fi
|
|
|
|
if [[ ! -f "$TMP_DIR/applied.yaml" ]]; then
|
|
echo "FAILURE: expected Backup manifest to be applied"
|
|
sed -n '1,200p' "$TMP_DIR/mock_calls.log" || true
|
|
sed -n '1,200p' "$TMP_DIR/stderr" || true
|
|
exit 1
|
|
fi
|
|
|
|
# Verify webhook endpoints were checked before apply.
|
|
endpoint_line=$(grep -n "get endpoints cnpg-webhook-service" "$TMP_DIR/mock_calls.log" | head -n1 | cut -d: -f1 || true)
|
|
apply_line=$(grep -n "apply -n test-ns -f -" "$TMP_DIR/mock_calls.log" | tail -n1 | cut -d: -f1 || true)
|
|
if [[ -z "$endpoint_line" || -z "$apply_line" || "$endpoint_line" -ge "$apply_line" ]]; then
|
|
echo "FAILURE: expected webhook endpoints check before applying Backup"
|
|
echo "endpoint_line=$endpoint_line apply_line=$apply_line"
|
|
sed -n '1,200p' "$TMP_DIR/mock_calls.log" || true
|
|
exit 1
|
|
fi
|
|
|
|
if ! grep -qE '^kind:\s*Backup\s*$' "$TMP_DIR/applied.yaml"; then
|
|
echo "FAILURE: expected applied manifest to be a Backup"
|
|
sed -n '1,120p' "$TMP_DIR/applied.yaml" || true
|
|
exit 1
|
|
fi
|
|
|
|
echo "SUCCESS"
|