mirror of
https://github.com/basicmachines-co/basic-memory
synced 2026-06-21 13:47:35 +00:00
phernandez
31 lines
1.1 KiB
Python
31 lines
1.1 KiB
Python
"""Tests for MCP delete_entities tool."""
|
|
|
|
import pytest
|
|
|
|
from basic_memory.mcp.server import handle_call_tool
|
|
from basic_memory.schemas import SearchNodesResponse
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_delete_entities(app):
|
|
"""Test deleting entities."""
|
|
# Create test entities
|
|
entities = {
|
|
"entities": [
|
|
{"name": "DeleteTest1", "entity_type": "test", "observations": ["To be deleted 1"]},
|
|
{"name": "DeleteTest2", "entity_type": "test", "observations": ["To be deleted 2"]},
|
|
]
|
|
}
|
|
await handle_call_tool("create_entities", entities)
|
|
|
|
# Delete first entity
|
|
await handle_call_tool("delete_entities", {"entity_ids": ["test/deletetest1"]})
|
|
|
|
# Verify through search
|
|
search_result = await handle_call_tool("search_nodes", {"query": "DeleteTest"})
|
|
search_response = SearchNodesResponse.model_validate_json(search_result[0].resource.text) # pyright: ignore [reportAttributeAccessIssue]
|
|
|
|
# Only second entity should remain
|
|
assert len(search_response.matches) == 1
|
|
assert search_response.matches[0].name == "DeleteTest2"
|