31 lines
826 B
Python
31 lines
826 B
Python
"""
|
|
Tests for basic MCP tools: prune, cognify_status
|
|
|
|
These are integration tests that test the actual tool behavior.
|
|
Run with: pytest tests/tools/test_basic_tools.py -v
|
|
"""
|
|
|
|
import pytest
|
|
import mcp.types as types
|
|
|
|
from src import server
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_prune():
|
|
"""Test prune tool - removes all data from knowledge graph"""
|
|
result = await server.prune()
|
|
|
|
assert len(result) == 1
|
|
assert isinstance(result[0], types.TextContent)
|
|
assert "Pruned" in result[0].text or "not available" in result[0].text
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_cognify_status():
|
|
"""Test cognify_status tool - gets status of cognify pipeline"""
|
|
result = await server.cognify_status()
|
|
|
|
assert len(result) == 1
|
|
assert isinstance(result[0], types.TextContent)
|
|
assert len(result[0].text) > 0
|