dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-29 08:44:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

fixed tests

Browse source
This commit is contained in:
Paul Gauthier 2023-06-20 17:31:17 -07:00
parent 124ab7dd99
commit f7ccdaa914
1 changed files with 8 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

16
tests/test_main.py
View file

@ -18,7 +18,7 @@ class TestMain(TestCase):
self.original_cwd = os.getcwd() self.original_cwd = os.getcwd()
self.tempdir = tempfile.mkdtemp() self.tempdir = tempfile.mkdtemp()
os.chdir(self.tempdir) os.chdir(self.tempdir)
self.patcher = patch("aider.main.Coder.check_model_availability") self.patcher = patch("aider.coders.base.check_model_availability")
self.mock_check = self.patcher.start() self.mock_check = self.patcher.start()
self.mock_check.return_value = True self.mock_check.return_value = True
@ -48,39 +48,39 @@ class TestMain(TestCase):
self.assertTrue(os.path.exists("foo.txt")) self.assertTrue(os.path.exists("foo.txt"))
def test_main_args(self): def test_main_args(self):
with patch("aider.main.Coder") as MockCoder: with patch("aider.main.Coder.create") as MockCoder:
main(["--no-auto-commits"]) main(["--no-auto-commits"])
_, kwargs = MockCoder.call_args _, kwargs = MockCoder.call_args
assert kwargs["auto_commits"] is False assert kwargs["auto_commits"] is False
with patch("aider.main.Coder") as MockCoder: with patch("aider.main.Coder.create") as MockCoder:
main(["--auto-commits"]) main(["--auto-commits"])
_, kwargs = MockCoder.call_args _, kwargs = MockCoder.call_args
assert kwargs["auto_commits"] is True assert kwargs["auto_commits"] is True
with patch("aider.main.Coder") as MockCoder: with patch("aider.main.Coder.create") as MockCoder:
main([]) main([])
_, kwargs = MockCoder.call_args _, kwargs = MockCoder.call_args
assert kwargs["dirty_commits"] is True assert kwargs["dirty_commits"] is True
assert kwargs["auto_commits"] is True assert kwargs["auto_commits"] is True
assert kwargs["pretty"] is True assert kwargs["pretty"] is True
with patch("aider.main.Coder") as MockCoder: with patch("aider.main.Coder.create") as MockCoder:
main(["--no-pretty"]) main(["--no-pretty"])
_, kwargs = MockCoder.call_args _, kwargs = MockCoder.call_args
assert kwargs["pretty"] is False assert kwargs["pretty"] is False
with patch("aider.main.Coder") as MockCoder: with patch("aider.main.Coder.create") as MockCoder:
main(["--pretty"]) main(["--pretty"])
_, kwargs = MockCoder.call_args _, kwargs = MockCoder.call_args
assert kwargs["pretty"] is True assert kwargs["pretty"] is True
with patch("aider.main.Coder") as MockCoder: with patch("aider.main.Coder.create") as MockCoder:
main(["--no-dirty-commits"]) main(["--no-dirty-commits"])
_, kwargs = MockCoder.call_args _, kwargs = MockCoder.call_args
assert kwargs["dirty_commits"] is False assert kwargs["dirty_commits"] is False
with patch("aider.main.Coder") as MockCoder: with patch("aider.main.Coder.create") as MockCoder:
main(["--dirty-commits"]) main(["--dirty-commits"])
_, kwargs = MockCoder.call_args _, kwargs = MockCoder.call_args
assert kwargs["dirty_commits"] is True assert kwargs["dirty_commits"] is True