Chris Coutinho
95 lines
2.9 KiB
Python
95 lines
2.9 KiB
Python
#!/usr/bin/env python3
|
|
"""Test script to verify OAuth MCP tools work correctly.
|
|
|
|
This script connects to the OAuth MCP server and tests tool execution.
|
|
Note: This currently requires a valid OAuth token, which must be obtained
|
|
through the browser-based OAuth flow.
|
|
"""
|
|
|
|
import asyncio
|
|
import sys
|
|
|
|
from mcp import ClientSession
|
|
from mcp.client.streamable_http import streamablehttp_client
|
|
|
|
|
|
async def test_oauth_mcp_tools():
|
|
"""Test OAuth MCP server tools."""
|
|
print("Connecting to OAuth MCP server on port 8001...")
|
|
|
|
streamable_context = streamablehttp_client("http://127.0.0.1:8001/mcp")
|
|
session_context = None
|
|
|
|
try:
|
|
read_stream, write_stream, _ = await streamable_context.__aenter__()
|
|
session_context = ClientSession(read_stream, write_stream)
|
|
session = await session_context.__aenter__()
|
|
|
|
print("Initializing session...")
|
|
await session.initialize()
|
|
print("✓ Session initialized successfully")
|
|
|
|
# List available tools
|
|
print("\nListing available tools...")
|
|
result = await session.list_tools()
|
|
print(f"✓ Found {len(result.tools)} tools")
|
|
|
|
for tool in result.tools[:5]: # Show first 5
|
|
print(f" - {tool.name}: {tool.description}")
|
|
|
|
if len(result.tools) > 5:
|
|
print(f" ... and {len(result.tools) - 5} more")
|
|
|
|
# Try to call a simple tool
|
|
print("\nTesting tool execution...")
|
|
print("Note: Tool execution will fail without a valid OAuth token")
|
|
print(" (OAuth token must be obtained through browser flow)")
|
|
|
|
try:
|
|
# Try to list tables (this will fail without OAuth token)
|
|
response = await session.call_tool("nc_tables_list_tables", {})
|
|
print(f"✓ Tool executed successfully: {response}")
|
|
except Exception as e:
|
|
print(f"✗ Tool execution failed (expected without OAuth token): {e}")
|
|
print("\nTo use OAuth tools, you need to:")
|
|
print(" 1. Implement the browser-based OAuth authorization flow")
|
|
print(" 2. Obtain an access token from Nextcloud OIDC")
|
|
print(" 3. Include the token in the Authorization header")
|
|
|
|
return True
|
|
|
|
except Exception as e:
|
|
print(f"✗ Error: {e}")
|
|
import traceback
|
|
|
|
traceback.print_exc()
|
|
return False
|
|
|
|
finally:
|
|
# Clean up
|
|
if session_context is not None:
|
|
try:
|
|
await session_context.__aexit__(None, None, None)
|
|
except Exception:
|
|
pass
|
|
|
|
try:
|
|
await streamable_context.__aexit__(None, None, None)
|
|
except Exception:
|
|
pass
|
|
|
|
|
|
if __name__ == "__main__":
|
|
print("OAuth MCP Server Tool Test")
|
|
print("=" * 50)
|
|
|
|
success = asyncio.run(test_oauth_mcp_tools())
|
|
|
|
print("\n" + "=" * 50)
|
|
if success:
|
|
print("✓ Test completed (tools accessible)")
|
|
sys.exit(0)
|
|
else:
|
|
print("✗ Test failed")
|
|
sys.exit(1)
|