Files
2026-08-17 22:07:11 +03:00

92 lines
3.3 KiB
Bash
Executable File

#!/usr/bin/env bash
# A supervised worker that produces a valid MCP error response is healthy.
# The CLI may exit nonzero when presenting that error to a human, but the worker
# transport must exit zero so its parent reads the response instead of inventing
# an exit_nonzero "crashed on a file" diagnosis.
set -euo pipefail
ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
# scripts/test.sh builds into $BUILD_DIR, which is NOT build/c on every leg (the
# Linux containers use build/linux-arm64 / build/linux-amd64). Honour the binary
# the caller built, exactly as test_parent_watchdog.sh and test_worker_watchdog.sh
# do; the build/c default keeps a bare manual invocation working.
BINARY="${CBM_TEST_BINARY:-${ROOT}/build/c/codebase-memory-mcp}"
if [[ ! -x "${BINARY}" && -x "${BINARY}.exe" ]]; then
BINARY="${BINARY}.exe"
fi
if [[ ! -x "${BINARY}" ]]; then
echo "missing binary: ${BINARY}" >&2
exit 2
fi
if command -v shasum >/dev/null 2>&1; then
BUILD_FINGERPRINT="$(shasum -a 256 "${BINARY}" | awk '{print $1}')"
elif command -v sha256sum >/dev/null 2>&1; then
BUILD_FINGERPRINT="$(sha256sum "${BINARY}" | awk '{print $1}')"
elif command -v openssl >/dev/null 2>&1; then
BUILD_FINGERPRINT="$(openssl dgst -sha256 "${BINARY}" | awk '{print $NF}')"
else
echo "no SHA-256 command available for worker build binding" >&2
exit 2
fi
if [[ ! "${BUILD_FINGERPRINT}" =~ ^[0-9a-f]{64}$ ]]; then
echo "invalid worker build fingerprint: ${BUILD_FINGERPRINT}" >&2
exit 2
fi
# shellcheck source=../scripts/test-runtime.sh
source "${ROOT}/scripts/test-runtime.sh"
cbm_test_runtime_init
tmpdir="${CBM_TEST_RUNTIME_ROOT}"
cleanup() {
CBM_CACHE_DIR="${tmpdir}/cache-supervisor" \
"${BINARY}" daemon stop >/dev/null 2>&1 || true
cbm_test_runtime_cleanup "${BINARY}"
}
trap cleanup EXIT
missing="${tmpdir}/repository-does-not-exist"
response="${tmpdir}/worker.response"
args="{\"repo_path\":\"${missing}\",\"mode\":\"fast\"}"
if ! CBM_CACHE_DIR="${tmpdir}/cache-worker" \
"${BINARY}" cli --index-worker \
--index-worker-build "${BUILD_FINGERPRINT}" \
index_repository "${args}" \
--response-out "${response}" >"${tmpdir}/worker.out" 2>"${tmpdir}/worker.err"; then
echo "worker treated a delivered MCP error as a process failure" >&2
cat "${tmpdir}/worker.err" >&2
exit 1
fi
if [[ ! -s "${response}" ]] || ! grep -q 'Pipeline failed' "${response}"; then
echo "worker did not deliver the underlying pipeline error" >&2
cat "${response}" 2>/dev/null >&2 || true
exit 1
fi
set +e
CBM_CACHE_DIR="${tmpdir}/cache-supervisor" \
"${BINARY}" cli index_repository --repo-path "${missing}" --mode fast \
>"${tmpdir}/supervisor.out" 2>"${tmpdir}/supervisor.err"
cli_rc=$?
set -e
if [[ ${cli_rc} -eq 0 ]]; then
echo "human-facing CLI unexpectedly accepted an indexing error" >&2
exit 1
fi
if grep -q 'crashed on a file' "${tmpdir}/supervisor.out" "${tmpdir}/supervisor.err"; then
echo "supervisor replaced a valid tool error with a false crash diagnosis" >&2
exit 1
fi
if ! grep -q 'Pipeline failed' "${tmpdir}/supervisor.out" "${tmpdir}/supervisor.err"; then
echo "supervisor did not preserve the worker's pipeline error" >&2
cat "${tmpdir}/supervisor.out" >&2
cat "${tmpdir}/supervisor.err" >&2
exit 1
fi
echo "ok: worker MCP errors remain tool errors, not process crashes"