mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-21 12:55:00 +00:00
aider: Replaced manual setting of abs_fnames
with a mocked git.Repo
object that has its ls_files
method return the desired file paths.
# Context: USER: don't set the files in abs_fnames. patch self.repo.git.ls_files to return them ASSISTANT: To patch `self.repo.git.ls_files` to return the files in `abs_fnames`, we will modify the `test_check_for_file_mentions` function in the `tests/test_coder.py` file. ```python tests/test_coder.py <<<<<<< ORIGINAL # Initialize the Coder object with the mocked IO coder = Coder(io=mock_io, openai_api_key="fake_key") # Add sample files to the Coder object coder.abs_fnames = { "/path/to/repo/file1.txt", "/path/to/repo/file2.py", } coder.root = "/path/to/repo" ======= # Mock the git repo mock_repo = MagicMock() mock_repo.git.ls_files.return_value = "file1.txt\nfile2.py" # Initialize the Coder object with the mocked IO and mocked repo coder = Coder(io=mock_io, openai_api_key="fake_key", repo=mock_repo) coder.root = "/path/to/repo" >>>>>>> UPDATED ``` This change replaces the manual setting of `abs_fnames` with a mocked `git.Repo` object that has its `ls_files` method return the desired file paths.
This commit is contained in:
parent
475f85bcc3
commit
0fc40df8e6
1 changed files with 5 additions and 7 deletions
|
@ -9,14 +9,12 @@ class TestCoder(unittest.TestCase):
|
|||
mock_io.get_input.return_value = "Please check file1.txt and file2.py"
|
||||
mock_io.confirm_ask.return_value = True
|
||||
|
||||
# Initialize the Coder object with the mocked IO
|
||||
coder = Coder(io=mock_io, openai_api_key="fake_key")
|
||||
# Mock the git repo
|
||||
mock_repo = MagicMock()
|
||||
mock_repo.git.ls_files.return_value = "file1.txt\nfile2.py"
|
||||
|
||||
# Add sample files to the Coder object
|
||||
coder.abs_fnames = {
|
||||
"/path/to/repo/file1.txt",
|
||||
"/path/to/repo/file2.py",
|
||||
}
|
||||
# Initialize the Coder object with the mocked IO and mocked repo
|
||||
coder = Coder(io=mock_io, openai_api_key="fake_key", repo=mock_repo)
|
||||
coder.root = "/path/to/repo"
|
||||
|
||||
# Call the check_for_file_mentions method
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue