echo -e "\n${YELLOW}Cleaning up...${NC}"

# Kill mcpproxy if running

if [ ! -z "$MCPPROXY_PID" ]; then

echo "Stopping mcpproxy (PID: $MCPPROXY_PID)"

kill $MCPPROXY_PID 2>/dev/null || true

# Wait for graceful shutdown with timeout

local count=0

while [ $count -lt 10 ]; do

if ! kill -0 $MCPPROXY_PID 2>/dev/null; then

echo "Process stopped gracefully"

break

fi

sleep 1

count=$((count + 1))

done

# Force kill if still running

if kill -0 $MCPPROXY_PID 2>/dev/null; then

echo "Force killing process"

kill -9 $MCPPROXY_PID 2>/dev/null || true

sleep 1

fi

fi

# Additional cleanup - find any remaining mcpproxy processes

pkill -f "mcpproxy.*serve" 2>/dev/null || true

sleep 1

# Clean up test data

if [ -d "$TEST_DATA_DIR" ]; then

rm -rf "$TEST_DATA_DIR"

fi

# Clean up test results

rm -f "$TEST_RESULTS_FILE"

echo "Cleanup complete"

if [ -f "/tmp/mcpproxy_e2e.log" ]; then

API_KEY=$(grep -o '"api_key": "[^"]*"' "/tmp/mcpproxy_e2e.log" | sed 's/.*"api_key": "\([^"]*\)".*/\1/' | head -1)

if [ ! -z "$API_KEY" ]; then

echo "Extracted API key: ${API_KEY:0:8}..."

fi

fi

local max_attempts=30

local attempt=1

echo "Waiting for server to be ready..."

while [ $attempt -le $max_attempts ]; do

# First extract API key from logs if available

extract_api_key

# Build curl command with CA certificate if it exists, otherwise use insecure for initial check

local curl_cmd="curl -s -f --max-time 5"

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

if [ -f "./test-data/certs/ca.pem" ]; then

curl_cmd="$curl_cmd --cacert ./test-data/certs/ca.pem"

elif [ -f "./certs/ca.pem" ]; then

curl_cmd="$curl_cmd --cacert ./certs/ca.pem"

else

# For initial startup, use insecure until certificates are generated

curl_cmd="$curl_cmd -k"

fi

fi

if [ ! -z "$API_KEY" ]; then

curl_cmd="$curl_cmd -H \"X-API-Key: $API_KEY\""

fi

curl_cmd="$curl_cmd \"${BASE_URL}/api/v1/servers\""

if eval $curl_cmd > /dev/null 2>&1; then

echo "Server is ready!"

# Update CURL_CA_OPTS for subsequent tests if certificates now exist

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

if [ -f "./test-data/certs/ca.pem" ]; then

CURL_CA_OPTS="--cacert ./test-data/certs/ca.pem"

elif [ -f "./certs/ca.pem" ]; then

CURL_CA_OPTS="--cacert ./certs/ca.pem"

fi

fi

return 0

fi

echo "Attempt $attempt/$max_attempts - server not ready yet"

sleep 1

attempt=$((attempt + 1))

done

echo "Server failed to start within $max_attempts seconds"

return 1

local max_attempts=30

local attempt=1

echo "Waiting for everything server to connect and be indexed..."

while [ $attempt -le $max_attempts ]; do

# Check if everything server is connected

local curl_cmd="curl -s --max-time 5 $CURL_CA_OPTS"

if [ ! -z "$API_KEY" ]; then

curl_cmd="$curl_cmd -H \"X-API-Key: $API_KEY\""

fi

curl_cmd="$curl_cmd \"${API_BASE}/servers\""

local response=$(eval $curl_cmd 2>/dev/null)

local connected=$(echo "$response" | jq -r '.data.servers[0].connected // false' 2>/dev/null)

local enabled=$(echo "$response" | jq -r '.data.servers[0].enabled // false' 2>/dev/null)

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

echo "Everything server is connected!"

# Wait a bit more for indexing to complete

sleep 3

return 0

fi

echo "Attempt $attempt/$max_attempts - connected: $connected, enabled: $enabled"

sleep 2

attempt=$((attempt + 1))

done

echo "Everything server failed to connect within $max_attempts attempts"

return 1

local test_name="$1"

local method="$2"

local url="$3"

local expected_status="$4"

local data="$5"

local extra_checks="$6"

log_test "$test_name"

# Clear previous test results

rm -f "$TEST_RESULTS_FILE"

local curl_args=("-s" "-w" "%{http_code}" "-o" "$TEST_RESULTS_FILE" "--max-time" "10")

