build: add server-rs smoke scripts
This commit is contained in:
188
server-rs/scripts/smoke.sh
Normal file
188
server-rs/scripts/smoke.sh
Normal file
@@ -0,0 +1,188 @@
|
||||
#!/usr/bin/env bash
|
||||
|
||||
set -euo pipefail
|
||||
|
||||
# 先编译并拉起临时 api-server 进程，再校验 /healthz 的裸响应、envelope 和关键响应头契约。
|
||||
|
||||
usage() {
|
||||
cat <<'EOF'
|
||||
用法:
|
||||
./server-rs/scripts/smoke.sh
|
||||
GENARRATIVE_SMOKE_PORT=3201 ./server-rs/scripts/smoke.sh
|
||||
|
||||
说明:
|
||||
1. 先执行 `cargo build -p api-server`
|
||||
2. 启动一个临时 `api-server` 进程并等待 `/healthz` 就绪
|
||||
3. 校验 raw `/healthz`、envelope `/healthz` 以及 `x-request-id / x-api-version / x-route-version / x-response-time-ms`
|
||||
4. 冒烟结束后自动回收临时进程
|
||||
EOF
|
||||
}
|
||||
|
||||
if [[ "${1:-}" == "-h" || "${1:-}" == "--help" ]]; then
|
||||
usage
|
||||
exit 0
|
||||
fi
|
||||
|
||||
SCRIPT_DIR="$(cd -- "$(dirname -- "${BASH_SOURCE[0]}")" && pwd)"
|
||||
SERVER_RS_DIR="$(cd -- "${SCRIPT_DIR}/.." && pwd)"
|
||||
MANIFEST_PATH="${SERVER_RS_DIR}/Cargo.toml"
|
||||
SERVER_BIN="${SERVER_RS_DIR}/target/debug/api-server"
|
||||
API_HOST="${GENARRATIVE_SMOKE_HOST:-127.0.0.1}"
|
||||
API_PORT="${GENARRATIVE_SMOKE_PORT:-3101}"
|
||||
API_LOG="${GENARRATIVE_SMOKE_LOG:-warn,tower_http=warn}"
|
||||
STARTUP_TIMEOUT_SECONDS="${GENARRATIVE_SMOKE_TIMEOUT_SECONDS:-30}"
|
||||
BASE_URL="http://${API_HOST}:${API_PORT}"
|
||||
HEALTHZ_URL="${BASE_URL}/healthz"
|
||||
RUN_ID="$(date +%s)-$$"
|
||||
STDOUT_LOG="${TMPDIR:-/tmp}/genarrative-server-rs-smoke-${RUN_ID}.stdout.log"
|
||||
STDERR_LOG="${TMPDIR:-/tmp}/genarrative-server-rs-smoke-${RUN_ID}.stderr.log"
|
||||
KEEP_LOGS=0
|
||||
SERVER_PID=""
|
||||
|
||||
assert_contains() {
|
||||
local haystack="$1"
|
||||
local needle="$2"
|
||||
local message="$3"
|
||||
|
||||
if [[ "${haystack}" != *"${needle}"* ]]; then
|
||||
echo "[server-rs:smoke] ${message}" >&2
|
||||
exit 1
|
||||
fi
|
||||
}
|
||||
|
||||
print_logs() {
|
||||
if [[ -f "${STDOUT_LOG}" ]]; then
|
||||
echo "[server-rs:smoke] stdout:"
|
||||
cat "${STDOUT_LOG}"
|
||||
fi
|
||||
|
||||
if [[ -f "${STDERR_LOG}" ]]; then
|
||||
echo "[server-rs:smoke] stderr:"
|
||||
cat "${STDERR_LOG}"
|
||||
fi
|
||||
}
|
||||
|
||||
cleanup() {
|
||||
if [[ -n "${SERVER_PID}" ]] && kill -0 "${SERVER_PID}" 2>/dev/null; then
|
||||
kill "${SERVER_PID}" 2>/dev/null || true
|
||||
wait "${SERVER_PID}" 2>/dev/null || true
|
||||
fi
|
||||
|
||||
if [[ "${KEEP_LOGS}" -eq 0 ]]; then
|
||||
rm -f "${STDOUT_LOG}" "${STDERR_LOG}"
|
||||
fi
|
||||
}
|
||||
|
||||
trap cleanup EXIT
|
||||
|
||||
if [[ ! -f "${MANIFEST_PATH}" ]]; then
|
||||
echo "[server-rs:smoke] 未找到 ${MANIFEST_PATH}，无法启动冒烟脚本。" >&2
|
||||
exit 1
|
||||
fi
|
||||
|
||||
cd "${SERVER_RS_DIR}"
|
||||
|
||||
echo "[server-rs:smoke] 步骤: cargo build -p api-server"
|
||||
cargo build -p api-server --manifest-path "${MANIFEST_PATH}"
|
||||
|
||||
if [[ ! -x "${SERVER_BIN}" ]]; then
|
||||
echo "[server-rs:smoke] 未找到 ${SERVER_BIN}，构建结果异常。" >&2
|
||||
exit 1
|
||||
fi
|
||||
|
||||
echo "[server-rs:smoke] 步骤: start api-server binary"
|
||||
GENARRATIVE_API_HOST="${API_HOST}" \
|
||||
GENARRATIVE_API_PORT="${API_PORT}" \
|
||||
GENARRATIVE_API_LOG="${API_LOG}" \
|
||||
"${SERVER_BIN}" >"${STDOUT_LOG}" 2>"${STDERR_LOG}" &
|
||||
SERVER_PID=$!
|
||||
|
||||
echo "[server-rs:smoke] 步骤: wait for /healthz readiness"
|
||||
READY=0
|
||||
for ((i = 0; i < STARTUP_TIMEOUT_SECONDS * 10; i++)); do
|
||||
if ! kill -0 "${SERVER_PID}" 2>/dev/null; then
|
||||
KEEP_LOGS=1
|
||||
print_logs
|
||||
echo "[server-rs:smoke] api-server exited before /healthz became ready." >&2
|
||||
exit 1
|
||||
fi
|
||||
|
||||
if curl -fsS "${HEALTHZ_URL}" >/dev/null 2>&1; then
|
||||
READY=1
|
||||
break
|
||||
fi
|
||||
|
||||
sleep 0.1
|
||||
done
|
||||
|
||||
if [[ "${READY}" -ne 1 ]]; then
|
||||
KEEP_LOGS=1
|
||||
print_logs
|
||||
echo "[server-rs:smoke] /healthz readiness timed out." >&2
|
||||
exit 1
|
||||
fi
|
||||
|
||||
RAW_BODY_FILE="${TMPDIR:-/tmp}/genarrative-server-rs-smoke-${RUN_ID}.raw.body.json"
|
||||
RAW_HEADER_FILE="${TMPDIR:-/tmp}/genarrative-server-rs-smoke-${RUN_ID}.raw.headers.txt"
|
||||
ENVELOPE_BODY_FILE="${TMPDIR:-/tmp}/genarrative-server-rs-smoke-${RUN_ID}.envelope.body.json"
|
||||
ENVELOPE_HEADER_FILE="${TMPDIR:-/tmp}/genarrative-server-rs-smoke-${RUN_ID}.envelope.headers.txt"
|
||||
|
||||
trap 'rm -f "${RAW_BODY_FILE}" "${RAW_HEADER_FILE}" "${ENVELOPE_BODY_FILE}" "${ENVELOPE_HEADER_FILE}"; cleanup' EXIT
|
||||
|
||||
RAW_REQUEST_ID="smoke-healthz-raw"
|
||||
echo "[server-rs:smoke] 步骤: verify raw /healthz contract"
|
||||
curl -fsS \
|
||||
-H "x-request-id: ${RAW_REQUEST_ID}" \
|
||||
-D "${RAW_HEADER_FILE}" \
|
||||
-o "${RAW_BODY_FILE}" \
|
||||
"${HEALTHZ_URL}"
|
||||
|
||||
RAW_BODY="$(tr -d '\n' <"${RAW_BODY_FILE}")"
|
||||
RAW_REQUEST_ID_HEADER="$(grep -i '^x-request-id:' "${RAW_HEADER_FILE}" | tail -n 1 | cut -d':' -f2- | tr -d '\r' | xargs)"
|
||||
API_VERSION_HEADER="$(grep -i '^x-api-version:' "${RAW_HEADER_FILE}" | tail -n 1 | cut -d':' -f2- | tr -d '\r' | xargs)"
|
||||
ROUTE_VERSION_HEADER="$(grep -i '^x-route-version:' "${RAW_HEADER_FILE}" | tail -n 1 | cut -d':' -f2- | tr -d '\r' | xargs)"
|
||||
RESPONSE_TIME_HEADER="$(grep -i '^x-response-time-ms:' "${RAW_HEADER_FILE}" | tail -n 1 | cut -d':' -f2- | tr -d '\r' | xargs)"
|
||||
|
||||
assert_contains "${RAW_BODY}" '"ok":true' "raw /healthz body 缺少 ok=true。"
|
||||
assert_contains "${RAW_BODY}" '"service":"genarrative-node-server"' "raw /healthz body 返回了意外的 service。"
|
||||
|
||||
if [[ "${RAW_REQUEST_ID_HEADER}" != "${RAW_REQUEST_ID}" ]]; then
|
||||
echo "[server-rs:smoke] raw /healthz 没有正确回写 x-request-id。" >&2
|
||||
exit 1
|
||||
fi
|
||||
|
||||
if [[ -z "${API_VERSION_HEADER}" ]]; then
|
||||
echo "[server-rs:smoke] raw /healthz 缺少 x-api-version。" >&2
|
||||
exit 1
|
||||
fi
|
||||
|
||||
if [[ "${ROUTE_VERSION_HEADER}" != "${API_VERSION_HEADER}" ]]; then
|
||||
echo "[server-rs:smoke] raw /healthz 的 x-route-version 与 x-api-version 不一致。" >&2
|
||||
exit 1
|
||||
fi
|
||||
|
||||
if [[ ! "${RESPONSE_TIME_HEADER}" =~ ^[0-9]+$ ]]; then
|
||||
echo "[server-rs:smoke] raw /healthz 的 x-response-time-ms 不是合法整数。" >&2
|
||||
exit 1
|
||||
fi
|
||||
|
||||
ENVELOPE_REQUEST_ID="smoke-healthz-envelope"
|
||||
echo "[server-rs:smoke] 步骤: verify envelope /healthz contract"
|
||||
curl -fsS \
|
||||
-H "x-request-id: ${ENVELOPE_REQUEST_ID}" \
|
||||
-H "x-genarrative-response-envelope: 1" \
|
||||
-D "${ENVELOPE_HEADER_FILE}" \
|
||||
-o "${ENVELOPE_BODY_FILE}" \
|
||||
"${HEALTHZ_URL}"
|
||||
|
||||
ENVELOPE_BODY="$(tr -d '\n' <"${ENVELOPE_BODY_FILE}")"
|
||||
|
||||
assert_contains "${ENVELOPE_BODY}" '"ok":true' "envelope /healthz body 缺少 ok=true。"
|
||||
assert_contains "${ENVELOPE_BODY}" '"error":null' "envelope /healthz body 缺少 error=null。"
|
||||
assert_contains "${ENVELOPE_BODY}" '"data":{"ok":true,"service":"genarrative-node-server"}' "envelope /healthz data 返回异常。"
|
||||
assert_contains "${ENVELOPE_BODY}" "\"requestId\":\"${ENVELOPE_REQUEST_ID}\"" "envelope /healthz meta.requestId 没有回写请求 ID。"
|
||||
assert_contains "${ENVELOPE_BODY}" "\"apiVersion\":\"${API_VERSION_HEADER}\"" "envelope /healthz meta.apiVersion 与响应头不一致。"
|
||||
assert_contains "${ENVELOPE_BODY}" "\"routeVersion\":\"${ROUTE_VERSION_HEADER}\"" "envelope /healthz meta.routeVersion 与响应头不一致。"
|
||||
assert_contains "${ENVELOPE_BODY}" '"operation":"GET /healthz"' "envelope /healthz meta.operation 异常。"
|
||||
|
||||
echo "[server-rs:smoke] all checks passed"
|
||||
Reference in New Issue
Block a user