add FileManager and provide issue fixing example

tests/metagpt/tools/libs/test_file_manager.py

@@ -0,0 +1,102 @@
+import pytest
+
+from metagpt.const import TEST_DATA_PATH
+from metagpt.tools.libs.file_manager import FileBlock, FileManager
+
+TEST_FILE_CONTENT = """
+# this is line one
+def test_function_for_fm():
+    "some docstring"
+    a = 1
+    b = 2
+    c = 3
+    # this is the 7th line
+""".strip()
+
+TEST_FILE_PATH = TEST_DATA_PATH / "tools/test_script_for_file_manager.py"
+
+
+@pytest.fixture
+def test_file():
+    with open(TEST_FILE_PATH, "w") as f:
+        f.write(TEST_FILE_CONTENT)
+    yield
+    with open(TEST_FILE_PATH, "w") as f:
+        f.write("")
+
+
+EXPECTED_SEARCHED_BLOCK = FileBlock(
+    file_path=str(TEST_FILE_PATH),
+    block_content='# this is line one\ndef test_function_for_fm():\n    "some docstring"\n    a = 1\n    b = 2\n',
+    block_start_line=1,
+    block_end_line=5,
+    symbol="def test_function_for_fm",
+    symbol_line=2,
+)
+
+
+def test_search_content(test_file):
+    block = FileManager().search_content("def test_function_for_fm", root_path=TEST_DATA_PATH, window=3)
+    assert block == EXPECTED_SEARCHED_BLOCK
+
+
+EXPECTED_CONTENT_AFTER_REPLACE = """
+# this is line one
+def test_function_for_fm():
+    This is the new line A replacing lines 3 to 5.
+    This is the new line B.
+    c = 3
+    # this is the 7th line
+""".strip()
+
+
+def test_replace_content(test_file):
+    FileManager().write_content(
+        file_path=str(TEST_FILE_PATH),
+        start_line=3,
+        end_line=5,
+        new_block_content="    This is the new line A replacing lines 3 to 5.\n    This is the new line B.",
+    )
+    with open(TEST_FILE_PATH, "r") as f:
+        new_content = f.read()
+    print(new_content)
+    assert new_content == EXPECTED_CONTENT_AFTER_REPLACE
+
+
+EXPECTED_CONTENT_AFTER_DELETE = """
+# this is line one
+def test_function_for_fm():
+    c = 3
+    # this is the 7th line
+""".strip()
+
+
+def test_delete_content(test_file):
+    FileManager().write_content(file_path=str(TEST_FILE_PATH), start_line=3, end_line=5)
+    with open(TEST_FILE_PATH, "r") as f:
+        new_content = f.read()
+    assert new_content == EXPECTED_CONTENT_AFTER_DELETE
+
+
+EXPECTED_CONTENT_AFTER_INSERT = """
+# this is line one
+def test_function_for_fm():
+    This is the new line to be inserted, at line 3
+    "some docstring"
+    a = 1
+    b = 2
+    c = 3
+    # this is the 7th line
+""".strip()
+
+
+def test_insert_content(test_file):
+    FileManager().write_content(
+        file_path=str(TEST_FILE_PATH),
+        start_line=3,
+        end_line=-1,
+        new_block_content="    This is the new line to be inserted, at line 3",
+    )
+    with open(TEST_FILE_PATH, "r") as f:
+        new_content = f.read()
+    assert new_content == EXPECTED_CONTENT_AFTER_INSERT