diff --git a/tests/test_main.py b/tests/test_main.py index b2aa337e2..9168f71e9 100644 --- a/tests/test_main.py +++ b/tests/test_main.py @@ -2,7 +2,6 @@ import os import tempfile import unittest from unittest import TestCase -from unittest.mock import MagicMock from aider.main import main import subprocess from prompt_toolkit.input import create_input @@ -18,16 +17,6 @@ class TestMain(TestCase): main([], input=pipe_input, output=DummyOutput()) pipe_input.close() - def test_main_with_no_auto_commits(self): - with tempfile.TemporaryDirectory() as temp_dir: - os.chdir(temp_dir) - pipe_input = create_input(StringIO("")) - with unittest.mock.patch("aider.main.Coder") as MockCoder: - main(["--no-auto-commits"], input=pipe_input, output=DummyOutput()) - _, kwargs = MockCoder.call_args - assert kwargs["auto_commits"] is False - pipe_input.close() - def test_main_with_empty_dir_new_file(self): with tempfile.TemporaryDirectory() as temp_dir: os.chdir(temp_dir) @@ -44,3 +33,11 @@ class TestMain(TestCase): main(["--yes", "foo.txt"], input=pipe_input, output=DummyOutput()) pipe_input.close() self.assertTrue(os.path.exists("foo.txt")) + + def test_main_with_no_auto_commits(self): + with tempfile.TemporaryDirectory() as temp_dir: + os.chdir(temp_dir) + with unittest.mock.patch("aider.main.Coder") as MockCoder: + main(["--no-auto-commits"]) + _, kwargs = MockCoder.call_args + assert kwargs["auto_commits"] is False