test.sh

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

# Local E2E Test Runner
#
# Runs test suites in full Docker isolation — each run gets its own Stellar
# node, PostgreSQL, and service instances. Tests use current local repo source
# code (mounted read-only). Does not require up.sh.
#
# Usage:
#   ./test.sh e2e           # Payment flow
#   ./test.sh governance    # UC2 governance flows
#   ./test.sh uc2           # Manual UC2 user flow
#   ./test.sh all           # All three in parallel
#   ./test.sh clean         # Remove all test containers/volumes

SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
BASE_DIR="${BASE_DIR:-$(dirname "$SCRIPT_DIR")}"

# Stellar quickstart pin (image tag + local-network protocol) — single source of
# truth, shared with every compose file, infra-up.sh, provider.ts and the drift
# alarm. Exported so the docker compose invocations below interpolate it.
set -a
# shellcheck source=stellar-pin.env
. "$SCRIPT_DIR/stellar-pin.env"
set +a

RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
NC='\033[0m'

info()  { echo -e "${GREEN}[test]${NC} $*"; }
warn()  { echo -e "${YELLOW}[test]${NC} $*"; }
error() { echo -e "${RED}[test]${NC} $*" >&2; exit 1; }

usage() {
  echo "Usage: $0 <suite>"
  echo ""
  echo "Suites:"
  echo "  e2e           Payment flow (8 steps)"
  echo "  otel          Payment flow + OTEL trace verification (16 checks)"
  echo "  governance    UC2 governance flows (40+ checks)"
  echo "  lifecycle     Full lifecycle (deploy → payment → remove)"
  echo "  multi-asset   Multi-asset UC6 (one council, XLM + USDC channels)"
  echo "  pos-instant   POS crypto instant payment (temp P256 hop)"
  echo "  playwright    Full UI flow (Freighter + 3 frontends, ~6min)"
  echo "  invite-gate   Invite-only allowlist gate (blocked + allowed wallets)"
  echo "  clean         Remove all test containers and volumes"
  exit 1
}

SUITE="${1:-}"
[ -z "$SUITE" ] && usage

# Ensure WASMs are built
ensure_wasms() {
  WASM_DIR="${WASM_DIR:-${BASE_DIR}/soroban-core/target/wasm32v1-none/release}"
  if [ ! -f "$WASM_DIR/channel_auth_contract.wasm" ] || [ ! -f "$WASM_DIR/privacy_channel.wasm" ]; then
    local soroban_path="${SOROBAN_CORE_PATH:-${BASE_DIR}/soroban-core}"
    [ -d "$soroban_path" ] || error "soroban-core not found at $soroban_path. Set SOROBAN_CORE_PATH or BASE_DIR."
    info "Building contracts..."
    (cd "$soroban_path" && stellar contract build) || error "Contract build failed"
    info "Contracts built."
  fi
  export WASM_DIR
}

