File size: 2,899 Bytes
1afb40b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 | import pytest
from pathlib import Path
from src.tools.file_ops import FileOperations
def test_file_operations_read_scope(tmp_path):
ops = FileOperations(repo_root=str(tmp_path))
test_file = tmp_path / "sample.py"
test_file.write_text("line one\nline two\nline three\n", encoding="utf-8")
res = ops.read_scope("sample.py", 1, 2)
assert res["status"] == "success"
assert res["total_lines"] == 3
assert "line one" in res["content"]
assert "line two" in res["content"]
def test_file_operations_read_scope_not_found(tmp_path):
ops = FileOperations(repo_root=str(tmp_path))
res = ops.read_scope("nonexistent.py", 1, 5)
assert res["status"] == "error"
assert "not found" in res["message"]
def test_file_operations_read_scope_generic_exception(tmp_path, monkeypatch):
ops = FileOperations(repo_root=str(tmp_path))
test_file = tmp_path / "error.py"
test_file.write_text("content", encoding="utf-8")
def mock_read_text(*args, **kwargs):
raise RuntimeError("Simulated read failure")
monkeypatch.setattr("pathlib.Path.read_text", mock_read_text)
res = ops.read_scope("error.py", 1, 2)
assert res["status"] == "error"
assert "Simulated read failure" in res["message"]
def test_file_operations_patch_file(tmp_path):
ops = FileOperations(repo_root=str(tmp_path))
test_file = tmp_path / "target.py"
test_file.write_text("def hello():\n return 'old'\n", encoding="utf-8")
res = ops.patch_file("target.py", "return 'old'", "return 'new'")
assert res["status"] == "success"
updated_content = test_file.read_text(encoding="utf-8")
assert "return 'new'" in updated_content
def test_file_operations_patch_file_missing_target(tmp_path):
ops = FileOperations(repo_root=str(tmp_path))
test_file = tmp_path / "target.py"
test_file.write_text("def hello():\n return 'old'\n", encoding="utf-8")
res = ops.patch_file("target.py", "nonexistent_code", "replacement")
assert res["status"] == "error"
assert "Target code block not found" in res["message"]
def test_file_operations_patch_file_exception(tmp_path):
ops = FileOperations(repo_root=str(tmp_path))
sub_dir = tmp_path / "subdir"
sub_dir.mkdir()
res = ops.patch_file("subdir", "target", "replacement")
assert res["status"] == "error"
def test_file_operations_path_traversal_security(tmp_path):
ops = FileOperations(repo_root=str(tmp_path))
res = ops.read_scope("../outside_file.py", 1, 5)
assert res["status"] == "error"
assert "Access denied" in res["message"]
def test_patch_file_not_found(tmp_path):
from src.tools.file_ops import FileOperations
ops = FileOperations(repo_root=str(tmp_path))
res = ops.patch_file("nonexistent.py", "target", "replacement")
assert res["status"] == "error"
assert "not found" in res["message"]
|