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

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.tempdir = tempfile.mkdtemp()
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.return_value = True
@ -48,39 +48,39 @@ class TestMain(TestCase):
self.assertTrue(os.path.exists("foo.txt"))
def test_main_args(self):
with patch("aider.main.Coder") as MockCoder:
with patch("aider.main.Coder.create") as MockCoder:
main(["--no-auto-commits"])
_, kwargs = MockCoder.call_args
assert kwargs["auto_commits"] is False
with patch("aider.main.Coder") as MockCoder:
with patch("aider.main.Coder.create") as MockCoder:
main(["--auto-commits"])
_, kwargs = MockCoder.call_args
assert kwargs["auto_commits"] is True
with patch("aider.main.Coder") as MockCoder:
with patch("aider.main.Coder.create") as MockCoder:
main([])
_, kwargs = MockCoder.call_args
assert kwargs["dirty_commits"] is True
assert kwargs["auto_commits"] 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"])
_, kwargs = MockCoder.call_args
assert kwargs["pretty"] is False
with patch("aider.main.Coder") as MockCoder:
with patch("aider.main.Coder.create") as MockCoder:
main(["--pretty"])
_, kwargs = MockCoder.call_args
assert kwargs["pretty"] is True
with patch("aider.main.Coder") as MockCoder:
with patch("aider.main.Coder.create") as MockCoder:
main(["--no-dirty-commits"])
_, kwargs = MockCoder.call_args
assert kwargs["dirty_commits"] is False
with patch("aider.main.Coder") as MockCoder:
with patch("aider.main.Coder.create") as MockCoder:
main(["--dirty-commits"])
_, kwargs = MockCoder.call_args
assert kwargs["dirty_commits"] is True