Bill
a7c87128ef
All checks were successful
Build and Push Docker Image / build (push) Successful in 8s
The MCP tool approach was impractical because it required the LLM to generate large base64 strings token-by-token, causing timeouts. Changes: - Remove upload_attachment MCP tool - Add POST /api/v1/attachments endpoint for multipart/form-data uploads - Update proxy documentation to show both endpoints - Uses existing GristClient.upload_attachment() method - Requires write permission in session token
101 lines
2.8 KiB
Python
101 lines
2.8 KiB
Python
import pytest
|
|
from mcp.types import ListToolsRequest
|
|
from grist_mcp.server import create_server
|
|
from grist_mcp.config import load_config
|
|
from grist_mcp.auth import Authenticator
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_create_server_registers_tools(tmp_path):
|
|
config_file = tmp_path / "config.yaml"
|
|
config_file.write_text("""
|
|
documents:
|
|
test-doc:
|
|
url: https://grist.example.com
|
|
doc_id: abc123
|
|
api_key: test-key
|
|
|
|
tokens:
|
|
- token: test-token
|
|
name: test-agent
|
|
scope:
|
|
- document: test-doc
|
|
permissions: [read, write, schema]
|
|
""")
|
|
|
|
config = load_config(str(config_file))
|
|
auth = Authenticator(config)
|
|
agent = auth.authenticate("test-token")
|
|
server = create_server(auth, agent)
|
|
|
|
# Server should have tools registered
|
|
assert server is not None
|
|
|
|
# Get the list_tools handler and call it
|
|
handler = server.request_handlers.get(ListToolsRequest)
|
|
assert handler is not None
|
|
|
|
req = ListToolsRequest(method="tools/list")
|
|
result = await handler(req)
|
|
|
|
# Check tool names are registered
|
|
tool_names = [t.name for t in result.root.tools]
|
|
assert "list_documents" in tool_names
|
|
assert "list_tables" in tool_names
|
|
assert "describe_table" in tool_names
|
|
assert "get_records" in tool_names
|
|
assert "sql_query" in tool_names
|
|
assert "add_records" in tool_names
|
|
assert "update_records" in tool_names
|
|
assert "delete_records" in tool_names
|
|
assert "create_table" in tool_names
|
|
assert "add_column" in tool_names
|
|
assert "modify_column" in tool_names
|
|
assert "delete_column" in tool_names
|
|
|
|
# Session tools (always registered)
|
|
assert "get_proxy_documentation" in tool_names
|
|
assert "request_session_token" in tool_names
|
|
|
|
# Should have all 14 tools
|
|
assert len(result.root.tools) == 14
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_create_server_registers_session_tools(tmp_path):
|
|
from grist_mcp.session import SessionTokenManager
|
|
|
|
config_file = tmp_path / "config.yaml"
|
|
config_file.write_text("""
|
|
documents:
|
|
test-doc:
|
|
url: https://grist.example.com
|
|
doc_id: abc123
|
|
api_key: test-key
|
|
|
|
tokens:
|
|
- token: valid-token
|
|
name: test-agent
|
|
scope:
|
|
- document: test-doc
|
|
permissions: [read, write, schema]
|
|
""")
|
|
|
|
config = load_config(str(config_file))
|
|
auth = Authenticator(config)
|
|
agent = auth.authenticate("valid-token")
|
|
token_manager = SessionTokenManager()
|
|
server = create_server(auth, agent, token_manager)
|
|
|
|
# Get the list_tools handler and call it
|
|
handler = server.request_handlers.get(ListToolsRequest)
|
|
assert handler is not None
|
|
|
|
req = ListToolsRequest(method="tools/list")
|
|
result = await handler(req)
|
|
|
|
tool_names = [t.name for t in result.root.tools]
|
|
|
|
assert "get_proxy_documentation" in tool_names
|
|
assert "request_session_token" in tool_names
|