# Add CA certificate for HTTPS if needed

if [ ! -z "$CURL_CA_OPTS" ]; then

curl_args+=($CURL_CA_OPTS)

fi

# Add API key header if available

if [ ! -z "$API_KEY" ]; then

curl_args+=("-H" "X-API-Key: $API_KEY")

fi

if [ "$method" = "POST" ]; then

curl_args+=("-X" "POST" "-H" "Content-Type: application/json")

if [ ! -z "$data" ]; then

curl_args+=("-d" "$data")

fi

fi

curl_args+=("$url")

local status_code=$(curl "${curl_args[@]}")

if [ "$status_code" = "$expected_status" ]; then

if [ ! -z "$extra_checks" ]; then

if eval "$extra_checks"; then

log_pass "$test_name"

return 0

else

log_fail "$test_name - extra checks failed"

return 1

fi

else

log_pass "$test_name"

return 0

fi

else

if [ "$status_code" = "000" ]; then

log_fail "$test_name - Connection failed (timeout or refused)"

echo "Note: Server may be down or not responding. Check server logs."

else

log_fail "$test_name - Expected status $expected_status, got $status_code"

fi

if [ -f "$TEST_RESULTS_FILE" ] && [ -s "$TEST_RESULTS_FILE" ]; then

echo "Response body:"

cat "$TEST_RESULTS_FILE"

echo

fi

return 1

fi

local test_name="$1"

log_test "$test_name"

# Test SSE endpoint by connecting and reading first few events

# Use Perl for cross-platform timeout (macOS doesn't have timeout command)

local curl_cmd="curl -s -N $CURL_CA_OPTS"

if [ ! -z "$API_KEY" ]; then

curl_cmd="$curl_cmd -H \"X-API-Key: $API_KEY\""

fi

curl_cmd="$curl_cmd \"${BASE_URL}/events\""

# Run with 5 second timeout using Perl

perl -e 'alarm 5; exec @ARGV' sh -c "$curl_cmd" | head -n 10 > "$TEST_RESULTS_FILE" 2>/dev/null || true

if [ -s "$TEST_RESULTS_FILE" ] && grep -q "data:" "$TEST_RESULTS_FILE"; then

log_pass "$test_name"

return 0

else

log_fail "$test_name - No SSE events received"

return 1

fi

local test_name="$1"

log_test "$test_name"

# Test SSE endpoint with API key as query parameter

local sse_url="${BASE_URL}/events"

if [ ! -z "$API_KEY" ]; then

sse_url="${sse_url}?apikey=${API_KEY}"

fi

# Use Perl for cross-platform timeout (macOS doesn't have timeout command)

perl -e 'alarm 5; exec @ARGV' curl -s -N $CURL_CA_OPTS "$sse_url" | head -n 10 > "$TEST_RESULTS_FILE" 2>/dev/null || true

if [ -s "$TEST_RESULTS_FILE" ] && grep -q "data:" "$TEST_RESULTS_FILE"; then

log_pass "$test_name"

return 0

else

log_fail "$test_name - No SSE events received with query parameter"

return 1

fi

local test_name="$1"

log_test "$test_name"

# Test that SSE endpoint establishes proper connection headers

local curl_cmd="curl -s -I --max-time 3 $CURL_CA_OPTS"

if [ ! -z "$API_KEY" ]; then

curl_cmd="$curl_cmd -H \"X-API-Key: $API_KEY\""

fi

curl_cmd="$curl_cmd \"${BASE_URL}/events\""

eval "$curl_cmd" > "$TEST_RESULTS_FILE" 2>/dev/null || true

if [ -s "$TEST_RESULTS_FILE" ] && grep -q "text/event-stream" "$TEST_RESULTS_FILE" && grep -q "Cache-Control: no-cache" "$TEST_RESULTS_FILE"; then

log_pass "$test_name"

return 0

else

log_fail "$test_name - Improper SSE headers"

echo "Headers received:"

cat "$TEST_RESULTS_FILE"

return 1

fi

local test_name="$1"

log_test "$test_name"

# Test SSE with wrong API key (if API key is configured)

if [ -z "$API_KEY" ]; then

log_pass "$test_name (skipped - no API key configured)"

return 0

fi

local status_code=$(curl -s --max-time 5 -w "%{http_code}" -o /dev/null $CURL_CA_OPTS -H "X-API-Key: wrong-api-key" "${BASE_URL}/events")

if [ "$status_code" = "401" ]; then

log_pass "$test_name"

return 0

else

log_fail "$test_name - Expected 401, got $status_code"

return 1

fi