# Run a single test suite
run_suite() {
  local suite=$1
  # Not local — must be visible to the EXIT trap after run_suite returns
  project_name="moonlight-test-${suite}-$(date +%s | tail -c 9)"

  local compose_file="$SCRIPT_DIR/docker-compose.${suite}.yml"
  if [ ! -f "$compose_file" ]; then
    error "No compose file for suite '$suite' at $compose_file"
  fi

  # Wipe and recreate traces directory so Badger never sees a stale lock
  # from a previous (possibly crashed) run. Data is ephemeral per-run.
  rm -rf "$SCRIPT_DIR/.traces/$suite"
  mkdir -p "$SCRIPT_DIR/.traces/$suite"
  chmod 777 "$SCRIPT_DIR/.traces/$suite"

  info "Running '$suite' (project: $project_name)..."

  # Cleanup on exit or interrupt
  cleanup() {
    info "Cleaning up $project_name..."
    docker compose -f "$compose_file" -p "$project_name" down -v --remove-orphans 2>/dev/null || true
  }
  trap cleanup EXIT INT TERM

  WASM_DIR="${WASM_DIR:-}" \
  PROVIDER_PLATFORM_PATH="${PROVIDER_PLATFORM_PATH:-${BASE_DIR}/provider-platform}" \
  COUNCIL_PLATFORM_PATH="${COUNCIL_PLATFORM_PATH:-${BASE_DIR}/council-platform}" \
  PAY_PLATFORM_PATH="${PAY_PLATFORM_PATH:-${BASE_DIR}/pay-platform}" \
  MOONLIGHT_PAY_PATH="${MOONLIGHT_PAY_PATH:-${BASE_DIR}/moonlight-pay}" \
  PROVIDER_CONSOLE_PATH="${PROVIDER_CONSOLE_PATH:-${BASE_DIR}/provider-console}" \
  COUNCIL_CONSOLE_PATH="${COUNCIL_CONSOLE_PATH:-${BASE_DIR}/council-console}" \
  STELLAR_QUICKSTART_IMAGE="$STELLAR_QUICKSTART_IMAGE" \
  STELLAR_PROTOCOL_VERSION="$STELLAR_PROTOCOL_VERSION" \
  docker compose -f "$compose_file" -p "$project_name" up -d

  # Stream test-runner logs and wait for it to finish
  info "Waiting for test-runner to complete..."
  docker compose -f "$compose_file" -p "$project_name" logs -f test-runner 2>&1 &
  local logs_pid=$!

  # Wait for the test-runner container to exit
  local container_name="${project_name}-test-runner-1"
  local exit_code=1
  while true; do
    local state
    state=$(docker inspect --format '{{.State.Status}}' "$container_name" 2>/dev/null || echo "not_found")
    case "$state" in
      exited)
        exit_code=$(docker inspect --format '{{.State.ExitCode}}' "$container_name" 2>/dev/null || echo "1")
        break
        ;;
      not_found|dead)
        break
        ;;
    esac
    sleep 3
  done

  kill "$logs_pid" 2>/dev/null || true
  wait "$logs_pid" 2>/dev/null || true
  cleanup
  trap - EXIT INT TERM

  if [ "$exit_code" -eq 0 ]; then
    info "'$suite' passed."
  else
    error "'$suite' failed (exit $exit_code)."
  fi
  return "$exit_code"
}

# Clean up any lingering test projects
clean_all() {
  info "Cleaning up all test projects..."
  for project in $(docker compose ls --format json 2>/dev/null | grep -o '"moonlight-test-[^"]*"' | tr -d '"'); do
    info "  Removing $project..."
    docker compose -p "$project" down -v --remove-orphans 2>/dev/null || true
  done
  info "Clean."
}

case "$SUITE" in
  playwright|invite-gate)
    # No WASMs needed — council-console downloads them during its Docker build
    "$SCRIPT_DIR/setup-freighter.sh"
    run_suite "$SUITE"
    ;;

  e2e|otel|governance|lifecycle|multi-asset|pos-instant)
    ensure_wasms
    run_suite "$SUITE"
    ;;

  all)
    ensure_wasms
    info "Running all suites in parallel..."
    pids=()
    results=()

    for s in e2e otel governance lifecycle multi-asset pos-instant; do
      # Intentional: each subshell gets its own trap handler from run_suite,
      # so cleanup runs independently per suite when it exits or is interrupted.
      (run_suite "$s") &
      pids+=($!)
    done

    all_passed=true
    for i in "${!pids[@]}"; do
      suite_names=(e2e otel governance lifecycle multi-asset pos-instant)
      if wait "${pids[$i]}"; then
        results+=("${suite_names[$i]}: passed")
      else
        results+=("${suite_names[$i]}: FAILED")
        all_passed=false
      fi
    done

    echo ""
    info "=== Results ==="
    for r in "${results[@]}"; do
      info "  $r"
    done

    if [ "$all_passed" = true ]; then
      info "All suites passed."
    else
      error "Some suites failed."
    fi
    ;;

  clean)
    clean_all
    ;;

  *)
    usage
    ;;
esac