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

Bugfix reference to self.get_tracked_files, plus test coverage

Browse source
This commit is contained in:
Paul Gauthier 2023-07-27 06:55:06 -03:00
parent c6f99c9b66
commit 832259b55b
2 changed files with 33 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

31
tests/test_coder.py
View file

@ -22,6 +22,37 @@ class TestCoder(unittest.TestCase):
def tearDown(self):
self.patcher.stop()
def test_allowed_to_edit(self):
with GitTemporaryDirectory():
repo = git.Repo(Path.cwd())
fname = Path("foo.txt")
fname.touch()
repo.git.add(str(fname))
repo.git.commit("-m", "init")
io = InputOutput(yes=True)
# Initialize the Coder object with the mocked IO and mocked repo
coder = Coder.create(models.GPT4, None, io, fnames=["foo.txt"])
self.assertTrue(coder.allowed_to_edit("foo.txt"))
self.assertTrue(coder.allowed_to_edit("new.txt"))
def test_allowed_to_edit_no(self):
with GitTemporaryDirectory():
repo = git.Repo(Path.cwd())
fname = Path("foo.txt")
fname.touch()
repo.git.add(str(fname))
repo.git.commit("-m", "init")
# say NO
io = InputOutput(yes=False)
coder = Coder.create(models.GPT4, None, io, fnames=["foo.txt"])
self.assertTrue(coder.allowed_to_edit("foo.txt"))
self.assertFalse(coder.allowed_to_edit("new.txt"))
def test_get_last_modified(self):
# Mock the IO object
mock_io = MagicMock()