set -e

SCRIPT_DIR=$( cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )

PROJECT_ROOT="$(dirname "$SCRIPT_DIR")"

COMPOSE_FILE="$SCRIPT_DIR/docker-compose.test.yml"

DEBUG_MODE=false

STOP_MODE=false

SERVICES=()

PYTEST_ARGS=()

while [[ $# -gt 0 ]]; do

case $1 in

--debug)

DEBUG_MODE=true

shift

;;

--stop)

STOP_MODE=true

shift

;;

--postgres)

SERVICES+=("postgres")

shift

;;

--mysql)

SERVICES+=("mysql")

shift

;;

*)

# Preserve other arguments for pytest

PYTEST_ARGS+=("$1")

shift

;;

esac

if [[ "$STOP_MODE" == "true" ]]; then

echo "Stopping test databases..."

docker compose -f "$COMPOSE_FILE" down

exit 0

if [[ ${#SERVICES[@]} -eq 0 ]]; then

SERVICES=("postgres" "mysql")

cleanup() {

echo "Stopping test databases..."

docker compose -f "$COMPOSE_FILE" down

}

echo "Starting/Verifying databases: ${SERVICES[*]}..."

docker compose -f "$COMPOSE_FILE" up -d --wait "${SERVICES[@]}"

for service in "${SERVICES[@]}"; do

if [[ "$service" == "postgres" ]]; then

export POSTGRES_TEST_DSN="postgresql+asyncpg://a2a:a2a_password@localhost:5432/a2a_test"

elif [[ "$service" == "mysql" ]]; then

export MYSQL_TEST_DSN="mysql+aiomysql://a2a:a2a_password@localhost:3306/a2a_test"

fi

if [[ "$DEBUG_MODE" == "true" ]]; then

echo "---------------------------------------------------"

echo "Debug mode enabled. Databases are running."

echo "You can connect to them using the following DSNs."

echo ""

echo "Run the following commands to set up your environment:"

echo ""

[[ -n "$POSTGRES_TEST_DSN" ]] && echo "export POSTGRES_TEST_DSN=\"$POSTGRES_TEST_DSN\""

[[ -n "$MYSQL_TEST_DSN" ]] && echo "export MYSQL_TEST_DSN=\"$MYSQL_TEST_DSN\""

echo ""

echo "---------------------------------------------------"

echo "Run ./scripts/run_integration_tests.sh --stop to shut databases down."

exit 0

trap cleanup EXIT

echo "Running integration tests..."

cd "$PROJECT_ROOT"

uv run pytest -v \

tests/server/tasks/test_database_task_store.py \

tests/server/tasks/test_database_push_notification_config_store.py \

"${PYTEST_ARGS[@]}"