#!/usr/bin/env bash
set -euo pipefail

ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/../.." && pwd)"
cd "$ROOT_DIR"

TMP_DIR="$(mktemp -d)"
trap 'rm -rf "$TMP_DIR"' EXIT

BAD_FILE="$TMP_DIR/bad-output.txt"
cat > "$BAD_FILE" <<'EOF'
long command output line 1
long command output line 2
EOF

set +e
bash scripts/review/partial_output_guard.sh "$BAD_FILE" >/tmp/partial_bad.out 2>&1
BAD_RC=$?
set -e
[[ "$BAD_RC" -ne 0 ]]
grep -q 'missing conservative partial-output template' /tmp/partial_bad.out

GOOD_FILE="$TMP_DIR/good-output.txt"
cat > "$GOOD_FILE" <<'EOF'
PARTIAL_OUTPUT_STATUS complete=false source=artifact-required conclusion=conservative
artifact_hint=artifact://123
EOF

bash scripts/review/partial_output_guard.sh "$GOOD_FILE" >/tmp/partial_good.out 2>&1
grep -q 'PARTIAL_OUTPUT_GUARD: PASS' /tmp/partial_good.out