cleaned up --no-autocommit test

This commit is contained in:
Paul Gauthier 2023-05-27 05:03:30 -07:00
parent 2bb1c34bd5
commit 24e78229b2

View file

@ -2,7 +2,6 @@ import os
import tempfile import tempfile
import unittest import unittest
from unittest import TestCase from unittest import TestCase
from unittest.mock import MagicMock
from aider.main import main from aider.main import main
import subprocess import subprocess
from prompt_toolkit.input import create_input from prompt_toolkit.input import create_input
@ -18,16 +17,6 @@ class TestMain(TestCase):
main([], input=pipe_input, output=DummyOutput()) main([], input=pipe_input, output=DummyOutput())
pipe_input.close() 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): def test_main_with_empty_dir_new_file(self):
with tempfile.TemporaryDirectory() as temp_dir: with tempfile.TemporaryDirectory() as temp_dir:
os.chdir(temp_dir) os.chdir(temp_dir)
@ -44,3 +33,11 @@ class TestMain(TestCase):
main(["--yes", "foo.txt"], input=pipe_input, output=DummyOutput()) main(["--yes", "foo.txt"], input=pipe_input, output=DummyOutput())
pipe_input.close() pipe_input.close()
self.assertTrue(os.path.exists("foo.txt")) 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