mirror of
https://github.com/basicmachines-co/basic-memory
synced 2026-06-21 13:47:35 +00:00
phernandez
44 lines
1.4 KiB
Python
44 lines
1.4 KiB
Python
"""Tests for the MCP server implementation using FastAPI TestClient."""
|
|
|
|
import pytest
|
|
|
|
from basic_memory.mcp.server import handle_call_tool
|
|
from basic_memory.schemas import SearchNodesResponse
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_create_relations(app):
|
|
"""Test creating relations between entities."""
|
|
# Create two test entities
|
|
entity_data = {
|
|
"entities": [
|
|
{"name": "TestEntityA", "entity_type": "test", "observations": ["Entity A"]},
|
|
{"name": "TestEntityB", "entity_type": "test", "observations": ["Entity B"]},
|
|
]
|
|
}
|
|
|
|
await handle_call_tool("create_entities", entity_data)
|
|
|
|
# Create relation between them
|
|
relation_data = {
|
|
"relations": [
|
|
{
|
|
"from_id": "test/TestEntityA",
|
|
"to_id": "test/TestEntityB",
|
|
"relation_type": "relates_to",
|
|
}
|
|
]
|
|
}
|
|
|
|
result = await handle_call_tool("create_relations", relation_data)
|
|
|
|
# Verify through search
|
|
search_result = await handle_call_tool("search_nodes", {"query": "TestEntityA"})
|
|
response = SearchNodesResponse.model_validate_json(search_result[0].resource.text) # pyright: ignore [reportAttributeAccessIssue]
|
|
|
|
assert len(response.matches) == 1
|
|
entity = response.matches[0]
|
|
assert len(entity.relations) == 1
|
|
assert entity.relations[0].to_id == "test/testentityb"
|
|
assert entity.relations[0].relation_type == "relates_to"
|