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

tests/basic/test_commands.py

@@ -561,6 +561,41 @@ class TestCommands(TestCase):
             del repo
 
     def test_cmd_undo_with_newly_committed_file(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)
+
+            # 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
+            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 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)
@@ -579,7 +614,6 @@ class TestCommands(TestCase):
             coder.aider_commit_hashes.add(last_commit_hash)
 
             # Attempt to undo the last commit
-            with self.assertRaises(git.exc.GitCommandError):
             commands.cmd_undo("")
 
             # Check that the last commit is still present