AI Assistant
107 lines
3.4 KiB
Bash
Executable File
107 lines
3.4 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
|
|
ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
|
|
TMP_DIR="$(mktemp -d)"
|
|
MOCK_PORT="${SELFTEST_MOCK_PORT:-18080}"
|
|
APP_PORT="${SELFTEST_APP_PORT:-18081}"
|
|
MOCK_PID=""
|
|
APP_PID=""
|
|
|
|
cleanup() {
|
|
if [[ -n "${APP_PID}" ]] && kill -0 "${APP_PID}" >/dev/null 2>&1; then
|
|
kill "${APP_PID}" >/dev/null 2>&1 || true
|
|
wait "${APP_PID}" >/dev/null 2>&1 || true
|
|
fi
|
|
if [[ -n "${MOCK_PID}" ]] && kill -0 "${MOCK_PID}" >/dev/null 2>&1; then
|
|
kill "${MOCK_PID}" >/dev/null 2>&1 || true
|
|
wait "${MOCK_PID}" >/dev/null 2>&1 || true
|
|
fi
|
|
rm -rf "${TMP_DIR}"
|
|
}
|
|
trap cleanup EXIT
|
|
|
|
cd "${ROOT_DIR}"
|
|
|
|
echo "[selftest] gofmt"
|
|
gofmt -w backend/main.go backend/handlers/api.go backend/models/db.go backend/services/cse.go backend/services/cse_test.go backend/services/search_collectors.go backend/services/ranker.go backend/services/gemini.go backend/services/gemini_test.go
|
|
|
|
echo "[selftest] python syntax"
|
|
python3 -m py_compile worker/downloader.py scripts/mock_searxng.py
|
|
|
|
echo "[selftest] go test"
|
|
go test ./...
|
|
|
|
echo "[selftest] go build"
|
|
go build -o "${TMP_DIR}/ai-media-hub" ./backend
|
|
|
|
echo "[selftest] start mock searxng"
|
|
python3 scripts/mock_searxng.py --port "${MOCK_PORT}" >"${TMP_DIR}/mock-searxng.log" 2>&1 &
|
|
MOCK_PID=$!
|
|
|
|
echo "[selftest] start app"
|
|
mkdir -p "${TMP_DIR}/downloads"
|
|
APP_ROOT="${ROOT_DIR}" \
|
|
APP_ADDR="127.0.0.1:${APP_PORT}" \
|
|
SQLITE_PATH="${TMP_DIR}/media.db" \
|
|
DOWNLOADS_DIR="${TMP_DIR}/downloads" \
|
|
FRONTEND_DIR="${ROOT_DIR}/frontend" \
|
|
WORKER_SCRIPT="${ROOT_DIR}/worker/downloader.py" \
|
|
SEARXNG_BASE_URL="http://127.0.0.1:${MOCK_PORT}" \
|
|
"${TMP_DIR}/ai-media-hub" >"${TMP_DIR}/app.log" 2>&1 &
|
|
APP_PID=$!
|
|
|
|
for _ in $(seq 1 30); do
|
|
if curl -fsS "http://127.0.0.1:${APP_PORT}/healthz" >/dev/null; then
|
|
break
|
|
fi
|
|
sleep 1
|
|
done
|
|
|
|
echo "[selftest] verify healthz"
|
|
curl -fsS "http://127.0.0.1:${APP_PORT}/healthz" >"${TMP_DIR}/healthz.json"
|
|
python3 - "${TMP_DIR}/healthz.json" <<'PY'
|
|
import json
|
|
import sys
|
|
with open(sys.argv[1], "r", encoding="utf-8") as handle:
|
|
payload = json.load(handle)
|
|
if payload.get("status") != "ok":
|
|
raise SystemExit(f"unexpected healthz payload: {payload}")
|
|
PY
|
|
|
|
echo "[selftest] verify search"
|
|
curl -fsS \
|
|
-H "Content-Type: application/json" \
|
|
-d '{"query":"city rain","platforms":["envato","artgrid","google video"]}' \
|
|
"http://127.0.0.1:${APP_PORT}/api/search" >"${TMP_DIR}/search.json"
|
|
python3 - "${TMP_DIR}/search.json" <<'PY'
|
|
import json
|
|
import sys
|
|
with open(sys.argv[1], "r", encoding="utf-8") as handle:
|
|
payload = json.load(handle)
|
|
results = payload.get("results") or []
|
|
if len(results) < 2:
|
|
raise SystemExit(f"expected >= 2 search results, got {len(results)}: {payload}")
|
|
if not all(item.get("link") for item in results):
|
|
raise SystemExit(f"search results must include links: {payload}")
|
|
PY
|
|
|
|
echo "[selftest] verify upload"
|
|
printf 'selftest upload\n' > "${TMP_DIR}/sample.txt"
|
|
curl -fsS -F "file=@${TMP_DIR}/sample.txt" "http://127.0.0.1:${APP_PORT}/api/upload" >"${TMP_DIR}/upload.json"
|
|
python3 - "${TMP_DIR}/upload.json" "${TMP_DIR}/downloads" <<'PY'
|
|
import json
|
|
import os
|
|
import sys
|
|
with open(sys.argv[1], "r", encoding="utf-8") as handle:
|
|
payload = json.load(handle)
|
|
filename = payload.get("filename")
|
|
if not filename:
|
|
raise SystemExit(f"missing filename in upload payload: {payload}")
|
|
target = os.path.join(sys.argv[2], filename)
|
|
if not os.path.exists(target):
|
|
raise SystemExit(f"expected uploaded file at {target}")
|
|
PY
|
|
|
|
echo "[selftest] ok"
|