diff --git a/scripts/generate_test_map.py b/scripts/generate_test_map.py index e4d88de..89e0b27 100644 --- a/scripts/generate_test_map.py +++ b/scripts/generate_test_map.py @@ -8,7 +8,6 @@ sys.path.append(os.path.join(os.path.dirname(__file__), "..", "src")) from strava_mcp_server.strava_client import StravaClient from mcp.server.fastmcp import Context -from mcp.types import TextContent, EmbeddedResource class MockContext(Context): @@ -60,6 +59,7 @@ async def generate(): class MockMCP: def __init__(self): self.tools = {} + self.resources = {} def tool(self): def decorator(func): @@ -68,6 +68,13 @@ async def generate(): return decorator + def resource(self, path): + def decorator(func): + self.resources[path] = func + return func + + return decorator + mcp = MockMCP() # Mock StravaClient that returns our mock activity and streams @@ -106,35 +113,10 @@ async def generate(): client = MockClient() register(mcp, client) - get_activity_map = mcp.tools["get_activity_map"] - ctx = MockContext() - - print("Generating mock map HTML...") - result = await get_activity_map(ctx, 99999) - - html_text = None - content_list = result.content if hasattr(result, "content") else result - for item in content_list: - if isinstance(item, TextContent): - if "" in item.text: - html_text = item.text - break - elif '