#!/bin/bash

# Run SDK tests in a Docker container

# This helps catch Docker-specific issues like #406

#

# Usage:

# ./scripts/test-docker.sh [unit|e2e|all]

#

# Examples:

# ./scripts/test-docker.sh unit # Run unit tests only

# ANTHROPIC_API_KEY=sk-... ./scripts/test-docker.sh e2e # Run e2e tests

# ANTHROPIC_API_KEY=sk-... ./scripts/test-docker.sh all # Run all tests

set -e

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"

PROJECT_DIR="$(dirname "$SCRIPT_DIR")"

cd "$PROJECT_DIR"

usage() {

echo "Usage: $0 [unit|e2e|all]"

echo ""

echo "Commands:"

echo " unit - Run unit tests only (no API key needed)"

echo " e2e - Run e2e tests (requires ANTHROPIC_API_KEY)"

echo " all - Run both unit and e2e tests"

echo ""

echo "Examples:"

echo " $0 unit"

echo " ANTHROPIC_API_KEY=sk-... $0 e2e"

exit 1

}

echo "Building Docker test image..."

docker build -f Dockerfile.test -t claude-sdk-test .

case "${1:-unit}" in

unit)

echo ""

echo "Running unit tests in Docker..."

docker run --rm claude-sdk-test \

python -m pytest tests/ -v

;;

e2e)

if [ -z "$ANTHROPIC_API_KEY" ]; then

echo "Error: ANTHROPIC_API_KEY environment variable is required for e2e tests"

echo ""

echo "Usage: ANTHROPIC_API_KEY=sk-... $0 e2e"

exit 1

fi

echo ""

echo "Running e2e tests in Docker..."

docker run --rm -e ANTHROPIC_API_KEY \

claude-sdk-test python -m pytest e2e-tests/ -v -m e2e

;;

all)

echo ""

echo "Running unit tests in Docker..."

docker run --rm claude-sdk-test \

python -m pytest tests/ -v

echo ""

if [ -n "$ANTHROPIC_API_KEY" ]; then

echo "Running e2e tests in Docker..."

docker run --rm -e ANTHROPIC_API_KEY \

claude-sdk-test python -m pytest e2e-tests/ -v -m e2e

else

echo "Skipping e2e tests (ANTHROPIC_API_KEY not set)"

fi

;;

*)

usage

;;

esac

echo ""

echo "Done!"