228 lines
8.4 KiB
Bash
228 lines
8.4 KiB
Bash
#!/bin/bash
|
|
|
|
set -euo pipefail
|
|
|
|
source "$(cd -- "$(dirname -- "${BASH_SOURCE[0]}")" && pwd)/base-test.sh"
|
|
|
|
require_command flock
|
|
|
|
tmp=$(mktemp -d)
|
|
trap 'rm -rf "$tmp"' EXIT
|
|
|
|
cache_home="$tmp/cache"
|
|
images="$tmp/images"
|
|
stub_bin="$tmp/bin"
|
|
mkdir -p "$images" "$stub_bin"
|
|
|
|
cat >"$stub_bin/vipsthumbnail" <<'EOF'
|
|
#!/bin/bash
|
|
|
|
image="$1"
|
|
shift
|
|
|
|
while (( $# > 0 )); do
|
|
if [[ $1 == "--path" ]]; then
|
|
output=${2%%\[*}
|
|
break
|
|
fi
|
|
shift
|
|
done
|
|
|
|
if [[ -f ${VIPSTHUMBNAIL_FAIL_FILE:-} ]] && grep -Fxq "$image" "$VIPSTHUMBNAIL_FAIL_FILE"; then
|
|
exit 1
|
|
fi
|
|
|
|
[[ -z ${VIPSTHUMBNAIL_CALLS_FILE:-} ]] || printf '%s\n' "$image" >>"$VIPSTHUMBNAIL_CALLS_FILE"
|
|
[[ -z ${VIPSTHUMBNAIL_DELAY:-} ]] || sleep "$VIPSTHUMBNAIL_DELAY"
|
|
printf 'thumbnail' >"$output"
|
|
EOF
|
|
chmod +x "$stub_bin/vipsthumbnail"
|
|
|
|
for name in one two three; do
|
|
printf 'image-%s' "$name" >"$images/$name.png"
|
|
done
|
|
|
|
cache_dir="$cache_home/omarchy/image-selector"
|
|
mkdir -p "$cache_dir"
|
|
|
|
stale_tmp=""
|
|
live_lock=""
|
|
for image in "$images"/*; do
|
|
signature=$(stat -Lc '%s:%Y' "$image")
|
|
hash=$(printf '%s\t%s' "$image" "$signature" | md5sum | cut -d ' ' -f 1)
|
|
mkdir "$cache_dir/$hash.jpg.lock"
|
|
touch -m -d '10 minutes ago' "$cache_dir/$hash.jpg.lock"
|
|
stale_tmp="$cache_dir/$hash.jpg.4242.jpg"
|
|
live_lock="$cache_dir/$hash.jpg.lock"
|
|
done
|
|
printf 'partial' >"$stale_tmp"
|
|
|
|
cache_key=$(printf '%s' "$images" | md5sum | cut -d ' ' -f 1)
|
|
printf '%s\t%s' "$images/one.png" "$cache_dir/missing.jpg" >"$cache_dir/$cache_key.rows"
|
|
printf 'v2\n%s:%s\n' "$images" "$(stat -Lc '%Y' "$images")" >"$cache_dir/$cache_key.signature"
|
|
printf 'v1\n%s:%s\n' "$images" "$(stat -Lc '%Y' "$images")" >"$cache_dir/$cache_key.fast-signature"
|
|
|
|
PATH="$stub_bin:$PATH" XDG_CACHE_HOME="$cache_home" \
|
|
"$ROOT/bin/omarchy-menu-images" --cache-only "$images"
|
|
|
|
(( $(find "$cache_dir" -maxdepth 1 -name '*.jpg' -type f | wc -l) == 3 )) ||
|
|
fail "image menu recovers thumbnails from stranded locks"
|
|
(( $(awk 'END { print NR }' "$cache_dir/$cache_key.rows") == 3 )) ||
|
|
fail "image menu rebuilds every row after cache invalidation"
|
|
[[ $(head -n 1 "$cache_dir/$cache_key.signature") == "v4" ]] ||
|
|
fail "image menu invalidates stale row caches"
|
|
[[ ! -e $stale_tmp ]] ||
|
|
fail "image menu clears partial thumbnails left by killed generators"
|
|
pass "image menu recovers stranded locks and stale rows"
|
|
|
|
rm -rf "$cache_home"
|
|
mkdir -p "$cache_dir"
|
|
mkdir "$live_lock"
|
|
|
|
PATH="$stub_bin:$PATH" XDG_CACHE_HOME="$cache_home" \
|
|
"$ROOT/bin/omarchy-menu-images" --cache-only "$images"
|
|
|
|
(( $(find "$cache_dir" -maxdepth 1 -name '*.jpg' -type f | wc -l) == 2 )) ||
|
|
fail "image menu skips a thumbnail whose fresh legacy lock may still be owned"
|
|
[[ -d $live_lock ]] ||
|
|
fail "image menu leaves a fresh legacy lock directory alone"
|
|
[[ ! -e $cache_dir/$cache_key.rows ]] ||
|
|
fail "image menu does not cache rows while a legacy generator holds a lock"
|
|
pass "image menu respects a live legacy generator's lock"
|
|
|
|
rm -rf "$cache_home"
|
|
mkdir -p "$cache_home"
|
|
printf '%s\n' "$images/two.png" >"$tmp/failures"
|
|
|
|
PATH="$stub_bin:$PATH" XDG_CACHE_HOME="$cache_home" VIPSTHUMBNAIL_FAIL_FILE="$tmp/failures" \
|
|
"$ROOT/bin/omarchy-menu-images" --cache-only "$images"
|
|
|
|
cache_dir="$cache_home/omarchy/image-selector"
|
|
[[ ! -e $cache_dir/$cache_key.rows ]] || fail "image menu does not cache incomplete rows"
|
|
[[ ! -e $cache_dir/$cache_key.signature ]] || fail "image menu does not sign incomplete rows"
|
|
[[ ! -e $cache_dir/$cache_key.fast-signature ]] || fail "image menu does not fast-cache incomplete rows"
|
|
pass "image menu leaves failed thumbnail batches uncached"
|
|
|
|
rm "$tmp/failures"
|
|
PATH="$stub_bin:$PATH" XDG_CACHE_HOME="$cache_home" \
|
|
"$ROOT/bin/omarchy-menu-images" --cache-only "$images"
|
|
|
|
(( $(find "$cache_dir" -maxdepth 1 -name '*.jpg' -type f | wc -l) == 3 )) ||
|
|
fail "image menu retries a previously failed thumbnail"
|
|
(( $(awk 'END { print NR }' "$cache_dir/$cache_key.rows") == 3 )) ||
|
|
fail "image menu caches every row after retry"
|
|
pass "image menu completes and caches a later retry"
|
|
|
|
rm -rf "$cache_home"
|
|
mkdir -p "$cache_home"
|
|
: >"$tmp/calls"
|
|
|
|
# The delay keeps both runs inside the generation window so the locks are
|
|
# actually contended rather than the second run arriving after the first.
|
|
pids=()
|
|
for run in 1 2; do
|
|
PATH="$stub_bin:$PATH" XDG_CACHE_HOME="$cache_home" \
|
|
VIPSTHUMBNAIL_CALLS_FILE="$tmp/calls" VIPSTHUMBNAIL_DELAY=0.25 \
|
|
"$ROOT/bin/omarchy-menu-images" --cache-only "$images" &
|
|
pids+=($!)
|
|
done
|
|
for pid in "${pids[@]}"; do
|
|
wait "$pid" || fail "concurrent image menu runs exit cleanly"
|
|
done
|
|
|
|
(( $(wc -l <"$tmp/calls") == 3 )) || fail "image menu serializes concurrent thumbnail generators"
|
|
|
|
rm -f "$cache_dir"/*.jpg
|
|
rm -f "$cache_dir/$cache_key.rows" "$cache_dir/$cache_key.signature" "$cache_dir/$cache_key.fast-signature"
|
|
PATH="$stub_bin:$PATH" XDG_CACHE_HOME="$cache_home" VIPSTHUMBNAIL_CALLS_FILE="$tmp/calls" \
|
|
"$ROOT/bin/omarchy-menu-images" --cache-only "$images"
|
|
|
|
(( $(wc -l <"$tmp/calls") == 6 )) || fail "image menu releases thumbnail locks after generation"
|
|
pass "image menu owns locks for exactly one generator lifetime"
|
|
|
|
rm -rf "$cache_home"
|
|
mkdir -p "$cache_home"
|
|
rows=$(PATH="$stub_bin:$PATH" XDG_CACHE_HOME="$cache_home" \
|
|
"$ROOT/bin/omarchy-menu-images" --print-rows "$images")
|
|
|
|
(( $(wc -l <<<"$rows") == 3 )) || fail "image menu prints one row per image"
|
|
while IFS=$'\t' read -r row_image row_thumbnail; do
|
|
[[ $row_image == "$images"/* && -f $row_thumbnail ]] ||
|
|
fail "image menu prints each image with its generated thumbnail"
|
|
done <<<"$rows"
|
|
pass "image menu prints its rows for the shell to hold"
|
|
|
|
# Block converters behind a gate: printing lazy rows must neither await them
|
|
# nor start one process per image. Repeated refreshes share one worker pool.
|
|
lazy_images="$tmp/lazy-images"
|
|
lazy_state="$tmp/lazy-state"
|
|
mkdir -p "$lazy_images" "$lazy_state"
|
|
for (( i = 0; i < 40; i++ )); do
|
|
printf 'image' >"$lazy_images/$i.png"
|
|
done
|
|
printf '0\n' >"$lazy_state/active"
|
|
printf '0\n' >"$lazy_state/peak"
|
|
cat >"$stub_bin/nproc" <<'EOF'
|
|
#!/bin/bash
|
|
echo "${FAKE_CORES:-2}"
|
|
EOF
|
|
cat >"$stub_bin/vipsthumbnail" <<'EOF'
|
|
#!/bin/bash
|
|
while (( $# > 0 )); do
|
|
if [[ $1 == "--path" ]]; then output=${2%%\[*}; break; fi
|
|
shift
|
|
done
|
|
exec 9>"$LAZY_STATE/lock"
|
|
priority=$(ps -o ni= -p "$$")
|
|
(( priority >= 10 )) || : >"$LAZY_STATE/priority-failed"
|
|
[[ $(ionice -p "$$") == "idle" ]] || : >"$LAZY_STATE/priority-failed"
|
|
flock 9
|
|
active=$(<"$LAZY_STATE/active")
|
|
active=$((active + 1))
|
|
printf '%s\n' "$active" >"$LAZY_STATE/active"
|
|
(( active <= $(<"$LAZY_STATE/peak") )) || printf '%s\n' "$active" >"$LAZY_STATE/peak"
|
|
flock -u 9
|
|
while [[ ! -f $LAZY_STATE/gate ]]; do sleep 0.02; done
|
|
printf 'thumbnail' >"$output"
|
|
flock 9
|
|
active=$(<"$LAZY_STATE/active")
|
|
printf '%s\n' "$((active - 1))" >"$LAZY_STATE/active"
|
|
echo done >>"$LAZY_STATE/completed"
|
|
EOF
|
|
chmod +x "$stub_bin/nproc" "$stub_bin/vipsthumbnail"
|
|
# Also release the gate on failure, so background fixtures cannot outlive us.
|
|
trap 'touch "$lazy_state/gate"' EXIT
|
|
|
|
for cores in 1 2 8; do
|
|
lazy_state="$tmp/lazy-state-$cores"
|
|
mkdir -p "$lazy_state"
|
|
printf '0\n' >"$lazy_state/active"
|
|
printf '0\n' >"$lazy_state/peak"
|
|
expected_workers=1
|
|
(( cores < 4 )) || expected_workers=2
|
|
for run in 1 2; do
|
|
rows=$(PATH="$stub_bin:$PATH" XDG_CACHE_HOME="$tmp/lazy-cache-$cores" LAZY_STATE="$lazy_state" FAKE_CORES="$cores" \
|
|
timeout 10 "$ROOT/bin/omarchy-menu-images" --lazy-thumbnails --print-rows "$lazy_images")
|
|
(( $(wc -l <<<"$rows") == 40 )) || fail "lazy image menu returns all rows before conversion"
|
|
done
|
|
for attempt in {1..100}; do
|
|
(( $(<"$lazy_state/active") == expected_workers )) && break
|
|
sleep 0.02
|
|
done
|
|
(( $(<"$lazy_state/active") == expected_workers && $(<"$lazy_state/peak") == expected_workers )) ||
|
|
fail "lazy image menu bounds repeated refreshes to $expected_workers workers on $cores cores"
|
|
[[ ! -e $lazy_state/completed ]] || fail "lazy image menu does not wait for conversion"
|
|
[[ ! -e $lazy_state/priority-failed ]] || fail "lazy image menu reserves CPU and I/O priority for the UI"
|
|
pass "lazy image menu opens with at most $expected_workers workers on $cores cores"
|
|
|
|
touch "$lazy_state/gate"
|
|
for attempt in {1..500}; do
|
|
if [[ -f $lazy_state/completed ]] && (( $(wc -l <"$lazy_state/completed") == 40 )); then break; fi
|
|
sleep 0.02
|
|
done
|
|
(( $(wc -l <"$lazy_state/completed") == 40 && $(<"$lazy_state/peak") == expected_workers )) ||
|
|
fail "lazy image menu completes the queue after its parent and queue path are gone"
|
|
pass "lazy image menu workers finish every queued thumbnail after the caller exits"
|
|
done
|
|
trap 'rm -rf "$tmp"' EXIT
|