dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-20 12:24:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor: sort new file mentions before confirmation

Browse source
This commit is contained in:
Paul Gauthier 2024-08-28 17:59:04 -07:00 • committed by Paul Gauthier (aider)
parent 41a43fc313
commit e45def7a98
2 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/basic/test_coder.py
View file

@ -199,13 +199,13 @@ class TestCoder(unittest.TestCase):
coder = Coder.create(self.GPT35, None, io)
# Mock get_file_mentions to return two file names
coder.get_file_mentions = MagicMock(return_value=["file1.txt", "file2.txt"])
coder.get_file_mentions = MagicMock(return_value=set(["file1.txt", "file2.txt"]))
# Mock confirm_ask to return False for the first call and True for the second
io.confirm_ask = MagicMock(side_effect=[False, True])
io.confirm_ask = MagicMock(side_effect=[False, True, True])
# First call to check_for_file_mentions
coder.check_for_file_mentions("Please check file1.txt and file2.txt")
coder.check_for_file_mentions("Please check file1.txt for the info")
# Assert that confirm_ask was called twice
self.assertEqual(io.confirm_ask.call_count, 2)