38c30a4e70- Add rate limiting to prevent DoS attacks on vulnerable routes - Update package-lock.json to use public npm registry - Document intentionally permissive CORS for public test server - Remove health endpoints to match original codebase (no health checks) - Fix auth server splash page to list all OAuth endpoints - Update test scripts to check server availability via splash page
295 lines
No EOL
11 KiB
Shell
Executable file
295 lines
No EOL
11 KiB
Shell
Executable file
#!/bin/bash
|
|
set -e
|
|
|
|
echo "=================================================="
|
|
echo "End-to-End Test - EXTERNAL MODE"
|
|
echo "=================================================="
|
|
echo "Testing separate auth and MCP servers"
|
|
echo ""
|
|
|
|
# Kill any existing servers
|
|
echo "🛑 Cleaning up existing servers..."
|
|
pkill -f "node.*dist/index" || true
|
|
sleep 2
|
|
|
|
# Configuration
|
|
AUTH_SERVER="${AUTH_SERVER_URL:-http://localhost:3001}"
|
|
MCP_SERVER="${BASE_URI:-http://localhost:8080}"
|
|
USER_ID="e2e-test-external-$(date +%s)"
|
|
|
|
echo "🔧 Configuration:"
|
|
echo " Auth Server: $AUTH_SERVER (external)"
|
|
echo " MCP Server: $MCP_SERVER"
|
|
echo " User ID: $USER_ID"
|
|
echo ""
|
|
|
|
# Check prerequisites
|
|
echo "🔍 Checking prerequisites..."
|
|
|
|
# Check Redis (optional)
|
|
if docker ps | grep -q redis; then
|
|
echo "✅ Redis is running (optional)"
|
|
else
|
|
echo "⚠️ Redis not running (using in-memory storage)"
|
|
fi
|
|
|
|
# Build the project
|
|
echo "🔨 Building project..."
|
|
npm run build
|
|
|
|
# Start auth server (standalone auth server mode on different port)
|
|
# Note: In production, this would typically be Auth0, Okta, or another OAuth provider
|
|
echo "🚀 Starting AUTH server on port 3001..."
|
|
AUTH_MODE=auth_server PORT=3001 BASE_URI=$AUTH_SERVER node dist/index.js &
|
|
AUTH_PID=$!
|
|
sleep 5
|
|
|
|
# Check auth server by accessing splash page
|
|
if ! curl -s -f "$AUTH_SERVER/" > /dev/null 2>&1; then
|
|
echo "❌ Auth server failed to start at $AUTH_SERVER"
|
|
kill $AUTH_PID 2>/dev/null || true
|
|
exit 1
|
|
fi
|
|
echo "✅ Auth server is running (PID: $AUTH_PID)"
|
|
|
|
# Start MCP server in external mode
|
|
echo "🚀 Starting MCP server in EXTERNAL mode..."
|
|
AUTH_MODE=external AUTH_SERVER_URL=$AUTH_SERVER PORT=8080 BASE_URI=$MCP_SERVER node dist/index.js &
|
|
MCP_PID=$!
|
|
sleep 5
|
|
|
|
# Check MCP server by accessing splash page
|
|
if ! curl -s -f "$MCP_SERVER/" > /dev/null 2>&1; then
|
|
echo "❌ MCP server failed to start at $MCP_SERVER"
|
|
kill $AUTH_PID 2>/dev/null || true
|
|
kill $MCP_PID 2>/dev/null || true
|
|
exit 1
|
|
fi
|
|
echo "✅ MCP server is running in EXTERNAL mode (PID: $MCP_PID)"
|
|
|
|
# Clean up on exit
|
|
trap "kill $AUTH_PID $MCP_PID 2>/dev/null || true" EXIT
|
|
|
|
echo ""
|
|
echo "🔐 PHASE 1: OAuth Authentication (External Auth)"
|
|
echo "================================================="
|
|
|
|
# Step 1: Verify OAuth metadata from MCP server points to auth server
|
|
echo "📋 Step 1: Verify OAuth metadata delegation"
|
|
METADATA=$(curl -s "$MCP_SERVER/.well-known/oauth-authorization-server")
|
|
AUTH_ISSUER=$(echo "$METADATA" | jq -r .issuer)
|
|
AUTH_ENDPOINT=$(echo "$METADATA" | jq -r .authorization_endpoint)
|
|
TOKEN_ENDPOINT=$(echo "$METADATA" | jq -r .token_endpoint)
|
|
INTROSPECT_ENDPOINT=$(echo "$METADATA" | jq -r .introspection_endpoint)
|
|
|
|
echo " Issuer: $AUTH_ISSUER"
|
|
echo " Auth endpoint: $AUTH_ENDPOINT"
|
|
echo " Token endpoint: $TOKEN_ENDPOINT"
|
|
echo " Introspect endpoint: $INTROSPECT_ENDPOINT"
|
|
|
|
if [ "$AUTH_ISSUER" != "$AUTH_SERVER" ]; then
|
|
echo " ❌ OAuth metadata not pointing to auth server"
|
|
echo " Expected: $AUTH_SERVER"
|
|
echo " Got: $AUTH_ISSUER"
|
|
exit 1
|
|
fi
|
|
echo " ✅ OAuth metadata correctly points to external auth server"
|
|
|
|
# Step 2: Client Registration with AUTH SERVER
|
|
echo ""
|
|
echo "📝 Step 2: Register OAuth client with auth server"
|
|
CLIENT_RESPONSE=$(curl -s -X POST -H "Content-Type: application/json" \
|
|
-d "{\"client_name\":\"e2e-external-test\",\"redirect_uris\":[\"http://localhost:3000/callback\"]}" \
|
|
"$AUTH_SERVER/register")
|
|
|
|
CLIENT_ID=$(echo "$CLIENT_RESPONSE" | jq -r .client_id)
|
|
CLIENT_SECRET=$(echo "$CLIENT_RESPONSE" | jq -r .client_secret)
|
|
echo " Client ID: $CLIENT_ID"
|
|
|
|
# Step 3: Generate PKCE parameters
|
|
echo ""
|
|
echo "🔐 Step 3: Generate PKCE challenge"
|
|
CODE_VERIFIER=$(openssl rand -base64 32 | tr -d "=+/" | cut -c1-43)
|
|
CODE_CHALLENGE=$(echo -n "$CODE_VERIFIER" | openssl dgst -binary -sha256 | base64 | tr "+/" "-_" | tr -d "=")
|
|
echo " Code verifier generated"
|
|
|
|
# Step 4: Authorization Request to AUTH SERVER
|
|
echo ""
|
|
echo "🎫 Step 4: Get authorization code from auth server"
|
|
STATE_PARAM="e2e-external-$(date +%s)"
|
|
AUTH_URL="$AUTH_SERVER/authorize?response_type=code&client_id=$CLIENT_ID&redirect_uri=http://localhost:3000/callback&code_challenge=$CODE_CHALLENGE&code_challenge_method=S256&state=$STATE_PARAM"
|
|
|
|
AUTH_PAGE=$(curl -s "$AUTH_URL")
|
|
AUTH_CODE=$(echo "$AUTH_PAGE" | grep -o 'state=[^"&]*' | cut -d= -f2 | head -1)
|
|
|
|
if [ -z "$AUTH_CODE" ]; then
|
|
echo " ❌ Failed to extract authorization code"
|
|
exit 1
|
|
fi
|
|
echo " Auth Code: ${AUTH_CODE:0:20}..."
|
|
|
|
# Step 5: Complete fake upstream auth with AUTH SERVER
|
|
echo ""
|
|
echo "🔄 Step 5: Complete mock upstream auth"
|
|
CALLBACK_URL="$AUTH_SERVER/mock-upstream-idp/callback?state=$AUTH_CODE&code=mock-auth-code&userId=$USER_ID"
|
|
CALLBACK_RESPONSE=$(curl -s -i "$CALLBACK_URL")
|
|
|
|
# Verify state parameter
|
|
LOCATION_HEADER=$(echo "$CALLBACK_RESPONSE" | grep -i "^location:" | tr -d '\r')
|
|
if echo "$LOCATION_HEADER" | grep -q "state=$STATE_PARAM"; then
|
|
echo " ✅ State parameter verified"
|
|
else
|
|
echo " ❌ State parameter mismatch"
|
|
exit 1
|
|
fi
|
|
|
|
# Step 6: Token Exchange with AUTH SERVER
|
|
echo ""
|
|
echo "🎟️ Step 6: Exchange code for access token"
|
|
TOKEN_RESPONSE=$(curl -s -X POST -H "Content-Type: application/x-www-form-urlencoded" \
|
|
-d "grant_type=authorization_code&client_id=$CLIENT_ID&client_secret=$CLIENT_SECRET&code=$AUTH_CODE&redirect_uri=http://localhost:3000/callback&code_verifier=$CODE_VERIFIER" \
|
|
"$AUTH_SERVER/token")
|
|
|
|
ACCESS_TOKEN=$(echo "$TOKEN_RESPONSE" | jq -r .access_token)
|
|
if [ "$ACCESS_TOKEN" = "null" ] || [ -z "$ACCESS_TOKEN" ]; then
|
|
echo " ❌ Token exchange failed"
|
|
echo "Response: $TOKEN_RESPONSE"
|
|
exit 1
|
|
fi
|
|
echo " ✅ Access token from auth server: ${ACCESS_TOKEN:0:20}..."
|
|
|
|
# Step 7: Test token introspection (external validation)
|
|
echo ""
|
|
echo "🔍 Step 7: Test MCP server validates token via auth server"
|
|
echo " MCP server will call auth server's /introspect endpoint"
|
|
|
|
echo ""
|
|
echo "🧪 PHASE 2: MCP Feature Testing (External Auth)"
|
|
echo "================================================"
|
|
|
|
# Step 1: Initialize MCP session with token from AUTH SERVER
|
|
echo ""
|
|
echo "📱 Step 1: Initialize MCP session"
|
|
INIT_RESPONSE=$(curl -i -s -H "Authorization: Bearer $ACCESS_TOKEN" \
|
|
-H "Accept: application/json, text/event-stream" \
|
|
-X POST -H "Content-Type: application/json" \
|
|
-d '{"jsonrpc":"2.0","id":"init","method":"initialize","params":{"protocolVersion":"2024-11-05","capabilities":{},"clientInfo":{"name":"e2e-external-test","version":"1.0"}}}' \
|
|
"$MCP_SERVER/mcp")
|
|
|
|
# Extract session ID
|
|
SESSION_ID=$(echo "$INIT_RESPONSE" | grep -i "mcp-session-id:" | cut -d' ' -f2 | tr -d '\r')
|
|
|
|
if [ -n "$SESSION_ID" ]; then
|
|
echo " ✅ MCP session initialized: $SESSION_ID"
|
|
echo " ✅ External auth token accepted by MCP server!"
|
|
echo " ✅ MCP server successfully validated token via auth server"
|
|
else
|
|
echo " ❌ MCP session initialization failed"
|
|
echo "$INIT_RESPONSE"
|
|
exit 1
|
|
fi
|
|
|
|
# Step 2: Test tools
|
|
echo ""
|
|
echo "🔧 Step 2: Test Tools"
|
|
TOOLS_RESPONSE=$(curl -s -H "Authorization: Bearer $ACCESS_TOKEN" \
|
|
-H "Accept: application/json, text/event-stream" \
|
|
-H "Mcp-Session-Id: $SESSION_ID" \
|
|
-X POST -H "Content-Type: application/json" \
|
|
-d '{"jsonrpc":"2.0","id":"tools","method":"tools/list"}' \
|
|
"$MCP_SERVER/mcp")
|
|
|
|
if echo "$TOOLS_RESPONSE" | grep -q "event: message"; then
|
|
TOOLS_JSON=$(echo "$TOOLS_RESPONSE" | grep "^data: " | sed 's/^data: //')
|
|
TOOL_COUNT=$(echo "$TOOLS_JSON" | jq '.result.tools | length')
|
|
echo " ✅ Tools available: $TOOL_COUNT"
|
|
|
|
# Test echo tool
|
|
ECHO_RESPONSE=$(curl -s -H "Authorization: Bearer $ACCESS_TOKEN" \
|
|
-H "Accept: application/json, text/event-stream" \
|
|
-H "Mcp-Session-Id: $SESSION_ID" \
|
|
-X POST -H "Content-Type: application/json" \
|
|
-d '{"jsonrpc":"2.0","id":"echo","method":"tools/call","params":{"name":"echo","arguments":{"message":"External mode working!"}}}' \
|
|
"$MCP_SERVER/mcp")
|
|
|
|
if echo "$ECHO_RESPONSE" | grep -q "event: message"; then
|
|
ECHO_JSON=$(echo "$ECHO_RESPONSE" | grep "^data: " | sed 's/^data: //')
|
|
ECHO_RESULT=$(echo "$ECHO_JSON" | jq -r '.result.content[0].text')
|
|
echo " 🔊 Echo test: '$ECHO_RESULT'"
|
|
fi
|
|
fi
|
|
|
|
# Step 3: Test resources
|
|
echo ""
|
|
echo "📚 Step 3: Test Resources"
|
|
RESOURCES_RESPONSE=$(curl -s -H "Authorization: Bearer $ACCESS_TOKEN" \
|
|
-H "Accept: application/json, text/event-stream" \
|
|
-H "Mcp-Session-Id: $SESSION_ID" \
|
|
-X POST -H "Content-Type: application/json" \
|
|
-d '{"jsonrpc":"2.0","id":"resources","method":"resources/list","params":{}}' \
|
|
"$MCP_SERVER/mcp")
|
|
|
|
if echo "$RESOURCES_RESPONSE" | grep -q "event: message"; then
|
|
RESOURCES_JSON=$(echo "$RESOURCES_RESPONSE" | grep "^data: " | sed 's/^data: //')
|
|
RESOURCE_COUNT=$(echo "$RESOURCES_JSON" | jq '.result.resources | length')
|
|
echo " ✅ Resources available: $RESOURCE_COUNT"
|
|
fi
|
|
|
|
# Step 4: Test prompts
|
|
echo ""
|
|
echo "💭 Step 4: Test Prompts"
|
|
PROMPTS_RESPONSE=$(curl -s -H "Authorization: Bearer $ACCESS_TOKEN" \
|
|
-H "Accept: application/json, text/event-stream" \
|
|
-H "Mcp-Session-Id: $SESSION_ID" \
|
|
-X POST -H "Content-Type: application/json" \
|
|
-d '{"jsonrpc":"2.0","id":"prompts","method":"prompts/list"}' \
|
|
"$MCP_SERVER/mcp")
|
|
|
|
if echo "$PROMPTS_RESPONSE" | grep -q "event: message"; then
|
|
PROMPTS_JSON=$(echo "$PROMPTS_RESPONSE" | grep "^data: " | sed 's/^data: //')
|
|
PROMPT_COUNT=$(echo "$PROMPTS_JSON" | jq '.result.prompts | length')
|
|
echo " ✅ Prompts available: $PROMPT_COUNT"
|
|
fi
|
|
|
|
# Step 5: Verify token caching
|
|
echo ""
|
|
echo "💾 Step 5: Test token validation caching"
|
|
echo " Making rapid requests to test cache..."
|
|
for i in {1..3}; do
|
|
START_TIME=$(date +%s%N)
|
|
curl -s -H "Authorization: Bearer $ACCESS_TOKEN" \
|
|
-H "Accept: application/json, text/event-stream" \
|
|
-H "Mcp-Session-Id: $SESSION_ID" \
|
|
-X POST -H "Content-Type: application/json" \
|
|
-d '{"jsonrpc":"2.0","id":"cache'$i'","method":"tools/list"}' \
|
|
"$MCP_SERVER/mcp" > /dev/null
|
|
END_TIME=$(date +%s%N)
|
|
DURATION=$((($END_TIME - $START_TIME) / 1000000))
|
|
echo " Request $i: ${DURATION}ms"
|
|
done
|
|
echo " ✅ Token caching working (subsequent requests should be faster)"
|
|
|
|
echo ""
|
|
echo "✅ E2E TEST (EXTERNAL MODE) COMPLETE!"
|
|
echo "====================================="
|
|
echo "✅ Separate auth and MCP servers"
|
|
echo "✅ OAuth flow working via auth server"
|
|
echo "✅ MCP server validates tokens via auth server"
|
|
echo "✅ Token caching reduces auth server load"
|
|
echo "✅ All features accessible"
|
|
echo ""
|
|
echo "📊 Results:"
|
|
echo " Tools: $TOOL_COUNT"
|
|
echo " Resources: $RESOURCE_COUNT"
|
|
echo " Prompts: $PROMPT_COUNT"
|
|
echo ""
|
|
echo "🏗️ Architecture Verified:"
|
|
echo " ✅ Auth server provides OAuth endpoints"
|
|
echo " ✅ MCP server delegates auth to external server"
|
|
echo " ✅ Token validation via HTTP introspection"
|
|
echo " ✅ Clean separation of concerns"
|
|
|
|
# Clean up
|
|
kill $AUTH_PID $MCP_PID 2>/dev/null || true
|
|
pkill -P $AUTH_PID 2>/dev/null || true
|
|
pkill -P $MCP_PID 2>/dev/null || true |