mirror of
https://github.com/obra/superpowers
synced 2026-07-26 12:34:30 +00:00
111 lines
3.6 KiB
Bash
111 lines
3.6 KiB
Bash
|
|
#!/usr/bin/env bash
|
||
|
|
set -euo pipefail
|
||
|
|
|
||
|
|
SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
|
||
|
|
REPO_ROOT="$(cd "$SCRIPT_DIR/../.." && pwd)"
|
||
|
|
HOOK_UNDER_TEST="$REPO_ROOT/hooks/session-start-codex"
|
||
|
|
EXAMPLE_UNDER_TEST="$REPO_ROOT/hooks/hooks-codex.json.example"
|
||
|
|
|
||
|
|
FAILURES=0
|
||
|
|
|
||
|
|
pass() {
|
||
|
|
echo " [PASS] $1"
|
||
|
|
}
|
||
|
|
|
||
|
|
fail() {
|
||
|
|
echo " [FAIL] $1"
|
||
|
|
FAILURES=$((FAILURES + 1))
|
||
|
|
}
|
||
|
|
|
||
|
|
# run_hook <stdin-payload> — echoes hook stdout; fails the calling test on
|
||
|
|
# non-zero exit. env -i mirrors the codex hook executor's clean environment.
|
||
|
|
run_hook() {
|
||
|
|
printf '%s' "$1" | env -i PATH="${PATH:-}" bash "$HOOK_UNDER_TEST"
|
||
|
|
}
|
||
|
|
|
||
|
|
echo "Codex SessionStart hook tests"
|
||
|
|
|
||
|
|
startup_payload='{"session_id":"s","hook_event_name":"SessionStart","model":"gpt-5.6-terra","source":"startup"}'
|
||
|
|
if output="$(run_hook "$startup_payload")" && [ -z "$output" ]; then
|
||
|
|
pass "source=startup emits nothing and exits 0"
|
||
|
|
else
|
||
|
|
fail "source=startup emits nothing and exits 0"
|
||
|
|
printf '%s\n' "$output" | head -3 | sed 's/^/ /'
|
||
|
|
fi
|
||
|
|
|
||
|
|
compact_payload='{"session_id":"s","hook_event_name":"SessionStart","model":"gpt-5.6-terra","source":"compact"}'
|
||
|
|
if output="$(run_hook "$compact_payload")"; then
|
||
|
|
ok=1
|
||
|
|
for needle in \
|
||
|
|
"<EXTREMELY_IMPORTANT>" \
|
||
|
|
"You have superpowers." \
|
||
|
|
"name: using-superpowers" \
|
||
|
|
"<CONTEXT_RESTORED>" \
|
||
|
|
"subagent-driven-development/SKILL.md" \
|
||
|
|
"references/codex-tools.md"; do
|
||
|
|
if [[ "$output" != *"$needle"* ]]; then
|
||
|
|
ok=0
|
||
|
|
echo " missing: $needle"
|
||
|
|
fi
|
||
|
|
done
|
||
|
|
if [ "$ok" -eq 1 ]; then
|
||
|
|
pass "source=compact emits bootstrap plus re-read addendum"
|
||
|
|
else
|
||
|
|
fail "source=compact emits bootstrap plus re-read addendum"
|
||
|
|
fi
|
||
|
|
else
|
||
|
|
fail "source=compact emits bootstrap plus re-read addendum (hook exited non-zero)"
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Whitespace-tolerant source matching (serializers vary).
|
||
|
|
spaced_payload='{"hook_event_name":"SessionStart", "source" : "compact"}'
|
||
|
|
if output="$(run_hook "$spaced_payload")" && [[ "$output" == *"<CONTEXT_RESTORED>"* ]]; then
|
||
|
|
pass "whitespace around the source key still triggers injection"
|
||
|
|
else
|
||
|
|
fail "whitespace around the source key still triggers injection"
|
||
|
|
fi
|
||
|
|
|
||
|
|
if output="$(printf '' | env -i PATH="${PATH:-}" bash "$HOOK_UNDER_TEST")" && [ -z "$output" ]; then
|
||
|
|
pass "empty stdin fails open to no output, exit 0"
|
||
|
|
else
|
||
|
|
fail "empty stdin fails open to no output, exit 0"
|
||
|
|
fi
|
||
|
|
|
||
|
|
if output="$(run_hook 'not json at all {{{')" && [ -z "$output" ]; then
|
||
|
|
pass "garbage stdin fails open to no output, exit 0"
|
||
|
|
else
|
||
|
|
fail "garbage stdin fails open to no output, exit 0"
|
||
|
|
fi
|
||
|
|
|
||
|
|
# A compact mention inside some other field must not trigger injection.
|
||
|
|
decoy_payload='{"hook_event_name":"SessionStart","source":"startup","cwd":"/tmp/compact"}'
|
||
|
|
if output="$(run_hook "$decoy_payload")" && [ -z "$output" ]; then
|
||
|
|
pass "compact appearing outside the source field does not trigger"
|
||
|
|
else
|
||
|
|
fail "compact appearing outside the source field does not trigger"
|
||
|
|
fi
|
||
|
|
|
||
|
|
if node -e '
|
||
|
|
const example = JSON.parse(require("fs").readFileSync(process.argv[1], "utf8"));
|
||
|
|
const entry = example.hooks.SessionStart[0].hooks[0];
|
||
|
|
if (entry.type !== "command") {
|
||
|
|
console.error(`example hook type is ${JSON.stringify(entry.type)}, expected "command"`);
|
||
|
|
process.exit(1);
|
||
|
|
}
|
||
|
|
if (!/session-start-codex"$/.test(entry.command)) {
|
||
|
|
console.error(`unexpected example command shape: ${entry.command}`);
|
||
|
|
process.exit(1);
|
||
|
|
}
|
||
|
|
' "$EXAMPLE_UNDER_TEST"; then
|
||
|
|
pass "hooks-codex.json.example parses and invokes session-start-codex"
|
||
|
|
else
|
||
|
|
fail "hooks-codex.json.example parses and invokes session-start-codex"
|
||
|
|
fi
|
||
|
|
|
||
|
|
if [[ "$FAILURES" -gt 0 ]]; then
|
||
|
|
echo "STATUS: FAILED ($FAILURES failure(s))"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
echo "STATUS: PASSED"
|