307e0e3b71
- Create test structure mirroring features folder organization - Add tests for document tools (create, list, update, delete) - Add tests for version tools (create, list, restore, invalid field handling) - Add tests for task tools (create with sources, list with filters, update, delete) - Add tests for project tools (create with polling, list, get) - Add tests for feature tools (get features with various structures) - Mock HTTP client for all external API calls - Test both success and error scenarios - 100% test coverage for critical tool functions
123 lines
4.3 KiB
Python
123 lines
4.3 KiB
Python
"""Unit tests for feature management tools."""
|
|
|
|
import json
|
|
from unittest.mock import AsyncMock, MagicMock, patch
|
|
|
|
import pytest
|
|
from mcp.server.fastmcp import Context
|
|
|
|
from src.mcp_server.features.feature_tools import register_feature_tools
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_mcp():
|
|
"""Create a mock MCP server for testing."""
|
|
mock = MagicMock()
|
|
# Store registered tools
|
|
mock._tools = {}
|
|
|
|
def tool_decorator():
|
|
def decorator(func):
|
|
mock._tools[func.__name__] = func
|
|
return func
|
|
return decorator
|
|
|
|
mock.tool = tool_decorator
|
|
return mock
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_context():
|
|
"""Create a mock context for testing."""
|
|
return MagicMock(spec=Context)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_project_features_success(mock_mcp, mock_context):
|
|
"""Test successful retrieval of project features."""
|
|
register_feature_tools(mock_mcp)
|
|
|
|
# Get the get_project_features function
|
|
get_project_features = mock_mcp._tools.get('get_project_features')
|
|
|
|
assert get_project_features is not None, "get_project_features tool not registered"
|
|
|
|
# Mock HTTP response with various feature structures
|
|
mock_response = MagicMock()
|
|
mock_response.status_code = 200
|
|
mock_response.json.return_value = {
|
|
"features": [
|
|
{"name": "authentication", "status": "completed", "components": ["oauth", "jwt"]},
|
|
{"name": "api", "status": "in_progress", "endpoints_done": 12, "endpoints_total": 20},
|
|
{"name": "database", "status": "planned"},
|
|
{"name": "payments", "provider": "stripe", "version": "2.0", "enabled": True}
|
|
]
|
|
}
|
|
|
|
with patch('src.mcp_server.features.feature_tools.httpx.AsyncClient') as mock_client:
|
|
mock_async_client = AsyncMock()
|
|
mock_async_client.get.return_value = mock_response
|
|
mock_client.return_value.__aenter__.return_value = mock_async_client
|
|
|
|
result = await get_project_features(mock_context, project_id="project-123")
|
|
|
|
result_data = json.loads(result)
|
|
assert result_data["success"] is True
|
|
assert result_data["count"] == 4
|
|
assert len(result_data["features"]) == 4
|
|
|
|
# Verify different feature structures are preserved
|
|
features = result_data["features"]
|
|
assert features[0]["components"] == ["oauth", "jwt"]
|
|
assert features[1]["endpoints_done"] == 12
|
|
assert features[2]["status"] == "planned"
|
|
assert features[3]["provider"] == "stripe"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_project_features_empty(mock_mcp, mock_context):
|
|
"""Test getting features for a project with no features defined."""
|
|
register_feature_tools(mock_mcp)
|
|
|
|
get_project_features = mock_mcp._tools.get('get_project_features')
|
|
|
|
# Mock response with empty features
|
|
mock_response = MagicMock()
|
|
mock_response.status_code = 200
|
|
mock_response.json.return_value = {"features": []}
|
|
|
|
with patch('src.mcp_server.features.feature_tools.httpx.AsyncClient') as mock_client:
|
|
mock_async_client = AsyncMock()
|
|
mock_async_client.get.return_value = mock_response
|
|
mock_client.return_value.__aenter__.return_value = mock_async_client
|
|
|
|
result = await get_project_features(mock_context, project_id="project-123")
|
|
|
|
result_data = json.loads(result)
|
|
assert result_data["success"] is True
|
|
assert result_data["count"] == 0
|
|
assert result_data["features"] == []
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_project_features_not_found(mock_mcp, mock_context):
|
|
"""Test getting features for a non-existent project."""
|
|
register_feature_tools(mock_mcp)
|
|
|
|
get_project_features = mock_mcp._tools.get('get_project_features')
|
|
|
|
# Mock 404 response
|
|
mock_response = MagicMock()
|
|
mock_response.status_code = 404
|
|
mock_response.text = "Project not found"
|
|
|
|
with patch('src.mcp_server.features.feature_tools.httpx.AsyncClient') as mock_client:
|
|
mock_async_client = AsyncMock()
|
|
mock_async_client.get.return_value = mock_response
|
|
mock_client.return_value.__aenter__.return_value = mock_async_client
|
|
|
|
result = await get_project_features(mock_context, project_id="non-existent")
|
|
|
|
result_data = json.loads(result)
|
|
assert result_data["success"] is False
|
|
assert "not found" in result_data["error"] |