test_api.py

#!/usr/bin/env python3
"""
Test script for the Games MCP API
Run this after deploying to verify everything works correctly
"""

import requests
import json
import sys
from typing import Optional

def test_endpoint(url: str, endpoint: str, params: Optional[dict] = None) -> bool:
    """Test a single endpoint and return success status"""
    try:
        full_url = f"{url}{endpoint}"
        print(f"Testing: {full_url}")
        if params:
            print(f"  Params: {params}")
        
        response = requests.get(full_url, params=params, timeout=10)
        
        if response.status_code == 200:
            print(f"  ✅ Success: {response.status_code}")
            if 'application/json' in response.headers.get('content-type', ''):
                data = response.json()
                print(f"  Response: {json.dumps(data, indent=2)}")
            else:
                print(f"  Response: {response.text[:100]}...")
            return True
        else:
            print(f"  ❌ Failed: {response.status_code}")
            print(f"  Error: {response.text}")
            return False
            
    except requests.exceptions.RequestException as e:
        print(f"  ❌ Request failed: {e}")
        return False

def main():
    # Get URL from command line or use localhost
    if len(sys.argv) > 1:
        base_url = sys.argv[1].rstrip('/')
    else:
        base_url = "http://localhost:8000"
    
    print(f"Testing Games MCP API at: {base_url}")
    print("=" * 50)
    
    tests = [
        # Health check
        ("/health", None),
        
        # Text endpoints
        ("/flip-coin", {"flips": 3}),
        ("/roll-dice", {"dice": 2, "sides": 6}),
        ("/roll-custom", {"expression": "2d6+5"}),
        
        # JSON endpoints
        ("/api/flip-coin", {"flips": 5}),
        ("/api/roll-dice", {"dice": 3, "sides": 20}),
        ("/api/roll-custom", {"expression": "4d6+2"}),
    ]
    
    passed = 0
    total = len(tests)
    
    for endpoint, params in tests:
        if test_endpoint(base_url, endpoint, params):
            passed += 1
        print()
    
    print("=" * 50)
    print(f"Results: {passed}/{total} tests passed")
    
    if passed == total:
        print("🎉 All tests passed! Your API is working correctly.")
        return 0
    else:
        print("❌ Some tests failed. Check the errors above.")
        return 1

if __name__ == "__main__":
    sys.exit(main())