ddadmin
b34eb6ac4c
Добавлены независимые пользовательские профили, их выбор в Bash-командах и явная установка службы systemd. Инструкции размещены в README пакета. Проверка: 67 тестов на Python 3.8 и 3.14, shell-наборы и интеграция Bash с CLI. Загрузка ОС и восстановление настоящей службы требуют отдельной проверки.
258 lines
10 KiB
Bash
Executable File
258 lines
10 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
# Тесты Bash-функций proxy-client: вместо настоящего CLI — поддельный
|
|
# исполняемый файл в PROXY_CLIENT_BIN.
|
|
# Строчные proxy-переменные задаёт proxy_on, shellcheck этого не видит.
|
|
# shellcheck disable=SC2154
|
|
set -euo pipefail
|
|
|
|
REPO_ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
|
|
FUNCTIONS_FILE="$REPO_ROOT/bash/.bash_functions"
|
|
CUSTOM_FILE="$REPO_ROOT/bash/.bash_custom"
|
|
SHELL_MODULE="$REPO_ROOT/proxy-client/.config/proxy-client/shell.bash"
|
|
TEST_ROOT="$(mktemp -d /tmp/proxy-launchers-test.XXXXXX)"
|
|
trap 'rm -rf "$TEST_ROOT"' EXIT
|
|
|
|
PROXY_KEYS=(HTTP_PROXY HTTPS_PROXY ALL_PROXY NO_PROXY
|
|
http_proxy https_proxy all_proxy no_proxy)
|
|
|
|
fail() {
|
|
echo "FAIL: $*" >&2
|
|
exit 1
|
|
}
|
|
|
|
ok() {
|
|
echo "ok - $*"
|
|
}
|
|
|
|
assert_eq() {
|
|
[ "$1" = "$2" ] || fail "$3 (ожидалось '$1', получено '$2')"
|
|
}
|
|
|
|
# Поддельный CLI: на "env" печатает $FAKE_ENV_OUTPUT, на остальное — свои
|
|
# аргументы по одному в строке; завершается с кодом $FAKE_RC.
|
|
PROXY_CLIENT_BIN="$TEST_ROOT/proxy-client"
|
|
cat > "$PROXY_CLIENT_BIN" <<'EOF'
|
|
#!/usr/bin/env bash
|
|
if [ -n "${FAKE_CALL_LOG:-}" ]; then
|
|
printf '<%s>\n' "$@" >> "$FAKE_CALL_LOG"
|
|
fi
|
|
if [ "$1" = env ]; then
|
|
printf '%s\n' "$FAKE_ENV_OUTPUT"
|
|
elif [ "$#" -eq 3 ] && [ "$1" = --profile ] && [ "$2" = work ] && [ "$3" = env ]; then
|
|
printf '%s\n' "$FAKE_PROFILE_ENV_OUTPUT"
|
|
else
|
|
printf '<%s>\n' "$@"
|
|
fi
|
|
exit "${FAKE_RC:-0}"
|
|
EOF
|
|
chmod +x "$PROXY_CLIENT_BIN"
|
|
export PROXY_CLIENT_BIN
|
|
|
|
VALID_ENV='HTTP_PROXY=http://proxy.test:1234
|
|
HTTPS_PROXY=http://proxy.test:1234
|
|
ALL_PROXY=socks5://proxy.test:1234
|
|
NO_PROXY=proxy.test,localhost,.example.test
|
|
http_proxy=http://proxy.test:1234
|
|
https_proxy=http://proxy.test:1234
|
|
all_proxy=socks5://proxy.test:1234
|
|
no_proxy=proxy.test,localhost,.example.test'
|
|
|
|
# shellcheck source=/dev/null
|
|
source "$SHELL_MODULE"
|
|
|
|
# proxy_on применяет все восемь переменных.
|
|
(
|
|
unset "${PROXY_KEYS[@]}"
|
|
FAKE_ENV_OUTPUT="$VALID_ENV" proxy_on >/dev/null
|
|
assert_eq http://proxy.test:1234 "$HTTP_PROXY" HTTP_PROXY
|
|
assert_eq http://proxy.test:1234 "$https_proxy" https_proxy
|
|
assert_eq socks5://proxy.test:1234 "$ALL_PROXY" ALL_PROXY
|
|
assert_eq proxy.test,localhost,.example.test "$no_proxy" no_proxy
|
|
for key in "${PROXY_KEYS[@]}"; do
|
|
[ -n "$(printenv "$key")" ] || fail "$key не экспортирован"
|
|
done
|
|
)
|
|
ok "proxy_on применяет восемь переменных"
|
|
|
|
# Выбор профиля меняет окружение только текущей оболочки.
|
|
(
|
|
unset "${PROXY_KEYS[@]}"
|
|
FAKE_ENV_OUTPUT="$VALID_ENV" \
|
|
FAKE_PROFILE_ENV_OUTPUT='HTTP_PROXY=http://work.test:4321
|
|
HTTPS_PROXY=http://work.test:4321
|
|
ALL_PROXY=socks5://work.test:4321
|
|
NO_PROXY=work.test,localhost
|
|
http_proxy=http://work.test:4321
|
|
https_proxy=http://work.test:4321
|
|
all_proxy=socks5://work.test:4321
|
|
no_proxy=work.test,localhost' \
|
|
proxy_on --profile work >/dev/null
|
|
assert_eq http://work.test:4321 "$HTTP_PROXY" "proxy_on не выбрал профиль work"
|
|
assert_eq socks5://work.test:4321 "$all_proxy" "SOCKS выбранного профиля"
|
|
assert_eq work.test,localhost "$NO_PROXY" "исключения выбранного профиля"
|
|
)
|
|
ok "proxy_on выбирает именованный профиль"
|
|
|
|
# Неверный синтаксис proxy_on отклоняется до изменения окружения.
|
|
assert_invalid_proxy_on() (
|
|
for key in "${PROXY_KEYS[@]}"; do export "$key=old"; done
|
|
local rc=0
|
|
FAKE_ENV_OUTPUT="$VALID_ENV" proxy_on "$@" >/dev/null 2>&1 || rc=$?
|
|
assert_eq 2 "$rc" "proxy_on принял неверные аргументы: $*"
|
|
for key in "${PROXY_KEYS[@]}"; do
|
|
assert_eq old "${!key}" "$key изменился после неверных аргументов"
|
|
done
|
|
)
|
|
assert_invalid_proxy_on env
|
|
assert_invalid_proxy_on --profile
|
|
assert_invalid_proxy_on --profile ""
|
|
assert_invalid_proxy_on --profile work extra
|
|
assert_invalid_proxy_on --unknown work
|
|
ok "proxy_on отклоняет неверные аргументы без изменения окружения"
|
|
|
|
# Ошибка CLI или непонятный вывод не меняют окружение ни на одну переменную.
|
|
for case_spec in "1|$VALID_ENV" "0|$VALID_ENV
|
|
EXTRA_PROXY=bad" "0|BROKEN-LINE"; do
|
|
(
|
|
for key in "${PROXY_KEYS[@]}"; do export "$key=old"; done
|
|
if FAKE_RC="${case_spec%%|*}" FAKE_ENV_OUTPUT="${case_spec#*|}" \
|
|
proxy_on >/dev/null 2>&1; then
|
|
fail "proxy_on принял ошибочный ответ: ${case_spec#*|}"
|
|
fi
|
|
for key in "${PROXY_KEYS[@]}"; do
|
|
assert_eq old "${!key}" "$key изменился после ошибки"
|
|
done
|
|
)
|
|
done
|
|
ok "неуспешный env не меняет окружение"
|
|
|
|
# Даже пригодный вывод не применяется, если проверка выбранного прокси не прошла.
|
|
(
|
|
for key in "${PROXY_KEYS[@]}"; do export "$key=old"; done
|
|
unset no_proxy
|
|
if FAKE_RC=1 FAKE_PROFILE_ENV_OUTPUT="$VALID_ENV" proxy_on --profile work >/dev/null 2>&1; then
|
|
fail "proxy_on принял отказ проверки выбранного профиля"
|
|
fi
|
|
for key in "${PROXY_KEYS[@]}"; do
|
|
if [[ "$key" == no_proxy ]]; then
|
|
[ -z "${!key+set}" ] || fail "$key появился после ошибки"
|
|
else
|
|
assert_eq old "${!key}" "$key изменился после ошибки выбранного профиля"
|
|
fi
|
|
done
|
|
)
|
|
ok "отказ проверки выбранного профиля сохраняет окружение"
|
|
|
|
# proxy_off очищает оба регистра.
|
|
(
|
|
for key in "${PROXY_KEYS[@]}"; do export "$key=x"; done
|
|
FAKE_CALL_LOG="$TEST_ROOT/off-calls" proxy_off >/dev/null
|
|
[ ! -e "$TEST_ROOT/off-calls" ] || fail "proxy_off вызвал CLI"
|
|
for key in "${PROXY_KEYS[@]}"; do
|
|
[ -z "${!key+set}" ] || fail "$key не очищен"
|
|
done
|
|
)
|
|
ok "proxy_off очищает восемь переменных без вызова CLI"
|
|
|
|
# Обёртки передают CLI команду, аргументы и код возврата без искажений.
|
|
(
|
|
export FAKE_RC=42
|
|
set +e
|
|
output="$(pclaude "a b" -- --flag)"
|
|
rc=$?
|
|
set -e
|
|
assert_eq 42 "$rc" "pclaude не сохранил код возврата"
|
|
assert_eq $'<run>\n<-->\n<claude>\n<a b>\n<-->\n<--flag>' "$output" \
|
|
"pclaude исказил аргументы"
|
|
|
|
set +e
|
|
output="$(pcodex --search)"
|
|
rc=$?
|
|
set -e
|
|
assert_eq 42 "$rc" "pcodex не сохранил код возврата"
|
|
assert_eq $'<run>\n<-->\n<codex>\n<--search>' "$output" \
|
|
"pcodex исказил аргументы"
|
|
)
|
|
ok "pclaude и pcodex передают аргументы и код выхода"
|
|
|
|
# Ведущий --proxy-profile относится к прокси, обычный --profile — к программе.
|
|
(
|
|
for wrapper in pclaude pcodex; do
|
|
rc=0
|
|
output="$(FAKE_RC=42 "$wrapper" --proxy-profile work --profile personal "a b" "")" || rc=$?
|
|
assert_eq 42 "$rc" "$wrapper не сохранил код возврата с профилем"
|
|
assert_eq $'<--profile>\n<work>\n<run>\n<-->\n<'"${wrapper#p}"$'>\n<--profile>\n<personal>\n<a b>\n<>' "$output" \
|
|
"$wrapper исказил выбор профиля или аргументы ребенка"
|
|
done
|
|
)
|
|
ok "обёртки выбирают профиль прокси и сохраняют аргументы ребенка"
|
|
|
|
# Отсутствующее имя профиля не должно запускать дочернюю команду.
|
|
(
|
|
for wrapper in pclaude pcodex; do
|
|
for empty_name in missing empty; do
|
|
args=(--proxy-profile)
|
|
if [[ "$empty_name" == empty ]]; then args+=(""); fi
|
|
rc=0
|
|
output="$("$wrapper" "${args[@]}" 2>/dev/null)" || rc=$?
|
|
assert_eq 2 "$rc" "$wrapper принял --proxy-profile без имени"
|
|
assert_eq "" "$output" "$wrapper вызвал CLI без имени профиля"
|
|
done
|
|
done
|
|
)
|
|
ok "обёртки отклоняют --proxy-profile без имени"
|
|
|
|
# Только ведущий --proxy-profile выбирает профиль прокси.
|
|
(
|
|
for wrapper in pclaude pcodex; do
|
|
output="$("$wrapper" --profile personal --proxy-profile child "a b" "")"
|
|
assert_eq $'<run>\n<-->\n<'"${wrapper#p}"$'>\n<--profile>\n<personal>\n<--proxy-profile>\n<child>\n<a b>\n<>' "$output" \
|
|
"$wrapper забрал аргументы ребенка"
|
|
done
|
|
)
|
|
ok "обычный --profile и неведущий --proxy-profile остаются ребенку"
|
|
|
|
# Запуск с профилем не меняет окружение вызывающей оболочки.
|
|
(
|
|
for key in "${PROXY_KEYS[@]}"; do export "$key=old"; done
|
|
unset no_proxy
|
|
for wrapper in pclaude pcodex; do
|
|
"$wrapper" --proxy-profile work >/dev/null
|
|
for key in "${PROXY_KEYS[@]}"; do
|
|
if [[ "$key" == no_proxy ]]; then
|
|
[ -z "${!key+set}" ] || fail "$wrapper экспортировал $key"
|
|
else
|
|
assert_eq old "${!key}" "$wrapper изменил $key"
|
|
fi
|
|
done
|
|
done
|
|
)
|
|
ok "обёртки сохраняют окружение вызывающей оболочки"
|
|
|
|
# .bash_custom подключает и общие функции, и proxy-модуль.
|
|
(
|
|
home="$TEST_ROOT/home"
|
|
mkdir -p "$home/.config/proxy-client"
|
|
ln -s "$CUSTOM_FILE" "$home/.bash_custom"
|
|
ln -s "$FUNCTIONS_FILE" "$home/.bash_functions"
|
|
ln -s "$SHELL_MODULE" "$home/.config/proxy-client/shell.bash"
|
|
output="$(
|
|
HOME="$home" bash --noprofile --norc -ic '
|
|
source "$HOME/.bash_custom"
|
|
declare -F proxy_on update_hproxy >/dev/null && printf ready
|
|
' 2>/dev/null
|
|
)"
|
|
assert_eq ready "$output" ".bash_custom не подключил модули"
|
|
)
|
|
ok ".bash_custom подключает общие функции и proxy-модуль"
|
|
|
|
if command -v shellcheck >/dev/null 2>&1; then
|
|
shellcheck -s bash "$SHELL_MODULE" "$0" \
|
|
|| fail "shellcheck нашёл проблемы"
|
|
ok "shellcheck чист"
|
|
else
|
|
echo "skip - shellcheck не установлен"
|
|
fi
|
|
|
|
echo "PASS: тесты Bash-функций proxy-client"
|