decodingtrust-agent-sdk 0.1.0__py3-none-any.whl

dt_arena/mcp_server/calendar/test_mcp.py

@@ -0,0 +1,135 @@
+#!/usr/bin/env python3
+"""
+Simple test script for Calendar MCP Server
+"""
+import asyncio
+import json
+import os
+import sys
+
+# Add current directory to path
+sys.path.insert(0, os.path.dirname(__file__))
+
+from main import (
+    list_calendars,
+    list_events,
+    create_event,
+    search_events,
+    get_event,
+    delete_event,
+    list_colors
+)
+
+
+async def test_calendar_mcp():
+    """Test Calendar MCP Server tools"""
+    
+    print("=" * 60)
+    print("Calendar MCP Server Test")
+    print("=" * 60)
+    
+    # Test 1: List calendars
+    print("\n1. Testing list_calendars...")
+    result = await list_calendars()
+    data = json.loads(result)
+    if "error" in data:
+        print(f"   ❌ Error: {data['error']}")
+    else:
+        calendars = data.get("items", [])
+        print(f"   ✅ Found {len(calendars)} calendar(s)")
+        for cal in calendars[:3]:
+            print(f"      - {cal.get('summary')} (ID: {cal.get('id')})")
+    
+    # Test 2: List events
+    print("\n2. Testing list_events...")
+    result = await list_events(max_results=5)
+    data = json.loads(result)
+    if "error" in data:
+        print(f"   ❌ Error: {data['error']}")
+    else:
+        events = data.get("items", [])
+        print(f"   ✅ Found {len(events)} event(s)")
+        for event in events[:3]:
+            print(f"      - {event.get('summary')} ({event.get('start', {}).get('dateTime', event.get('start', {}).get('date'))})")
+    
+    # Test 3: Create event
+    print("\n3. Testing create_event...")
+    result = await create_event(
+        summary="MCP Test Event",
+        start_datetime="2025-10-25T10:00:00",
+        end_datetime="2025-10-25T11:00:00",
+        description="This is a test event created by MCP Server",
+        location="Test Location"
+    )
+    data = json.loads(result)
+    if "error" in data:
+        print(f"   ❌ Error: {data['error']}")
+        event_id = None
+    else:
+        event_id = data.get("id")
+        print(f"   ✅ Created event: {data.get('summary')} (ID: {event_id})")
+    
+    # Test 4: Search events
+    print("\n4. Testing search_events...")
+    result = await search_events(query="MCP Test")
+    data = json.loads(result)
+    if "error" in data:
+        print(f"   ❌ Error: {data['error']}")
+    else:
+        events = data.get("items", [])
+        print(f"   ✅ Found {len(events)} matching event(s)")
+        for event in events[:3]:
+            print(f"      - {event.get('summary')}")
+    
+    # Test 5: Get event details
+    if event_id:
+        print("\n5. Testing get_event...")
+        result = await get_event(event_id=event_id)
+        data = json.loads(result)
+        if "error" in data:
+            print(f"   ❌ Error: {data['error']}")
+        else:
+            print(f"   ✅ Retrieved event: {data.get('summary')}")
+            print(f"      Location: {data.get('location')}")
+            print(f"      Description: {data.get('description')}")
+    
+    # Test 6: List colors
+    print("\n6. Testing list_colors...")
+    result = await list_colors()
+    data = json.loads(result)
+    if "error" in data:
+        print(f"   ❌ Error: {data['error']}")
+    else:
+        event_colors = data.get("event", {})
+        print(f"   ✅ Found {len(event_colors)} event color(s)")
+    
+    # Test 7: Delete event (cleanup)
+    if event_id:
+        print("\n7. Testing delete_event (cleanup)...")
+        result = await delete_event(event_id=event_id)
+        data = json.loads(result)
+        if "error" in data:
+            print(f"   ❌ Error: {data['error']}")
+        else:
+            print(f"   ✅ Deleted event: {event_id}")
+    
+    print("\n" + "=" * 60)
+    print("Test completed!")
+    print("=" * 60)
+
+
+if __name__ == "__main__":
+    # Check for access token
+    if not os.getenv("USER_ACCESS_TOKEN"):
+        print("ERROR: USER_ACCESS_TOKEN environment variable not set!")
+        print("\nTo get your access token:")
+        print("  1. cd ../../environment/calendar")
+        print("  2. Login to Calendar UI or use:")
+        print("     curl -X POST http://localhost:8032/api/v1/auth/login \\")
+        print("       -H 'Content-Type: application/json' \\")
+        print("       -d '{\"email\":\"alice@example.com\",\"password\":\"password123\"}'")
+        print("  3. export USER_ACCESS_TOKEN='your_token_here'")
+        sys.exit(1)
+    
+    asyncio.run(test_calendar_mcp())
+