mirror of
https://github.com/basicmachines-co/basic-memory
synced 2026-06-21 13:47:35 +00:00
phernandez
33 lines
874 B
Python
33 lines
874 B
Python
"""Tests for the MCP server implementation using FastAPI TestClient."""
|
|
|
|
import pytest
|
|
|
|
from basic_memory.mcp.server import handle_list_tools
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_list_tools(app):
|
|
"""Test that server exposes expected tools."""
|
|
|
|
tools = await handle_list_tools()
|
|
|
|
# Check each expected tool is present
|
|
expected_tools = {
|
|
"create_entities",
|
|
"search_nodes",
|
|
"open_nodes",
|
|
"add_observations",
|
|
"create_relations",
|
|
"delete_entities",
|
|
"delete_observations",
|
|
"delete_relations",
|
|
}
|
|
|
|
found_tools = {t.name: t for t in tools}
|
|
assert found_tools.keys() == expected_tools
|
|
|
|
# Verify schemas include required fields
|
|
search_schema = found_tools["search_nodes"].inputSchema
|
|
assert "query" in search_schema["properties"]
|
|
assert search_schema["required"] == ["query"]
|