Chris Coutinho
61 lines
2.0 KiB
Python
61 lines
2.0 KiB
Python
import json
|
|
import logging
|
|
|
|
import pytest
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
pytestmark = [pytest.mark.integration, pytest.mark.oauth]
|
|
|
|
|
|
async def test_mcp_oauth_server_connection(nc_mcp_oauth_client):
|
|
"""Test connection to OAuth-enabled MCP server."""
|
|
result = await nc_mcp_oauth_client.list_tools()
|
|
assert result is not None
|
|
assert len(result.tools) > 0
|
|
|
|
logger.info(f"OAuth MCP server has {len(result.tools)} tools available")
|
|
|
|
|
|
async def test_mcp_oauth_tool_execution(nc_mcp_oauth_client):
|
|
"""Test executing a tool on the OAuth-enabled MCP server."""
|
|
import json
|
|
|
|
# Example: Execute the 'nc_notes_search_notes' tool
|
|
result = await nc_mcp_oauth_client.call_tool(
|
|
"nc_notes_search_notes", arguments={"query": ""}
|
|
)
|
|
|
|
assert result.isError is False, f"Tool execution failed: {result.content}"
|
|
assert result.content is not None
|
|
response_data = json.loads(result.content[0].text)
|
|
|
|
# The search response should have a 'results' field containing the list
|
|
assert "results" in response_data
|
|
assert isinstance(response_data["results"], list)
|
|
|
|
logger.info(
|
|
f"Successfully executed 'nc_notes_search_notes' tool on OAuth MCP server and got {len(response_data['results'])} notes."
|
|
)
|
|
|
|
|
|
async def test_mcp_oauth_client_with_playwright(nc_mcp_oauth_client):
|
|
"""Test that MCP OAuth client via Playwright can execute tools."""
|
|
|
|
# Test: Execute the 'nc_notes_search_notes' tool
|
|
result = await nc_mcp_oauth_client.call_tool(
|
|
"nc_notes_search_notes", arguments={"query": ""}
|
|
)
|
|
|
|
assert result.isError is False, f"Tool execution failed: {result.content}"
|
|
assert result.content is not None
|
|
response_data = json.loads(result.content[0].text)
|
|
|
|
# The search response should have a 'results' field containing the list
|
|
assert "results" in response_data
|
|
assert isinstance(response_data["results"], list)
|
|
|
|
logger.info(
|
|
f"Successfully executed 'nc_notes_search_notes' tool on Playwright OAuth MCP server and got {len(response_data['results'])} notes."
|
|
)
|