test_full_system.py

#!/usr/bin/env python3
"""
Full system test for the MCP Chatbot application
Tests both backend API and verifies frontend is accessible
"""

import requests
import time
import json
import sys

def test_backend():
    """Test backend API endpoints"""
    print("🔧 Testing Backend API...")
    
    # Test 1: Ping endpoint
    try:
        response = requests.get("http://localhost:8000/ping", timeout=5)
        if response.status_code == 200:
            print("✅ Ping endpoint: PASS")
            print(f"   Response: {response.json()}")
        else:
            print(f"❌ Ping endpoint: FAIL (status {response.status_code})")
            return False
    except Exception as e:
        print(f"❌ Ping endpoint: FAIL ({e})")
        return False
    
    # Test 2: API Documentation
    try:
        response = requests.get("http://localhost:8000/docs", timeout=5)
        if response.status_code == 200:
            print("✅ API Documentation: PASS")
        else:
            print(f"❌ API Documentation: FAIL (status {response.status_code})")
    except Exception as e:
        print(f"❌ API Documentation: FAIL ({e})")
    
    # Test 3: OpenAPI Spec
    try:
        response = requests.get("http://localhost:8000/openapi.json", timeout=5)
        if response.status_code == 200:
            openapi_data = response.json()
            endpoints = list(openapi_data.get("paths", {}).keys())
            print(f"✅ OpenAPI Spec: PASS ({len(endpoints)} endpoints)")
            print(f"   Available endpoints: {', '.join(endpoints)}")
        else:
            print(f"❌ OpenAPI Spec: FAIL (status {response.status_code})")
    except Exception as e:
        print(f"❌ OpenAPI Spec: FAIL ({e})")
    
    # Test 4: Chat endpoint (simple test)
    try:
        payload = {
            "message": "Hello, this is a test message",
            "conversation_id": "test-conv-123"
        }
        response = requests.post(
            "http://localhost:8000/chat", 
            json=payload,
            timeout=10,
            headers={"Content-Type": "application/json"}
        )
        print(f"✅ Chat endpoint: Response status {response.status_code}")
        if response.status_code == 200:
            print("   Chat endpoint is responding")
        else:
            print(f"   Response: {response.text[:200]}...")
    except Exception as e:
        print(f"❌ Chat endpoint: FAIL ({e})")
    
    return True

def test_frontend():
    """Test frontend accessibility"""
    print("\n🎨 Testing Frontend...")
    
    # Test frontend accessibility
    try:
        response = requests.get("http://localhost:5173", timeout=5)
        if response.status_code == 200:
            html_content = response.text
            if "react" in html_content.lower() or "vite" in html_content.lower():
                print("✅ Frontend: PASS (React app accessible)")
                return True
            else:
                print("❌ Frontend: FAIL (unexpected content)")
                return False
        else:
            print(f"❌ Frontend: FAIL (status {response.status_code})")
            return False
    except Exception as e:
        print(f"❌ Frontend: FAIL ({e})")
        return False

def test_cors():
    """Test CORS configuration"""
    print("\n🌐 Testing CORS Configuration...")
    
    try:
        # Simulate a CORS preflight request
        response = requests.options(
            "http://localhost:8000/ping",
            headers={
                "Origin": "http://localhost:5173",
                "Access-Control-Request-Method": "POST",
                "Access-Control-Request-Headers": "Content-Type"
            },
            timeout=5
        )
        
        cors_headers = {
            "access-control-allow-origin": response.headers.get("access-control-allow-origin"),
            "access-control-allow-methods": response.headers.get("access-control-allow-methods"),
            "access-control-allow-headers": response.headers.get("access-control-allow-headers")
        }
        
        if cors_headers["access-control-allow-origin"]:
            print("✅ CORS: PASS (properly configured)")
            print(f"   Allowed origins: {cors_headers['access-control-allow-origin']}")
        else:
            print("❌ CORS: FAIL (not properly configured)")
            
    except Exception as e:
        print(f"❌ CORS: FAIL ({e})")

def main():
    print("🚀 Starting Full System Test for MCP Chatbot")
    print("=" * 50)
    
    backend_ok = test_backend()
    frontend_ok = test_frontend()
    test_cors()
    
    print("\n" + "=" * 50)
    print("📊 Test Summary:")
    print(f"   Backend: {'✅ PASS' if backend_ok else '❌ FAIL'}")
    print(f"   Frontend: {'✅ PASS' if frontend_ok else '❌ FAIL'}")
    
    if backend_ok and frontend_ok:
        print("\n🎉 All systems are GO! Your ChatGPT-inspired MCP Chatbot is ready!")
        print("   Frontend: http://localhost:5173")
        print("   Backend API: http://localhost:8000")
        print("   API Docs: http://localhost:8000/docs")
        return 0
    else:
        print("\n⚠️  Some issues detected. Check the logs above.")
        return 1

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