fix: Remove unused mock_stdout in test_main

This commit is contained in:
Paul Gauthier (aider) 2025-06-08 10:19:01 -07:00
parent 1953c9815a
commit 0ac6068e1d

View file

@ -180,16 +180,15 @@ class TestMain(TestCase):
self.assertNotIn(abs_ignored_file, coder.abs_fnames) self.assertNotIn(abs_ignored_file, coder.abs_fnames)
# Test with --add-gitignore-files set to True # Test with --add-gitignore-files set to True
with patch("sys.stdout", new_callable=StringIO) as mock_stdout: coder = main(
coder = main( ["--add-gitignore-files", "--exit", "--yes", abs_ignored_file],
["--add-gitignore-files", "--exit", "--yes", abs_ignored_file], input=DummyInput(),
input=DummyInput(), output=DummyOutput(),
output=DummyOutput(), return_coder=True,
return_coder=True, force_git_root=git_dir,
force_git_root=git_dir, )
) # Verify the ignored file is in the chat
# Verify the ignored file is in the chat self.assertIn(abs_ignored_file, coder.abs_fnames)
self.assertIn(abs_ignored_file, coder.abs_fnames)
# Test with --add-gitignore-files set to False # Test with --add-gitignore-files set to False
coder = main( coder = main(
@ -235,19 +234,18 @@ class TestMain(TestCase):
self.assertNotIn(abs_ignored_file, coder.abs_fnames) self.assertNotIn(abs_ignored_file, coder.abs_fnames)
# Test with --add-gitignore-files set to True # Test with --add-gitignore-files set to True
with patch("sys.stdout", new_callable=StringIO) as mock_stdout: coder = main(
coder = main( ["--add-gitignore-files", "--exit", "--yes"],
["--add-gitignore-files", "--exit", "--yes"], input=DummyInput(),
input=DummyInput(), output=DummyOutput(),
output=DummyOutput(), return_coder=True,
return_coder=True, force_git_root=git_dir,
force_git_root=git_dir, )
) with patch.object(coder.io, "confirm_ask", return_value=True):
with patch.object(coder.io, "confirm_ask", return_value=True): coder.commands.cmd_add(rel_ignored_file)
coder.commands.cmd_add(rel_ignored_file)
# Verify the ignored file is in the chat # Verify the ignored file is in the chat
self.assertIn(abs_ignored_file, coder.abs_fnames) self.assertIn(abs_ignored_file, coder.abs_fnames)
# Test with --add-gitignore-files set to False # Test with --add-gitignore-files set to False
coder = main( coder = main(