1aa19548d1
Major redesign of Gemini MCP Server with modular architecture: - Removed all emoji characters from tool outputs for clean terminal display - Kept review category emojis (🔴🟠🟡🟢) per user preference - Added 4 specialized tools: - think_deeper: Extended reasoning and problem-solving (temp 0.7) - review_code: Professional code review with severity levels (temp 0.2) - debug_issue: Root cause analysis and debugging (temp 0.2) - analyze: General-purpose file analysis (temp 0.2) - Modular architecture with base tool class and Pydantic models - Verbose tool descriptions with natural language triggers - Updated README with comprehensive examples and real-world use cases - All 25 tests passing, type checking clean, critical linting clean BREAKING CHANGE: Removed analyze_code tool in favor of specialized tools 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
92 lines
2.9 KiB
Python
92 lines
2.9 KiB
Python
"""
|
|
Tests for utility functions
|
|
"""
|
|
|
|
from utils import (
|
|
read_file_content,
|
|
read_files,
|
|
estimate_tokens,
|
|
check_token_limit,
|
|
)
|
|
|
|
|
|
class TestFileUtils:
|
|
"""Test file reading utilities"""
|
|
|
|
def test_read_file_content_success(self, tmp_path):
|
|
"""Test successful file reading"""
|
|
test_file = tmp_path / "test.py"
|
|
test_file.write_text(
|
|
"def hello():\n return 'world'", encoding="utf-8"
|
|
)
|
|
|
|
content = read_file_content(str(test_file))
|
|
assert "--- BEGIN FILE:" in content
|
|
assert "--- END FILE:" in content
|
|
assert "def hello():" in content
|
|
assert "return 'world'" in content
|
|
|
|
def test_read_file_content_not_found(self):
|
|
"""Test reading non-existent file"""
|
|
content = read_file_content("/nonexistent/file.py")
|
|
assert "--- FILE NOT FOUND:" in content
|
|
assert "Error: File does not exist" in content
|
|
|
|
def test_read_file_content_directory(self, tmp_path):
|
|
"""Test reading a directory"""
|
|
content = read_file_content(str(tmp_path))
|
|
assert "--- NOT A FILE:" in content
|
|
assert "Error: Path is not a file" in content
|
|
|
|
def test_read_files_multiple(self, tmp_path):
|
|
"""Test reading multiple files"""
|
|
file1 = tmp_path / "file1.py"
|
|
file1.write_text("print('file1')", encoding="utf-8")
|
|
file2 = tmp_path / "file2.py"
|
|
file2.write_text("print('file2')", encoding="utf-8")
|
|
|
|
content, summary = read_files([str(file1), str(file2)])
|
|
|
|
assert "--- BEGIN FILE:" in content
|
|
assert "file1.py" in content
|
|
assert "file2.py" in content
|
|
assert "print('file1')" in content
|
|
assert "print('file2')" in content
|
|
|
|
assert "Reading 2 file(s)" in summary
|
|
|
|
def test_read_files_with_code(self):
|
|
"""Test reading with direct code"""
|
|
code = "def test():\n pass"
|
|
content, summary = read_files([], code)
|
|
|
|
assert "--- BEGIN DIRECT CODE ---" in content
|
|
assert "--- END DIRECT CODE ---" in content
|
|
assert code in content
|
|
|
|
assert "Direct code:" in summary
|
|
|
|
|
|
class TestTokenUtils:
|
|
"""Test token counting utilities"""
|
|
|
|
def test_estimate_tokens(self):
|
|
"""Test token estimation"""
|
|
# Rough estimate: 1 token ≈ 4 characters
|
|
text = "a" * 400 # 400 characters
|
|
assert estimate_tokens(text) == 100
|
|
|
|
def test_check_token_limit_within(self):
|
|
"""Test token limit check - within limit"""
|
|
text = "a" * 4000 # 1000 tokens
|
|
within_limit, tokens = check_token_limit(text)
|
|
assert within_limit is True
|
|
assert tokens == 1000
|
|
|
|
def test_check_token_limit_exceeded(self):
|
|
"""Test token limit check - exceeded"""
|
|
text = "a" * 5_000_000 # 1.25M tokens
|
|
within_limit, tokens = check_token_limit(text)
|
|
assert within_limit is False
|
|
assert tokens == 1_250_000
|