NeoTachyon/tests/ai_doctor_local.sh

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

ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
TACHYON_LIB="$ROOT_DIR/tachyon/files/usr/lib"
export TACHYON_LIB="$TACHYON_LIB"
WORK_DIR="$(mktemp -d)"

cleanup() {
  rm -rf "$WORK_DIR"
}
trap cleanup EXIT

fail() {
  printf 'FAIL: %s\n' "$1" >&2
  exit 1
}

# 1. Verify runtime.uc owns local_rule_doctor
grep -Fq 'function local_rule_doctor(' "$TACHYON_LIB/diagnostics/runtime.uc" ||
  fail "runtime.uc must define local_rule_doctor heuristic engine"

# 2. Run ai-doctor without API key (local offline mode)
output="$(ucode -L "$TACHYON_LIB" "$TACHYON_LIB/diagnostics/runtime.uc" ai-doctor 2>/dev/null || true)"

echo "$output" | grep -Fq '"success": true' ||
  fail "ai-doctor in offline mode must return success: true"

echo "$output" | grep -Fq '"provider": "local_heuristic"' ||
  fail "ai-doctor in offline mode must report provider local_heuristic"

# 3. Test apply-quick-fix for new codes
fix_out="$(ucode -L "$TACHYON_LIB" "$TACHYON_LIB/diagnostics/runtime.uc" apply-quick-fix "restart_zapret,optimize_memory,restore_native_internet,fix_system_time,flush_conntrack,fix_bootstrap_dns,optimize_mtu" 2>/dev/null || true)"

echo "$fix_out" | grep -Fq '"code": "restart_zapret"' ||
  fail "apply_quick_fix must handle restart_zapret"

echo "$fix_out" | grep -Fq '"code": "optimize_memory"' ||
  fail "apply_quick_fix must handle optimize_memory"

echo "$fix_out" | grep -Fq '"code": "restore_native_internet"' ||
  fail "apply_quick_fix must handle restore_native_internet"

echo "$fix_out" | grep -Fq '"code": "fix_system_time"' ||
  fail "apply_quick_fix must handle fix_system_time"

echo "$fix_out" | grep -Fq '"code": "flush_conntrack"' ||
  fail "apply_quick_fix must handle flush_conntrack"

echo "$fix_out" | grep -Fq '"code": "fix_bootstrap_dns"' ||
  fail "apply_quick_fix must handle fix_bootstrap_dns"

echo "$fix_out" | grep -Fq '"code": "optimize_mtu"' ||
  fail "apply_quick_fix must handle optimize_mtu"

# 4. Doctor must work with the service stopped (recovery mode: restore stock
#    internet, never start Tachyon) and the AI Doctor must verify the system
#    end-to-end instead of trusting a point-in-time snapshot.
grep -Fq 'function tachyon_is_running()' "$TACHYON_LIB/diagnostics/runtime.uc" ||
  fail "runtime.uc must define tachyon_is_running()"
grep -Fq 'return run_recovery_checks();' "$TACHYON_LIB/diagnostics/runtime.uc" ||
  fail "doctor no longer dispatches to recovery mode when the service is stopped"
grep -Fq 'function run_recovery_checks()' "$TACHYON_LIB/diagnostics/runtime.uc" ||
  fail "runtime.uc must define run_recovery_checks()"
grep -Fq 'function verify_system()' "$TACHYON_LIB/diagnostics/runtime.uc" ||
  fail "runtime.uc must define verify_system()"
grep -Fq 'let verify = (pre_verify != null) ? pre_verify : verify_system();' "$TACHYON_LIB/diagnostics/runtime.uc" ||
  fail "local_rule_doctor no longer feeds live verification into its analysis"
# ai_doctor must run the suite exactly once and share results with the
# heuristic engine instead of executing every probe twice.
grep -Fq 'let local_res = local_rule_doctor(res, verify);' "$TACHYON_LIB/diagnostics/runtime.uc" ||
  fail "ai_doctor must reuse doctor/verify results instead of re-running the suite"
grep -Fq 'wan_fail_streak: int(st.wan_fail_streak || 0)' "$TACHYON_LIB/service/watchdog.uc" ||
  fail "watchdog status export no longer reports probe streaks to AI Doctor"

# 5. Fix usage tracker: applying the same fix must be recorded, and fixes
#    applied 3+ times within the hour must be withheld from recommendations
#    (issue #31: doctor proposing the same repairs forever).
#    Use fix_resolv_symlink (ln -sf works without system services in Docker).
rm -f /tmp/tachyon_doctor_fixes.json
for i in 1 2 3; do
  ucode -L "$TACHYON_LIB" "$TACHYON_LIB/diagnostics/runtime.uc" apply-quick-fix "fix_resolv_symlink" >/dev/null 2>&1 || true
done
grep -Fq '"fix_resolv_symlink"' /tmp/tachyon_doctor_fixes.json ||
  fail "apply_quick_fix must record applied fixes in the doctor usage tracker"
grep -Eq '"count": ?3' /tmp/tachyon_doctor_fixes.json ||
  fail "doctor usage tracker must count repeated applications"
grep -Fq 'function doctor_fix_overused(' "$TACHYON_LIB/diagnostics/runtime.uc" ||
  fail "runtime.uc must define doctor_fix_overused()"
grep -Fq 'if (doctor_fix_overused(code)) return;' "$TACHYON_LIB/diagnostics/runtime.uc" ||
  fail "add_fix must withhold overused fixes"

# 6. fix_uci_config must use uci_backup_restore() (correct path)
grep -Fq 'uci_backup_restore()' "$TACHYON_LIB/diagnostics/runtime.uc" ||
  fail "fix_uci_config must call uci_backup_restore() for correct backup path"

# 7. apply_quick_fix must capture honest exit codes
grep -Fq 'let rc = command_status(' "$TACHYON_LIB/diagnostics/runtime.uc" ||
  fail "apply_quick_fix must capture command exit codes"

# 8. doctor() must propagate busy flag
grep -Fq 'busy: res.busy == true' "$TACHYON_LIB/diagnostics/runtime.uc" ||
  fail "doctor() must propagate busy flag in output"

# 9. diagnose_json must use structured checks[] when available
grep -Fq 'Structured path: build problems directly from checks' "$TACHYON_LIB/diagnostics/runtime.uc" ||
  fail "diagnose_json must build problems from structured checks[]"
grep -Fq 'Legacy fallback for recovery mode' "$TACHYON_LIB/diagnostics/runtime.uc" ||
  fail "diagnose_json must have legacy emoji fallback for recovery mode"

# 10. rag.uc push args must be correct (push array, value)
grep -Fq 'push(texts, index.chunks[i].text)' "$TACHYON_LIB/diagnostics/rag.uc" ||
  fail "rag.uc must push with correct argument order: push(array, value)"

printf 'local AI Doctor checks passed\n'