mirror of
https://github.com/dredx/prole.git
synced 2026-09-23 11:03:59 +00:00
81 lines
2.1 KiB
Bash
81 lines
2.1 KiB
Bash
#!/usr/bin/env bash
|
|
# Unit test for etc/init_k3s_registry.sh
|
|
|
|
SCRIPT_DIR=$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)
|
|
KNOE_HOME=$(cd "$SCRIPT_DIR/../.." && pwd)
|
|
ETC_DIR="$KNOE_HOME/etc"
|
|
SCRIPT_UNDER_TEST="$ETC_DIR/init_k3s_registry.sh"
|
|
|
|
TMP_DIR=$(mktemp -d)
|
|
trap 'rm -rf "$TMP_DIR"' EXIT
|
|
|
|
# Mock tools
|
|
mock_tool() {
|
|
cat <<M_EOF > "$TMP_DIR/$1"
|
|
#!/usr/bin/env bash
|
|
echo "Mocked $1 called with \$@" >> "$TMP_DIR/mock_calls.log"
|
|
exit 0
|
|
M_EOF
|
|
chmod +x "$TMP_DIR/$1"
|
|
}
|
|
|
|
# Create mocks for common tools
|
|
mock_tool kubectl
|
|
mock_tool curl
|
|
mock_tool docker
|
|
mock_tool k3d
|
|
mock_tool ansible-playbook
|
|
mock_tool tofu
|
|
mock_tool terraform
|
|
mock_tool ollama
|
|
mock_tool jq
|
|
|
|
# Mock knoe.cfg
|
|
mkdir -p "$TMP_DIR/conf"
|
|
cat <<C_EOF > "$TMP_DIR/conf/knoe.cfg"
|
|
[globals]
|
|
knoe.home = $KNOE_HOME
|
|
knoe.mode = k3d
|
|
[k3s]
|
|
server = https://localhost:6443
|
|
token = test-token
|
|
C_EOF
|
|
|
|
export PATH="$TMP_DIR:$PATH"
|
|
export KNOE_HOME="$KNOE_HOME"
|
|
export KNOE_CONF="$TMP_DIR/conf"
|
|
export NAMESPACE="test-ns"
|
|
export PROLE_PASSWD="test-password"
|
|
|
|
# Run the script
|
|
if [[ "init_k3s_registry.sh" == *.py ]]; then
|
|
if [[ "init_k3s_registry.sh" == "render_manifest.py" ]]; then
|
|
echo "apiVersion: v1" > "$TMP_DIR/test.yaml"
|
|
python3 "$SCRIPT_UNDER_TEST" "$TMP_DIR/test.yaml" > "$TMP_DIR/stdout" 2> "$TMP_DIR/stderr"
|
|
else
|
|
python3 "$SCRIPT_UNDER_TEST" --help > "$TMP_DIR/stdout" 2> "$TMP_DIR/stderr"
|
|
fi
|
|
RC=$?
|
|
else
|
|
# Check for usage() or if it's a script that likely needs arguments
|
|
if grep -q "usage()" "$SCRIPT_UNDER_TEST"; then
|
|
bash "$SCRIPT_UNDER_TEST" --help > "$TMP_DIR/stdout" 2> "$TMP_DIR/stderr"
|
|
RC=$?
|
|
else
|
|
# Try a few common non-destructive actions
|
|
bash "$SCRIPT_UNDER_TEST" status > "$TMP_DIR/stdout" 2> "$TMP_DIR/stderr" || \
|
|
bash "$SCRIPT_UNDER_TEST" --help > "$TMP_DIR/stdout" 2> "$TMP_DIR/stderr" || \
|
|
bash "$SCRIPT_UNDER_TEST" > "$TMP_DIR/stdout" 2> "$TMP_DIR/stderr"
|
|
RC=$?
|
|
fi
|
|
fi
|
|
|
|
if [[ $RC -eq 0 || $RC -eq 1 || $RC -eq 2 ]]; then
|
|
echo "SUCCESS (RC=$RC)"
|
|
exit 0
|
|
else
|
|
echo "FAILURE with RC $RC"
|
|
cat "$TMP_DIR/stderr"
|
|
exit 1
|
|
fi
|