gpt4free

#!/usr/bin/env python
"""Test HTTP MCP server functionality

This script tests the HTTP transport for the MCP server.
"""

import asyncio
import json
from g4f.mcp.server import MCPServer, MCPRequest


async def test_http_server():
    """Test HTTP server methods"""
    server = MCPServer()

    print("Testing HTTP MCP Server Functionality")
    print("=" * 70)

    # Test that server can be initialized
    print("\n✓ Server initialized successfully")
    print(f"  Server: {server.server_info['name']}")
    print(f"  Version: {server.server_info['version']}")

    # Test that run_http method exists
    if hasattr(server, "run_http"):
        print("\n✓ HTTP transport method (run_http) available")
        print(f"  Signature: run_http(host, port)")
    else:
        print("\n✗ HTTP transport method not found")
        return

    # Test request handling (same for both transports)
    print("\n✓ Testing request handling...")

    init_request = MCPRequest(jsonrpc="2.0", id=1, method="initialize", params={})
    response = await server.handle_request(init_request)

    if response.result and response.result.get("protocolVersion"):
        print(f"  Protocol Version: {response.result['protocolVersion']}")
        print("  ✓ Request handling works correctly")

    print("\n" + "=" * 70)
    print("HTTP MCP Server Tests Passed!")
    print("\nTo start HTTP server:")
    print("  g4f mcp --http --port 8765")
    print("\nHTTP endpoints:")
    print("  POST http://localhost:8765/mcp - MCP JSON-RPC endpoint")
    print("  GET  http://localhost:8765/health - Health check")
    print("\nExample HTTP request:")
    print("  curl -X POST http://localhost:8765/mcp \\")
    print('    -H "Content-Type: application/json" \\')
    print('    -d \'{"jsonrpc":"2.0","id":1,"method":"initialize","params":{}}\'')


if __name__ == "__main__":
    asyncio.run(test_http_server())