test.sh

#!/usr/bin/env bash
# test.sh — Run main.py --testing with different flag combinations.
#
# Usage:
#   ./test.sh           # Run all tests
#   ./test.sh preflight # Check LLM, embedding, book APIs
#   ./test.sh pipeline  # Full run, all test books
#   ./test.sh filter    # Filter flag tests
#   ./test.sh tracker   # Skip/force behavior
#   ./test.sh quick     # Single easy book
#   ./test.sh region    # UK region fallback
#   ./test.sh cleaning  # Title cleaning tests
#   ./test.sh hard      # Books expected to fail

set -e

DIR="$(cd "$(dirname "$0")" && pwd)"
cd "$DIR"

GREEN='\033[32m'
RED='\033[31m'
CYAN='\033[36m'
BOLD='\033[1m'
RESET='\033[0m'

PASS=0
FAIL=0

run_test() {
    local name="$1"
    shift
    echo ""
    echo -e "${CYAN}--- TEST: ${name} ---${RESET}"
    echo -e "  cmd: python main.py --testing $*"
    if python main.py --testing "$@"; then
        echo -e "  ${GREEN}OK${RESET}"
        ((PASS++))
    else
        echo -e "  ${RED}FAILED${RESET}"
        ((FAIL++))
    fi
}

clean() {
    rm -f processed_test.json
}

# ============================================================
# Suites
# ============================================================

test_preflight() {
    echo -e "\n${BOLD}========== SUITE: preflight — check services ==========${RESET}"
    if python3 preflight.py; then
        ((PASS++))
    else
        ((FAIL++))
    fi
}

test_pipeline() {
    echo -e "\n${BOLD}========== SUITE: pipeline — full run, all books ==========${RESET}"
    clean
    run_test "all books, force" --force
}

test_filter() {
    echo -e "\n${BOLD}========== SUITE: filter — author/series/title flags ==========${RESET}"

    clean
    run_test "filter --author Flanagan" --force --author "Flanagan"

    clean
    run_test "filter --author Tolkien" --force --author "Tolkien"

    clean
    run_test "filter --series Dune" --force --series "Dune"

    clean
    run_test "filter --title Hobbit" --force --title "Hobbit"

    clean
    run_test "filter --author Tolkien --title Hobbit" --force --author "Tolkien" --title "Hobbit"

    clean
    run_test "filter --author Herbert --series Dune" --force --author "Herbert" --series "Dune"

    clean
    run_test "filter --author NonexistentAuthor (expect 0 total)" --force --author "Zzzzzzzz"
}

test_tracker() {
    echo -e "\n${BOLD}========== SUITE: tracker — skip/force behavior ==========${RESET}"

    clean
    run_test "tracker: first run (populates)" --force --title "Hail Mary"

    run_test "tracker: second run (should skip)" --title "Hail Mary"

    run_test "tracker: third run with --force (should reprocess)" --force --title "Hail Mary"
}

test_quick() {
    echo -e "\n${BOLD}========== SUITE: quick — single easy book ==========${RESET}"
    clean
    run_test "quick: The Hobbit" --force --title "The Hobbit" --author "Tolkien"
}

test_region() {
    echo -e "\n${BOLD}========== SUITE: region — UK region fallback ==========${RESET}"
    clean
    run_test "region: Icebound Land (needs UK)" --force --title "Icebound"
}

test_cleaning() {
    echo -e "\n${BOLD}========== SUITE: cleaning — titles that need clean_title ==========${RESET}"
    clean
    run_test "cleaning: 'Dune 01 - Dune'" --force --title "Dune 01"

    clean
    run_test "cleaning: series suffix" --force --title "Lost Stories"

    clean
    run_test "cleaning: MI prefix" --force --title "City of Fallen"
}

test_hard() {
    echo -e "\n${BOLD}========== SUITE: hard — books expected to fail ==========${RESET}"
    clean
    run_test "hard: Tolkien Reads and Sings (obscure)" --force --title "Tolkien Reads"

    clean
    run_test "hard: Oakleaf Bearers (AU title)" --force --title "Oakleaf"

    clean
    run_test "hard: Poems and Songs (should not false-match)" --force --title "Poems and Songs"
}

# ============================================================
# Main
# ============================================================

SUITE="${1:-all}"

case "$SUITE" in
    all)
        test_preflight
        test_pipeline
        test_filter
        test_tracker
        test_quick
        test_region
        test_cleaning
        test_hard
        ;;
    preflight) test_preflight ;;
    pipeline)  test_pipeline ;;
    filter)    test_filter ;;
    tracker)   test_tracker ;;
    quick)     test_quick ;;
    region)    test_region ;;
    cleaning)  test_cleaning ;;
    hard)      test_hard ;;
    *)
        echo "Unknown suite: $SUITE"
        echo "Available: all, preflight, pipeline, filter, tracker, quick, region, cleaning, hard"
        exit 1
        ;;
esac

# Summary
echo ""
echo -e "${BOLD}============================================${RESET}"
if [ "$FAIL" -eq 0 ]; then
    echo -e "  ${GREEN}ALL $PASS TESTS PASSED${RESET}"
else
    echo -e "  ${RED}$FAIL FAILED${RESET} / $((PASS + FAIL)) total"
fi
echo -e "${BOLD}============================================${RESET}"

clean
[ "$FAIL" -eq 0 ]