Add comprehensive test suites for all new tools (20+ tests)

tests/test_search_code.py

@@ -0,0 +1,91 @@
+from __future__ import annotations
+
+import pytest
+from pathlib import Path
+import tempfile
+
+from core.tools.base import ToolContext
+from tools.search_code import SearchCodeTool
+
+
+class TestSearchCodeTool:
+    """Test code search tool."""
+
+    @pytest.fixture
+    def tool(self) -> SearchCodeTool:
+        return SearchCodeTool()
+
+    @pytest.fixture
+    def temp_dir(self) -> Path:
+        """Create a temporary directory with test files."""
+        tmpdir = Path(tempfile.mkdtemp())
+        
+        (tmpdir / "test1.py").write_text("def hello():\n    print('hello')\n")
+        (tmpdir / "test2.py").write_text("def world():\n    print('world')\n")
+        (tmpdir / "data.txt").write_text("some data\n")
+        
+        yield tmpdir
+
+    def test_search_literal(self, tool: SearchCodeTool, temp_dir: Path) -> None:
+        """Test literal string search."""
+        result = tool.search_literal({
+            "pattern": "def",
+            "path": str(temp_dir),
+            "file_pattern": "*.py"
+        })
+        assert result["pattern"] == "def"
+        assert result["matches_found"] >= 2
+
+    def test_search_literal_no_matches(self, tool: SearchCodeTool, temp_dir: Path) -> None:
+        """Test literal search with no matches."""
+        result = tool.search_literal({
+            "pattern": "nonexistent_pattern",
+            "path": str(temp_dir),
+            "file_pattern": "*.py"
+        })
+        assert result["matches_found"] == 0
+
+    def test_search_regex(self, tool: SearchCodeTool, temp_dir: Path) -> None:
+        """Test regex search."""
+        result = tool.search_regex({
+            "pattern": r"def \w+",
+            "path": str(temp_dir),
+            "file_pattern": "*.py"
+        })
+        assert result["matches_found"] >= 2
+
+    def test_search_regex_invalid(self, tool: SearchCodeTool, temp_dir: Path) -> None:
+        """Test regex search with invalid pattern."""
+        with pytest.raises(ValueError, match="Invalid regex"):
+            tool.search_regex({
+                "pattern": "[invalid",
+                "path": str(temp_dir)
+            })
+
+    def test_search_in_file(self, tool: SearchCodeTool, temp_dir: Path) -> None:
+        """Test search within a specific file."""
+        test_file = temp_dir / "test1.py"
+        result = tool.search_in_file({
+            "path": str(test_file),
+            "pattern": "hello"
+        })
+        assert result["count"] >= 1
+
+    def test_execute_literal(self, tool: SearchCodeTool, temp_dir: Path) -> None:
+        """Test execute with literal action."""
+        ctx = ToolContext(dry_run=False)
+        result = tool.execute(
+            {
+                "action": "literal",
+                "pattern": "def",
+                "path": str(temp_dir)
+            },
+            ctx
+        )
+        assert result["matches_found"] >= 2
+
+    def test_execute_invalid_action(self, tool: SearchCodeTool) -> None:
+        """Test with invalid action."""
+        ctx = ToolContext(dry_run=False)
+        with pytest.raises(ValueError, match="Unknown search action"):
+            tool.execute({"action": "invalid_action"}, ctx)