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

Added a new test case for cmd_undo with a newly committed file.

Browse source
This commit is contained in:
Paul Gauthier 2024-07-18 16:17:20 +01:00 • committed by Paul Gauthier (aider)
parent 9aeeaf383f
commit e7f29c2dbc
1 changed files with 36 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

38
tests/basic/test_commands.py
View file

@ -567,6 +567,13 @@ class TestCommands(TestCase):
coder = Coder.create(self.GPT35, None, io) coder = Coder.create(self.GPT35, None, io)
commands = Commands(io, coder) commands = Commands(io, coder)
# Put in a random first commit
filename = "first_file.txt"
file_path = Path(repo_dir) / filename
file_path.write_text("new file content")
repo.git.add(filename)
repo.git.commit("-m", "Add new file")
# Create and commit a new file # Create and commit a new file
filename = "new_file.txt" filename = "new_file.txt"
file_path = Path(repo_dir) / filename file_path = Path(repo_dir) / filename
@ -579,8 +586,35 @@ class TestCommands(TestCase):
coder.aider_commit_hashes.add(last_commit_hash) coder.aider_commit_hashes.add(last_commit_hash)
# Attempt to undo the last commit # Attempt to undo the last commit
with self.assertRaises(git.exc.GitCommandError): commands.cmd_undo("")
commands.cmd_undo("")
# Check that the last commit was undone
self.assertNotEqual(last_commit_hash, repo.head.commit.hexsha[:7])
del coder
del commands
del repo
def test_cmd_undo_on_first_commit(self):
with GitTemporaryDirectory() as repo_dir:
repo = git.Repo(repo_dir)
io = InputOutput(pretty=False, yes=True)
coder = Coder.create(self.GPT35, None, io)
commands = Commands(io, coder)
# Create and commit a new file
filename = "new_file.txt"
file_path = Path(repo_dir) / filename
file_path.write_text("new file content")
repo.git.add(filename)
repo.git.commit("-m", "Add new file")
# Store the commit hash
last_commit_hash = repo.head.commit.hexsha[:7]
coder.aider_commit_hashes.add(last_commit_hash)
# Attempt to undo the last commit
commands.cmd_undo("")
# Check that the last commit is still present # Check that the last commit is still present
self.assertEqual(last_commit_hash, repo.head.commit.hexsha[:7]) self.assertEqual(last_commit_hash, repo.head.commit.hexsha[:7])