#!/bin/bash
# ============================================================
# Run pgclone tests locally using Docker
#
# Usage:
#   ./test/run_all.sh              # Test on all PG versions
#   ./test/run_all.sh 18           # Test on PG 18 only
#   ./test/run_all.sh 16 17        # Test on PG 16 and 17
# ============================================================

set -e

VERSIONS="${@:-14 15 16 17 18}"

echo "============================================"
echo "pgclone multi-version test suite"
echo "Testing on PostgreSQL: $VERSIONS"
echo "============================================"
echo ""

# Build source database
echo "Starting source database..."
docker compose up -d source-db
sleep 3

PASS=0
FAIL=0
RESULTS=""

for ver in $VERSIONS; do
    echo ""
    echo "============================================"
    echo "Testing on PostgreSQL $ver..."
    echo "============================================"

    SERVICE="test-pg${ver}"

    if docker compose build "$SERVICE" 2>&1 | tail -5; then
        if docker compose run --rm "$SERVICE" 2>&1; then
            echo "✅ PostgreSQL $ver: PASSED"
            RESULTS="$RESULTS\n  ✅ PostgreSQL $ver: PASSED"
            PASS=$((PASS + 1))
        else
            echo "❌ PostgreSQL $ver: FAILED"
            RESULTS="$RESULTS\n  ❌ PostgreSQL $ver: FAILED"
            FAIL=$((FAIL + 1))
        fi
    else
        echo "❌ PostgreSQL $ver: BUILD FAILED"
        RESULTS="$RESULTS\n  ❌ PostgreSQL $ver: BUILD FAILED"
        FAIL=$((FAIL + 1))
    fi
done

# Cleanup
echo ""
echo "Cleaning up..."
docker compose down -v 2>/dev/null

echo ""
echo "============================================"
echo "RESULTS SUMMARY"
echo "============================================"
echo -e "$RESULTS"
echo ""
echo "Passed: $PASS / $((PASS + FAIL))"

if [ $FAIL -gt 0 ]; then
    echo "STATUS: SOME TESTS FAILED"
    exit 1
else
    echo "STATUS: ALL TESTS PASSED ✅"
    exit 0
fi