mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-24 22:34:59 +00:00
Added a new test to check for filename mentions of longer paths.
This commit is contained in:
parent
7e75cfb1fa
commit
c8cf78b5cf
1 changed files with 21 additions and 0 deletions
|
@ -26,6 +26,27 @@ class TestCoder(unittest.TestCase):
|
||||||
expected_files = {os.path.abspath("file1.txt"), os.path.abspath("file2.py")}
|
expected_files = {os.path.abspath("file1.txt"), os.path.abspath("file2.py")}
|
||||||
self.assertEqual(coder.abs_fnames, expected_files)
|
self.assertEqual(coder.abs_fnames, expected_files)
|
||||||
|
|
||||||
|
def test_check_for_filename_mentions_of_longer_paths(self):
|
||||||
|
# Mock the IO object
|
||||||
|
mock_io = MagicMock()
|
||||||
|
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 and mocked repo
|
||||||
|
coder = Coder(io=mock_io, openai_api_key="fake_key")
|
||||||
|
|
||||||
|
# Mock the git repo
|
||||||
|
mock_repo = MagicMock()
|
||||||
|
mock_repo.git.ls_files.return_value = "path/to/file1.txt\npath/to/file2.py"
|
||||||
|
coder.repo = mock_repo
|
||||||
|
|
||||||
|
# Call the check_for_file_mentions method
|
||||||
|
coder.check_for_file_mentions("Please check file1.txt and file2.py")
|
||||||
|
|
||||||
|
# Check if coder.abs_fnames contains both files
|
||||||
|
expected_files = {os.path.abspath("file1.txt"), os.path.abspath("file2.py")}
|
||||||
|
self.assertEqual(coder.abs_fnames, expected_files)
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
unittest.main()
|
unittest.main()
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue