9082357414
Stubbing uname alone is not enough on a WSL host: the script's 'grep -qi microsoft /proc/version' probe still fires and the Linux test cases would run the real WSL cleanup against the host's /mnt/* shortcuts and /etc profile. Add a PATH-stubbed grep that fails only the /proc/version probe (delegating everything else to the real grep via an absolute path so the stub can never self-exec), plus no-op powershell.exe and sudo stubs as defense in depth.
129 lines
6.2 KiB
Bash
Executable file
129 lines
6.2 KiB
Bash
Executable file
#!/bin/bash
|
|
# SPDX-License-Identifier: AGPL-3.0-only
|
|
# Copyright 2026-present the Unsloth AI Inc. team. All rights reserved. See /studio/LICENSE.AGPL-3.0
|
|
# Regression tests for WebView runtime-data cleanup in scripts/uninstall.sh.
|
|
#
|
|
# WKWebView (macOS) and webkit2gtk (Linux) create data keyed by the bundle id
|
|
# at first app launch, not at install time, so the uninstaller used to miss it
|
|
# and a leftover cache served a stale frontend to the next install. Runs the
|
|
# full script against a fixture HOME (pkill/defaults stubbed via PATH, OS
|
|
# branch picked by a stubbed uname, /proc/version WSL probe force-failed) and
|
|
# asserts bundle-id paths are removed while unrelated app data survives.
|
|
set -e
|
|
|
|
SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
|
|
UNINSTALL_SH="$SCRIPT_DIR/../../scripts/uninstall.sh"
|
|
BID="ai.unsloth.studio"
|
|
PASS=0
|
|
FAIL=0
|
|
|
|
_TMP_ROOT=$(mktemp -d)
|
|
trap 'rm -rf "$_TMP_ROOT"' EXIT
|
|
|
|
assert_gone() { _l="$1"; if [ -e "$2" ]; then echo " FAIL: $_l (still present: $2)"; FAIL=$((FAIL+1)); else echo " PASS: $_l"; PASS=$((PASS+1)); fi; }
|
|
assert_present() { _l="$1"; if [ -e "$2" ]; then echo " PASS: $_l"; PASS=$((PASS+1)); else echo " FAIL: $_l (missing: $2)"; FAIL=$((FAIL+1)); fi; }
|
|
|
|
# Stub out process kills and macOS pref/LaunchServices tools so the script can
|
|
# run against a fixture HOME without touching the real system.
|
|
STUB_BIN="$_TMP_ROOT/stubbin"
|
|
mkdir -p "$STUB_BIN"
|
|
for _tool in pkill defaults; do
|
|
printf '#!/bin/sh\nexit 0\n' > "$STUB_BIN/$_tool"
|
|
chmod +x "$STUB_BIN/$_tool"
|
|
done
|
|
# Force the non-WSL path: with uname stubbed to Linux on a WSL host, the
|
|
# script's `grep -qi microsoft /proc/version` probe would still fire and the
|
|
# real WSL cleanup would touch the host's /mnt/* shortcuts and /etc profile.
|
|
# Fail that one probe; delegate every other grep call to the real grep.
|
|
# REAL_GREP must be an absolute path: a bare "grep" (e.g. from an alias-shaped
|
|
# `command -v` result) would resolve back to this stub and self-exec forever.
|
|
REAL_GREP=$(command -v grep)
|
|
case "$REAL_GREP" in /*) ;; *) REAL_GREP=/usr/bin/grep ;; esac
|
|
cat > "$STUB_BIN/grep" <<EOF
|
|
#!/bin/sh
|
|
for _a in "\$@"; do
|
|
[ "\$_a" = "/proc/version" ] && exit 1
|
|
done
|
|
exec "$REAL_GREP" "\$@"
|
|
EOF
|
|
chmod +x "$STUB_BIN/grep"
|
|
# Belt and braces should the WSL branch ever be entered anyway: powershell.exe
|
|
# exiting 0 makes its no-op path taken (and skips the /mnt/* drvfs fallback);
|
|
# sudo exiting 0 without running its argv keeps /etc untouched.
|
|
for _tool in powershell.exe sudo; do
|
|
printf '#!/bin/sh\nexit 0\n' > "$STUB_BIN/$_tool"
|
|
chmod +x "$STUB_BIN/$_tool"
|
|
done
|
|
|
|
# run_uninstall <home> <uname_output> : run the full script with a stubbed OS.
|
|
run_uninstall() {
|
|
printf '#!/bin/sh\necho %s\n' "$2" > "$STUB_BIN/uname"
|
|
chmod +x "$STUB_BIN/uname"
|
|
env -u UNSLOTH_STUDIO_HOME -u STUDIO_HOME \
|
|
-u XDG_CACHE_HOME -u XDG_DATA_HOME -u XDG_CONFIG_HOME -u XDG_STATE_HOME \
|
|
HOME="$1" PATH="$STUB_BIN:$PATH" sh "$UNINSTALL_SH" >/dev/null 2>&1
|
|
}
|
|
|
|
# ── 1. macOS: every bundle-id-keyed ~/Library path is removed ──
|
|
H=$(mktemp -d -p "$_TMP_ROOT")
|
|
mkdir -p "$H/Library/Caches/$BID/WebKit/NetworkCache" \
|
|
"$H/Library/WebKit/$BID/WebsiteData/CacheStorage" \
|
|
"$H/Library/Application Support/$BID" \
|
|
"$H/Library/HTTPStorages/$BID" \
|
|
"$H/Library/Saved Application State/$BID.savedState" \
|
|
"$H/Library/Preferences" \
|
|
"$H/Library/Cookies" \
|
|
"$H/Library/Caches/com.other.app"
|
|
: > "$H/Library/HTTPStorages/$BID.binarycookies"
|
|
: > "$H/Library/Cookies/$BID.binarycookies"
|
|
: > "$H/Library/Preferences/$BID.plist"
|
|
: > "$H/Library/Caches/$BID/stale-frontend.js"
|
|
: > "$H/Library/Caches/com.other.app/keepme"
|
|
run_uninstall "$H" Darwin
|
|
assert_gone "macOS: Caches/$BID removed" "$H/Library/Caches/$BID"
|
|
assert_gone "macOS: WebKit/$BID removed" "$H/Library/WebKit/$BID"
|
|
assert_gone "macOS: Application Support/$BID removed" "$H/Library/Application Support/$BID"
|
|
assert_gone "macOS: HTTPStorages/$BID removed" "$H/Library/HTTPStorages/$BID"
|
|
assert_gone "macOS: HTTPStorages/$BID.binarycookies removed" "$H/Library/HTTPStorages/$BID.binarycookies"
|
|
assert_gone "macOS: Cookies/$BID.binarycookies removed" "$H/Library/Cookies/$BID.binarycookies"
|
|
assert_gone "macOS: Saved Application State removed" "$H/Library/Saved Application State/$BID.savedState"
|
|
assert_gone "macOS: Preferences/$BID.plist removed" "$H/Library/Preferences/$BID.plist"
|
|
assert_present "macOS: unrelated app cache kept" "$H/Library/Caches/com.other.app/keepme"
|
|
|
|
# ── 2. Linux: bundle-id-keyed XDG default paths are removed ──
|
|
H=$(mktemp -d -p "$_TMP_ROOT")
|
|
mkdir -p "$H/.cache/$BID" "$H/.local/share/$BID" "$H/.config/$BID" \
|
|
"$H/.local/state/$BID" "$H/.cache/other.app"
|
|
run_uninstall "$H" Linux
|
|
assert_gone "linux: ~/.cache/$BID removed" "$H/.cache/$BID"
|
|
assert_gone "linux: ~/.local/share/$BID removed" "$H/.local/share/$BID"
|
|
assert_gone "linux: ~/.config/$BID removed" "$H/.config/$BID"
|
|
assert_gone "linux: ~/.local/state/$BID removed" "$H/.local/state/$BID"
|
|
assert_present "linux: unrelated app cache kept" "$H/.cache/other.app"
|
|
|
|
# ── 3. Linux: XDG_*_HOME overrides are honored ──
|
|
H=$(mktemp -d -p "$_TMP_ROOT")
|
|
XDG=$(mktemp -d -p "$_TMP_ROOT")
|
|
mkdir -p "$XDG/cache/$BID" "$XDG/data/$BID" "$XDG/config/$BID" "$XDG/state/$BID"
|
|
printf '#!/bin/sh\necho Linux\n' > "$STUB_BIN/uname"
|
|
chmod +x "$STUB_BIN/uname"
|
|
env -u UNSLOTH_STUDIO_HOME -u STUDIO_HOME \
|
|
XDG_CACHE_HOME="$XDG/cache" XDG_DATA_HOME="$XDG/data" \
|
|
XDG_CONFIG_HOME="$XDG/config" XDG_STATE_HOME="$XDG/state" \
|
|
HOME="$H" PATH="$STUB_BIN:$PATH" sh "$UNINSTALL_SH" >/dev/null 2>&1
|
|
assert_gone "linux: XDG_CACHE_HOME override honored" "$XDG/cache/$BID"
|
|
assert_gone "linux: XDG_DATA_HOME override honored" "$XDG/data/$BID"
|
|
assert_gone "linux: XDG_CONFIG_HOME override honored" "$XDG/config/$BID"
|
|
assert_gone "linux: XDG_STATE_HOME override honored" "$XDG/state/$BID"
|
|
|
|
# ── 4. Nothing to remove is a clean no-op (fresh HOME, exit 0) ──
|
|
H=$(mktemp -d -p "$_TMP_ROOT")
|
|
if run_uninstall "$H" Darwin; then
|
|
echo " PASS: empty HOME -> no-op exit 0"; PASS=$((PASS+1))
|
|
else
|
|
echo " FAIL: empty HOME -> nonzero exit"; FAIL=$((FAIL+1))
|
|
fi
|
|
|
|
echo ""
|
|
echo "Results: $PASS passed, $FAIL failed"
|
|
[ "$FAIL" = 0 ]
|