fix tests

Author: kopekC

tests/unit/codegen/extensions/test_tools.py

@@ -68,7 +68,8 @@ def large_codebase(tmpdir):
     for i in range(1, 401):  # This will create a 400+ line file
         if i % 20 == 0:
             # Add some class methods periodically
-            large_file_lines.extend(["    @classmethod", f"    def class_method_{i}(cls) -> None:", f"        print('Class method {i}')", "        return None", ""])
+            large_file_lines.extend(
+                ["    @classmethod", f"    def class_method_{i}(cls) -> None:", f"        print('Class method {i}')", "        return None", ""])
         else:
             # Add regular methods
             large_file_lines.extend(
@@ -123,7 +124,8 @@ def test_view_file_pagination(large_codebase):
     assert "def method_251" not in result.content  # Method after page 1
 
     # Test custom pagination range
-    result = view_file(large_codebase, "src/large_file.py", start_line=200, end_line=250)
+    result = view_file(large_codebase, "src/large_file.py",
+                       start_line=200, end_line=250)
     assert result.status == "success"
     assert result.start_line == 200
     assert result.end_line == 250
@@ -138,10 +140,12 @@ def test_view_file_pagination(large_codebase):
     result = view_file(large_codebase, "src/large_file.py", start_line=350)
     assert result.status == "success"
     assert result.start_line == 350
-    assert result.has_more is True  # File has 2010 lines, so there should be more content
+    # File has 2010 lines, so there should be more content
+    assert result.has_more is True
     assert "def method_69" in result.content  # Regular method
     assert "def class_method_80" in result.content  # Class method at 80
-    assert result.end_line == 599  # Should show 250 lines from start (350 to 599)
+    # Should show 250 lines from start (350 to 599)
+    assert result.end_line == 599
 
     # Test custom max_lines
     result = view_file(large_codebase, "src/large_file.py", max_lines=100)
@@ -154,7 +158,8 @@ def test_view_file_pagination(large_codebase):
     assert len(result.content.splitlines()) <= 100
 
     # Test line numbers display
-    result = view_file(large_codebase, "src/large_file.py", start_line=198, end_line=202, line_numbers=True)
+    result = view_file(large_codebase, "src/large_file.py",
+                       start_line=198, end_line=202, line_numbers=True)
     assert result.status == "success"
     assert "198|" in result.content
     assert "199|" in result.content
@@ -163,7 +168,8 @@ def test_view_file_pagination(large_codebase):
     assert "202|" in result.content
 
     # Test without line numbers
-    result = view_file(large_codebase, "src/large_file.py", start_line=198, end_line=202, line_numbers=False)
+    result = view_file(large_codebase, "src/large_file.py",
+                       start_line=198, end_line=202, line_numbers=False)
     assert result.status == "success"
     assert "198|" not in result.content
     assert "199|" not in result.content
@@ -172,7 +178,8 @@ def test_view_file_pagination(large_codebase):
 def test_view_file_pagination_edge_cases(large_codebase):
     """Test edge cases for file pagination."""
     # Test start_line > end_line (should respect provided end_line)
-    result = view_file(large_codebase, "src/large_file.py", start_line=200, end_line=100)
+    result = view_file(large_codebase, "src/large_file.py",
+                       start_line=200, end_line=100)
     assert result.status == "success"
     assert result.start_line == 200
     assert result.end_line == 100  # Should respect provided end_line
@@ -186,10 +193,12 @@ def test_view_file_pagination_edge_cases(large_codebase):
     assert result.has_more is False
 
     # Test end_line > file length (should truncate to file length)
-    result = view_file(large_codebase, "src/large_file.py", start_line=200, end_line=2000)
+    result = view_file(large_codebase, "src/large_file.py",
+                       start_line=200, end_line=2000)
     assert result.status == "success"
     assert result.start_line == 200
-    assert result.end_line == min(200 + 250 - 1, 2010)  # Should respect max_lines and file length
+    # Should respect max_lines and file length
+    assert result.end_line == min(200 + 250 - 1, 2010)
 
     # Test negative start_line (should default to 1)
     result = view_file(large_codebase, "src/large_file.py", start_line=-10)
@@ -205,7 +214,8 @@ def test_list_directory(codebase):
     create_file(codebase, "src/core/models.py", "")
     create_file(codebase, "src/utils.py", "")
 
-    result = list_directory(codebase, "./", depth=2)  # Ensure we get nested structure
+    # Ensure we get nested structure
+    result = list_directory(codebase, "./", depth=2)
     assert result.status == "success"
 
     # Check directory structure
@@ -239,7 +249,8 @@ def test_search(codebase):
     assert len(result.results) > 0
 
     # Check that we found the right content
-    assert any("hello" in match.match.lower() for file_result in result.results for match in file_result.matches)
+    assert any("hello" in match.match.lower()
+               for file_result in result.results for match in file_result.matches)
 
     # Check pagination info
     assert result.page == 1
@@ -255,7 +266,8 @@ def test_search_regex(codebase):
     assert len(result.results) > 0
 
     # Should find both 'def hello' and 'def greet'
-    matches = [match.line for file_result in result.results for match in file_result.matches]
+    matches = [
+        match.line for file_result in result.results for match in file_result.matches]
     assert any("def hello" in match for match in matches)
     assert any("def greet" in match for match in matches)
 
@@ -279,7 +291,8 @@ def test_search_pagination(codebase, tmpdir):
         # If we have enough results for multiple pages
         if result_page1.total_pages > 1:
             # Get page 2
-            result_page2 = search(pagination_codebase, "Hello", page=2, files_per_page=5)
+            result_page2 = search(pagination_codebase,
+                                  "Hello", page=2, files_per_page=5)
             assert result_page2.status == "success"
             assert result_page2.page == 2
             assert len(result_page2.results) <= 5
@@ -290,40 +303,6 @@ def test_search_pagination(codebase, tmpdir):
             assert not page1_files.intersection(page2_files)
 
 
-def test_search_fallback(codebase, monkeypatch):
-    """Test fallback to Python implementation when ripgrep fails."""
-
-    # Mock subprocess.run to simulate ripgrep failure
-    def mock_subprocess_run(*args, **kwargs):
-        msg = "Simulated ripgrep failure"
-        raise subprocess.SubprocessError(msg)
-
-    # Apply the mock
-    monkeypatch.setattr(subprocess, "run", mock_subprocess_run)
-
-    # Search should still work using Python fallback
-    result = search(codebase, "hello")
-    assert result.status == "success"
-    assert len(result.results) > 0
-
-
-def test_search_ripgrep_not_found(codebase, monkeypatch):
-    """Test fallback to Python implementation when ripgrep is not installed."""
-
-    # Mock subprocess.run to simulate ripgrep not found
-    def mock_subprocess_run(*args, **kwargs):
-        msg = "Simulated ripgrep not found"
-        raise FileNotFoundError(msg)
-
-    # Apply the mock
-    monkeypatch.setattr(subprocess, "run", mock_subprocess_run)
-
-    # Search should still work using Python fallback
-    result = search(codebase, "hello")
-    assert result.status == "success"
-    assert len(result.results) > 0
-
-
 def test_search_uses_ripgrep(codebase, monkeypatch):
     """Test that ripgrep is used when available."""
     # Track if ripgrep was called