102 lines
3.5 KiB
Bash
Executable File
102 lines
3.5 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
cd "$(dirname "$0")/.."
|
|
|
|
test_name="component-resolution-ttl-test-$$"
|
|
site_directory="${UCE_TEST_SITE_DIRECTORY:-site}"
|
|
worker_count="${UCE_TEST_WORKER_COUNT:-4}"
|
|
if [[ -r /etc/uce/settings.cfg ]]; then
|
|
if [[ -z "${UCE_TEST_SITE_DIRECTORY:-}" ]]; then
|
|
configured_site_directory=$(awk -F= '/^[[:space:]]*SITE_DIRECTORY[[:space:]]*=/ {gsub(/^[[:space:]]+|[[:space:]]+$/, "", $2); print $2; exit}' /etc/uce/settings.cfg)
|
|
site_directory="${configured_site_directory:-$site_directory}"
|
|
fi
|
|
if [[ -z "${UCE_TEST_WORKER_COUNT:-}" ]]; then
|
|
configured_worker_count=$(awk -F= '/^[[:space:]]*WORKER_COUNT[[:space:]]*=/ {gsub(/^[[:space:]]+|[[:space:]]+$/, "", $2); print $2; exit}' /etc/uce/settings.cfg)
|
|
worker_count="${configured_worker_count:-$worker_count}"
|
|
fi
|
|
fi
|
|
http_host="${UCE_TEST_HTTP_HOST:-uce.openfu.com}"
|
|
source_dir="$site_directory/$test_name"
|
|
|
|
cleanup() {
|
|
rm -rf "$source_dir"
|
|
}
|
|
trap cleanup EXIT
|
|
mkdir -p "$source_dir"
|
|
|
|
printf '%s\n' 'RENDER(Request& context) { print("parent:", request_perf()["worker_pid"].to_string(), ":", component("chosen", context)); }' >"$source_dir/parent.uce"
|
|
printf '%s\n' 'COMPONENT(Request& context) { print("a"); }' >"$source_dir/a.uce"
|
|
printf '%s\n' 'COMPONENT(Request& context) { print("b"); }' >"$source_dir/b.uce"
|
|
ln -s a.uce "$source_dir/chosen.uce"
|
|
|
|
declare -A seen
|
|
collect_workers() {
|
|
local method="$1"
|
|
local marker="$2"
|
|
local path="${3:-parent.uce}"
|
|
local deadline=$((SECONDS + 30))
|
|
seen=()
|
|
while (( SECONDS < deadline && ${#seen[@]} < worker_count )); do
|
|
local response status body pid
|
|
response=$(curl -sS --max-time 10 -X "$method" -H "Host: $http_host" -w $'\n%{http_code}' "http://127.0.0.1/$test_name/$path")
|
|
status=${response##*$'\n'}
|
|
body=${response%$'\n'*}
|
|
if [[ "$status" == "503" ]]; then
|
|
sleep 0.1
|
|
continue
|
|
fi
|
|
if [[ "$status" != "200" || "$body" != parent:*":$marker" ]]; then
|
|
echo "component resolution TTL probe failed: method=$method status=$status expected=$marker body=$body" >&2
|
|
exit 1
|
|
fi
|
|
pid=${body#parent:}
|
|
pid=${pid%%:*}
|
|
seen["$pid"]=1
|
|
done
|
|
if (( ${#seen[@]} != worker_count )); then
|
|
echo "component resolution TTL probe reached ${#seen[@]}/$worker_count workers" >&2
|
|
exit 1
|
|
fi
|
|
}
|
|
|
|
collect_missing_workers() {
|
|
local deadline=$((SECONDS + 30))
|
|
seen=()
|
|
while (( SECONDS < deadline && ${#seen[@]} < worker_count )); do
|
|
local response status body pid
|
|
response=$(curl -sS --max-time 10 -H "Host: $http_host" -w $'\n%{http_code}' "http://127.0.0.1/$test_name/missing.uce")
|
|
status=${response##*$'\n'}
|
|
body=${response%$'\n'*}
|
|
if [[ "$status" == "503" ]]; then
|
|
sleep 0.1
|
|
continue
|
|
fi
|
|
if [[ "$status" != "500" || "$body" != parent:* || "$body" != *"component not found: later"* ]]; then
|
|
echo "missing component TTL probe failed: status=$status body=$body" >&2
|
|
exit 1
|
|
fi
|
|
pid=${body#parent:}
|
|
pid=${pid%%:*}
|
|
seen["$pid"]=1
|
|
done
|
|
if (( ${#seen[@]} != worker_count )); then
|
|
echo "missing component TTL probe reached ${#seen[@]}/$worker_count workers" >&2
|
|
exit 1
|
|
fi
|
|
}
|
|
|
|
collect_workers GET a
|
|
ln -sfn b.uce "$source_dir/chosen.uce"
|
|
collect_workers GET a
|
|
collect_workers POST b
|
|
collect_workers GET a
|
|
sleep 10.2
|
|
collect_workers GET b
|
|
|
|
printf '%s\n' 'RENDER(Request& context) { print("parent:", request_perf()["worker_pid"].to_string(), ":", component("later", context)); }' >"$source_dir/missing.uce"
|
|
collect_missing_workers
|
|
printf '%s\n' 'COMPONENT(Request& context) { print("created"); }' >"$source_dir/later.uce"
|
|
collect_workers GET created missing.uce
|
|
|
|
echo "component resolution TTL passed"
|