#!/usr/bin/env bash

set -euo pipefail

ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
COMPOSE_FILE="${ROOT_DIR}/tests/docker-compose.yml"
KEEP_DB="${KEEP_DB:-0}"
COMPOSE_PROJECT_NAME="${COMPOSE_PROJECT_NAME:-wpkdb-postgres-correctness}"
TEST_URL="${TEST_URL:-postgres://postgres:demo@127.0.0.1:5432/postgres}"
WAIT_SECONDS="${WAIT_SECONDS:-90}"
POSTGRES_TESTS=(
  postgres_provider_reconnects_after_backend_termination
  postgres_provider_init_and_query_inside_tokio_runtime
  postgres_provider_query_and_pool
  postgres_provider_sqlx_type_compatibility
)

. "${ROOT_DIR}/tests/provider-test-common.sh"
install_compose_cleanup_trap

cd "${ROOT_DIR}"

log_inherited_test_url "WP_KDB_TEST_POSTGRES_URL"

start_service postgres

CONTAINER_ID="$(container_id_for postgres)"
if [[ -z "${CONTAINER_ID}" ]]; then
  echo "[wp-knowledge] postgres container id not found" >&2
  exit 1
fi

wait_for_container_ready postgres "${CONTAINER_ID}" "${WAIT_SECONDS}"

if ! wait_for_exec_success postgres "${WAIT_SECONDS}" pg_isready -U postgres -d postgres; then
  echo "[wp-knowledge] postgres is not ready after ${WAIT_SECONDS}s" >&2
  compose logs postgres >&2 || true
  STATUS="$(docker inspect -f '{{if .State.Health}}{{.State.Health.Status}}{{else}}{{.State.Status}}{{end}}' "${CONTAINER_ID}" 2>/dev/null || true)"
  echo "[wp-knowledge] final container status: ${STATUS:-unknown}" >&2
  exit 1
fi

if ! wait_for_exec_success postgres 5 psql -U postgres -d postgres -c 'select 1'; then
  require_exec_success postgres "[wp-knowledge] postgres is accepting connections but simple SQL probe still fails" \
    psql -U postgres -d postgres -c 'select 1'
  exit 1
fi

export WP_KDB_TEST_POSTGRES_URL="${TEST_URL}"

echo "[wp-knowledge] running postgres correctness tests with ${WP_KDB_TEST_POSTGRES_URL} (compose project=${COMPOSE_PROJECT_NAME})"
for test_name in "${POSTGRES_TESTS[@]}"; do
  echo "[wp-knowledge] running postgres_provider::${test_name}"
  cargo test --test postgres_provider "${test_name}" -- --ignored --nocapture
done