import asyncio
import os
import sys
import re
import pytest
from mcp import ClientSession, StdioServerParameters
from mcp.client.stdio import stdio_client
# Ensure functionality of imports if run directly
project_root = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.append(project_root)
@pytest.mark.asyncio
async def test_agent_vs_computer():
print("Starting E2E Test: Agent vs Computer...")
# Define server parameters
# Running the module directly
server_params = StdioServerParameters(
command=sys.executable,
args=["-m", "src.mcp_server"],
env=os.environ.copy() # Pass current env
)
async with stdio_client(server_params) as (read, write):
async with ClientSession(read, write) as session:
# Initialize
await session.initialize()
# List tools to verify connection
tools = await session.list_tools()
assert len(tools.tools) > 0, "No tools found on server"
# --- Step 1: Create Game ---
print("\n[Step 1] Creating Game (Agent White vs Computer)...")
result = await session.call_tool(
"createGame",
arguments={"type": "computer", "color": "white", "difficulty": 1}
)
# Parse output
content_text = result.content[0].text
print(f"Server Response:\n{content_text}")
# Extract Game ID
match = re.search(r"Game ID: ([a-fA-F0-9\-]+)", content_text)
assert match, "Could not find Game ID in server response"
game_id = match.group(1)
print(f"Game ID found: {game_id}")
# --- Step 2: Make Move (e2e4) ---
print("\n[Step 2] Making Move 'e2e4'...")
result = await session.call_tool(
"finishTurn",
arguments={"game_id": game_id, "move": "e2e4"}
)
content_text = result.content[0].text
print(f"Server Response:\n{content_text}")
assert "Waiting for Computer" in content_text, "Expected 'Waiting for Computer' message"
# --- Step 3: Wait for Computer Move ---
print("\n[Step 3] Waiting for Computer Turn...")
result = await session.call_tool(
"waitForNextTurn",
arguments={"game_id": game_id}
)
content_text = result.content[0].text
print(f"Server Response:\n{content_text}")
assert "Timeout" not in content_text, "Timed out waiting for computer"
# Verify board state
assert "**FEN**" in content_text, "Did not receive board state"
# Additional check: Parsing FEN to see if black moved?
# Basic check is fine for now.
print("\nE2E Test COMPLETED SUCCESSFULLY.")
