mirror of
https://github.com/catlog22/Claude-Code-Workflow.git
synced 2026-02-06 01:54:11 +08:00
f14418603a
重构 ccw cli 模板系统： - 新增 template-discovery.ts 模块，支持扁平化模板自动发现 - 添加 --rule <template> 选项，自动加载 protocol 和 template - 模板目录从嵌套结构 (prompts/category/file.txt) 迁移到扁平结构 (prompts/category-function.txt) - 更新所有 agent/command 文件，使用 $PROTO $TMPL 环境变量替代 $(cat ...) 模式 - 支持模糊匹配：--rule 02-review-architecture 可匹配 analysis-review-architecture.txt 其他更新： - Dashboard: 添加 Claude Manager 和 Issue Manager 页面 - Codex-lens: 增强 chain_search 和 clustering 模块 Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
384 lines
13 KiB
Python
384 lines
13 KiB
Python
"""Tests for MCP provider."""
|
|
|
|
import pytest
|
|
from unittest.mock import Mock, MagicMock, patch
|
|
from pathlib import Path
|
|
import tempfile
|
|
import os
|
|
|
|
from codexlens.mcp.provider import MCPProvider
|
|
from codexlens.mcp.schema import MCPContext, SymbolInfo, ReferenceInfo
|
|
|
|
|
|
class TestMCPProvider:
|
|
"""Test MCPProvider class."""
|
|
|
|
@pytest.fixture
|
|
def mock_global_index(self):
|
|
"""Create a mock global index."""
|
|
return Mock()
|
|
|
|
@pytest.fixture
|
|
def mock_search_engine(self):
|
|
"""Create a mock search engine."""
|
|
return Mock()
|
|
|
|
@pytest.fixture
|
|
def mock_registry(self):
|
|
"""Create a mock registry."""
|
|
return Mock()
|
|
|
|
@pytest.fixture
|
|
def provider(self, mock_global_index, mock_search_engine, mock_registry):
|
|
"""Create an MCPProvider with mocked dependencies."""
|
|
return MCPProvider(mock_global_index, mock_search_engine, mock_registry)
|
|
|
|
def test_build_context_returns_none_for_unknown_symbol(self, provider, mock_global_index):
|
|
"""build_context returns None when symbol is not found."""
|
|
mock_global_index.search.return_value = []
|
|
|
|
result = provider.build_context("unknown_symbol")
|
|
|
|
assert result is None
|
|
mock_global_index.search.assert_called_once_with(
|
|
"unknown_symbol", prefix_mode=False, limit=1
|
|
)
|
|
|
|
def test_build_context_returns_mcp_context(
|
|
self, provider, mock_global_index, mock_search_engine
|
|
):
|
|
"""build_context returns MCPContext for known symbol."""
|
|
mock_symbol = Mock()
|
|
mock_symbol.name = "my_func"
|
|
mock_symbol.kind = "function"
|
|
mock_symbol.file = "/test.py"
|
|
mock_symbol.range = (10, 20)
|
|
|
|
mock_global_index.search.return_value = [mock_symbol]
|
|
mock_search_engine.search_references.return_value = []
|
|
|
|
result = provider.build_context("my_func")
|
|
|
|
assert result is not None
|
|
assert isinstance(result, MCPContext)
|
|
assert result.symbol is not None
|
|
assert result.symbol.name == "my_func"
|
|
assert result.symbol.kind == "function"
|
|
assert result.context_type == "symbol_explanation"
|
|
|
|
def test_build_context_with_custom_context_type(
|
|
self, provider, mock_global_index, mock_search_engine
|
|
):
|
|
"""build_context respects custom context_type."""
|
|
mock_symbol = Mock()
|
|
mock_symbol.name = "my_func"
|
|
mock_symbol.kind = "function"
|
|
mock_symbol.file = "/test.py"
|
|
mock_symbol.range = (10, 20)
|
|
|
|
mock_global_index.search.return_value = [mock_symbol]
|
|
mock_search_engine.search_references.return_value = []
|
|
|
|
result = provider.build_context("my_func", context_type="refactor_context")
|
|
|
|
assert result is not None
|
|
assert result.context_type == "refactor_context"
|
|
|
|
def test_build_context_includes_references(
|
|
self, provider, mock_global_index, mock_search_engine
|
|
):
|
|
"""build_context includes references when include_references=True."""
|
|
mock_symbol = Mock()
|
|
mock_symbol.name = "my_func"
|
|
mock_symbol.kind = "function"
|
|
mock_symbol.file = "/test.py"
|
|
mock_symbol.range = (10, 20)
|
|
|
|
mock_ref = Mock()
|
|
mock_ref.file_path = "/caller.py"
|
|
mock_ref.line = 25
|
|
mock_ref.column = 4
|
|
mock_ref.context = "result = my_func()"
|
|
mock_ref.relationship_type = "call"
|
|
|
|
mock_global_index.search.return_value = [mock_symbol]
|
|
mock_search_engine.search_references.return_value = [mock_ref]
|
|
|
|
result = provider.build_context("my_func", include_references=True)
|
|
|
|
assert result is not None
|
|
assert len(result.references) == 1
|
|
assert result.references[0].file_path == "/caller.py"
|
|
assert result.references[0].line == 25
|
|
assert result.references[0].relationship_type == "call"
|
|
|
|
def test_build_context_excludes_references_when_disabled(
|
|
self, provider, mock_global_index, mock_search_engine
|
|
):
|
|
"""build_context excludes references when include_references=False."""
|
|
mock_symbol = Mock()
|
|
mock_symbol.name = "my_func"
|
|
mock_symbol.kind = "function"
|
|
mock_symbol.file = "/test.py"
|
|
mock_symbol.range = (10, 20)
|
|
|
|
mock_global_index.search.return_value = [mock_symbol]
|
|
mock_search_engine.search_references.return_value = []
|
|
|
|
# Disable both references and related to avoid any search_references calls
|
|
result = provider.build_context(
|
|
"my_func", include_references=False, include_related=False
|
|
)
|
|
|
|
assert result is not None
|
|
assert len(result.references) == 0
|
|
mock_search_engine.search_references.assert_not_called()
|
|
|
|
def test_build_context_respects_max_references(
|
|
self, provider, mock_global_index, mock_search_engine
|
|
):
|
|
"""build_context passes max_references to search engine."""
|
|
mock_symbol = Mock()
|
|
mock_symbol.name = "my_func"
|
|
mock_symbol.kind = "function"
|
|
mock_symbol.file = "/test.py"
|
|
mock_symbol.range = (10, 20)
|
|
|
|
mock_global_index.search.return_value = [mock_symbol]
|
|
mock_search_engine.search_references.return_value = []
|
|
|
|
# Disable include_related to test only the references call
|
|
provider.build_context("my_func", max_references=5, include_related=False)
|
|
|
|
mock_search_engine.search_references.assert_called_once_with(
|
|
"my_func", limit=5
|
|
)
|
|
|
|
def test_build_context_includes_metadata(
|
|
self, provider, mock_global_index, mock_search_engine
|
|
):
|
|
"""build_context includes source metadata."""
|
|
mock_symbol = Mock()
|
|
mock_symbol.name = "my_func"
|
|
mock_symbol.kind = "function"
|
|
mock_symbol.file = "/test.py"
|
|
mock_symbol.range = (10, 20)
|
|
|
|
mock_global_index.search.return_value = [mock_symbol]
|
|
mock_search_engine.search_references.return_value = []
|
|
|
|
result = provider.build_context("my_func")
|
|
|
|
assert result is not None
|
|
assert result.metadata.get("source") == "codex-lens"
|
|
|
|
def test_extract_definition_with_valid_file(self, provider):
|
|
"""_extract_definition reads file content correctly."""
|
|
# Create a temporary file with some content
|
|
with tempfile.NamedTemporaryFile(mode='w', suffix='.py', delete=False) as f:
|
|
f.write("# Line 1\n")
|
|
f.write("# Line 2\n")
|
|
f.write("def my_func():\n") # Line 3
|
|
f.write(" pass\n") # Line 4
|
|
f.write("# Line 5\n")
|
|
temp_path = f.name
|
|
|
|
try:
|
|
mock_symbol = Mock()
|
|
mock_symbol.file = temp_path
|
|
mock_symbol.range = (3, 4) # 1-based line numbers
|
|
|
|
definition = provider._extract_definition(mock_symbol)
|
|
|
|
assert definition is not None
|
|
assert "def my_func():" in definition
|
|
assert "pass" in definition
|
|
finally:
|
|
os.unlink(temp_path)
|
|
|
|
def test_extract_definition_returns_none_for_missing_file(self, provider):
|
|
"""_extract_definition returns None for non-existent file."""
|
|
mock_symbol = Mock()
|
|
mock_symbol.file = "/nonexistent/path/file.py"
|
|
mock_symbol.range = (1, 5)
|
|
|
|
definition = provider._extract_definition(mock_symbol)
|
|
|
|
assert definition is None
|
|
|
|
def test_extract_definition_returns_none_for_none_file(self, provider):
|
|
"""_extract_definition returns None when symbol.file is None."""
|
|
mock_symbol = Mock()
|
|
mock_symbol.file = None
|
|
mock_symbol.range = (1, 5)
|
|
|
|
definition = provider._extract_definition(mock_symbol)
|
|
|
|
assert definition is None
|
|
|
|
def test_build_context_for_file_returns_context(
|
|
self, provider, mock_global_index
|
|
):
|
|
"""build_context_for_file returns MCPContext."""
|
|
mock_global_index.search.return_value = []
|
|
|
|
result = provider.build_context_for_file(
|
|
Path("/test/file.py"),
|
|
context_type="file_overview",
|
|
)
|
|
|
|
assert result is not None
|
|
assert isinstance(result, MCPContext)
|
|
assert result.context_type == "file_overview"
|
|
assert result.metadata.get("file_path") == str(Path("/test/file.py"))
|
|
|
|
def test_build_context_for_file_includes_symbols(
|
|
self, provider, mock_global_index
|
|
):
|
|
"""build_context_for_file includes symbols from the file."""
|
|
# Create temp file to get resolved path
|
|
with tempfile.NamedTemporaryFile(mode='w', suffix='.py', delete=False) as f:
|
|
f.write("def func(): pass\n")
|
|
temp_path = f.name
|
|
|
|
try:
|
|
mock_symbol = Mock()
|
|
mock_symbol.name = "func"
|
|
mock_symbol.kind = "function"
|
|
mock_symbol.file = temp_path
|
|
mock_symbol.range = (1, 1)
|
|
|
|
mock_global_index.search.return_value = [mock_symbol]
|
|
|
|
result = provider.build_context_for_file(Path(temp_path))
|
|
|
|
assert result is not None
|
|
# Symbols from this file should be in related_symbols
|
|
assert len(result.related_symbols) >= 0 # May be 0 if filtering doesn't match
|
|
finally:
|
|
os.unlink(temp_path)
|
|
|
|
|
|
class TestMCPProviderRelatedSymbols:
|
|
"""Test related symbols functionality."""
|
|
|
|
@pytest.fixture
|
|
def provider(self):
|
|
"""Create provider with mocks."""
|
|
mock_global_index = Mock()
|
|
mock_search_engine = Mock()
|
|
mock_registry = Mock()
|
|
return MCPProvider(mock_global_index, mock_search_engine, mock_registry)
|
|
|
|
def test_get_related_symbols_from_references(self, provider):
|
|
"""_get_related_symbols extracts symbols from references."""
|
|
mock_symbol = Mock()
|
|
mock_symbol.name = "my_func"
|
|
mock_symbol.file = "/test.py"
|
|
|
|
mock_ref1 = Mock()
|
|
mock_ref1.file_path = "/caller1.py"
|
|
mock_ref1.relationship_type = "call"
|
|
|
|
mock_ref2 = Mock()
|
|
mock_ref2.file_path = "/caller2.py"
|
|
mock_ref2.relationship_type = "import"
|
|
|
|
provider.search_engine.search_references.return_value = [mock_ref1, mock_ref2]
|
|
|
|
related = provider._get_related_symbols(mock_symbol)
|
|
|
|
assert len(related) == 2
|
|
assert related[0].relationship == "call"
|
|
assert related[1].relationship == "import"
|
|
|
|
def test_get_related_symbols_limits_results(self, provider):
|
|
"""_get_related_symbols limits to 10 unique relationship types."""
|
|
mock_symbol = Mock()
|
|
mock_symbol.name = "my_func"
|
|
mock_symbol.file = "/test.py"
|
|
|
|
# Create 15 references with unique relationship types
|
|
refs = []
|
|
for i in range(15):
|
|
ref = Mock()
|
|
ref.file_path = f"/file{i}.py"
|
|
ref.relationship_type = f"type{i}"
|
|
refs.append(ref)
|
|
|
|
provider.search_engine.search_references.return_value = refs
|
|
|
|
related = provider._get_related_symbols(mock_symbol)
|
|
|
|
assert len(related) <= 10
|
|
|
|
def test_get_related_symbols_handles_exception(self, provider):
|
|
"""_get_related_symbols handles exceptions gracefully."""
|
|
mock_symbol = Mock()
|
|
mock_symbol.name = "my_func"
|
|
mock_symbol.file = "/test.py"
|
|
|
|
provider.search_engine.search_references.side_effect = Exception("Search failed")
|
|
|
|
related = provider._get_related_symbols(mock_symbol)
|
|
|
|
assert related == []
|
|
|
|
|
|
class TestMCPProviderIntegration:
|
|
"""Integration-style tests for MCPProvider."""
|
|
|
|
def test_full_context_workflow(self):
|
|
"""Test complete context building workflow."""
|
|
# Create temp file
|
|
with tempfile.NamedTemporaryFile(mode='w', suffix='.py', delete=False) as f:
|
|
f.write("def my_function(arg1, arg2):\n")
|
|
f.write(" '''This is my function.'''\n")
|
|
f.write(" return arg1 + arg2\n")
|
|
temp_path = f.name
|
|
|
|
try:
|
|
# Setup mocks
|
|
mock_global_index = Mock()
|
|
mock_search_engine = Mock()
|
|
mock_registry = Mock()
|
|
|
|
mock_symbol = Mock()
|
|
mock_symbol.name = "my_function"
|
|
mock_symbol.kind = "function"
|
|
mock_symbol.file = temp_path
|
|
mock_symbol.range = (1, 3)
|
|
|
|
mock_ref = Mock()
|
|
mock_ref.file_path = "/user.py"
|
|
mock_ref.line = 10
|
|
mock_ref.column = 4
|
|
mock_ref.context = "result = my_function(1, 2)"
|
|
mock_ref.relationship_type = "call"
|
|
|
|
mock_global_index.search.return_value = [mock_symbol]
|
|
mock_search_engine.search_references.return_value = [mock_ref]
|
|
|
|
provider = MCPProvider(mock_global_index, mock_search_engine, mock_registry)
|
|
context = provider.build_context("my_function")
|
|
|
|
assert context is not None
|
|
assert context.symbol.name == "my_function"
|
|
assert context.definition is not None
|
|
assert "def my_function" in context.definition
|
|
assert len(context.references) == 1
|
|
assert context.references[0].relationship_type == "call"
|
|
|
|
# Test serialization
|
|
json_str = context.to_json()
|
|
assert "my_function" in json_str
|
|
|
|
# Test prompt injection
|
|
prompt = context.to_prompt_injection()
|
|
assert "<code_context>" in prompt
|
|
assert "my_function" in prompt
|
|
assert "</code_context>" in prompt
|
|
|
|
finally:
|
|
os.unlink(temp_path)
|