dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-21 12:55:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: Update Coder import path in test_main.py

Browse source
This commit is contained in:
Paul Gauthier 2024-08-01 15:32:39 -03:00 • committed by Paul Gauthier (aider)
parent 387b2710b0
commit aad52eaae2
1 changed files with 12 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

24
tests/basic/test_main.py
View file

@ -75,13 +75,13 @@ class TestMain(TestCase):
make_repo() make_repo()
Path(".aider.conf.yml").write_text("auto-commits: false\n") Path(".aider.conf.yml").write_text("auto-commits: false\n")
with patch("aider.main.Coder.create") as MockCoder: with patch("aider.coders.Coder.create") as MockCoder:
main(["--yes"], input=DummyInput(), output=DummyOutput()) main(["--yes"], input=DummyInput(), output=DummyOutput())
_, kwargs = MockCoder.call_args _, kwargs = MockCoder.call_args
assert kwargs["auto_commits"] is False assert kwargs["auto_commits"] is False
Path(".aider.conf.yml").write_text("auto-commits: true\n") Path(".aider.conf.yml").write_text("auto-commits: true\n")
with patch("aider.main.Coder.create") as MockCoder: with patch("aider.coders.Coder.create") as MockCoder:
main([], input=DummyInput(), output=DummyOutput()) main([], input=DummyInput(), output=DummyOutput())
_, kwargs = MockCoder.call_args _, kwargs = MockCoder.call_args
assert kwargs["auto_commits"] is True assert kwargs["auto_commits"] is True
@ -132,41 +132,41 @@ class TestMain(TestCase):
del os.environ["GIT_CONFIG_GLOBAL"] del os.environ["GIT_CONFIG_GLOBAL"]
def test_main_args(self): def test_main_args(self):
with patch("aider.main.Coder.create") as MockCoder: with patch("aider.coders.Coder.create") as MockCoder:
# --yes will just ok the git repo without blocking on input # --yes will just ok the git repo without blocking on input
# following calls to main will see the new repo already # following calls to main will see the new repo already
main(["--no-auto-commits", "--yes"], input=DummyInput()) main(["--no-auto-commits", "--yes"], input=DummyInput())
_, 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.create") as MockCoder: with patch("aider.coders.Coder.create") as MockCoder:
main(["--auto-commits"], input=DummyInput()) main(["--auto-commits"], input=DummyInput())
_, 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.create") as MockCoder: with patch("aider.coders.Coder.create") as MockCoder:
main([], input=DummyInput()) main([], input=DummyInput())
_, 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.create") as MockCoder: with patch("aider.coders.Coder.create") as MockCoder:
main(["--no-pretty"], input=DummyInput()) main(["--no-pretty"], input=DummyInput())
_, kwargs = MockCoder.call_args _, kwargs = MockCoder.call_args
assert kwargs["pretty"] is False assert kwargs["pretty"] is False
with patch("aider.main.Coder.create") as MockCoder: with patch("aider.coders.Coder.create") as MockCoder:
main(["--pretty"], input=DummyInput()) main(["--pretty"], input=DummyInput())
_, kwargs = MockCoder.call_args _, kwargs = MockCoder.call_args
assert kwargs["pretty"] is True assert kwargs["pretty"] is True
with patch("aider.main.Coder.create") as MockCoder: with patch("aider.coders.Coder.create") as MockCoder:
main(["--no-dirty-commits"], input=DummyInput()) main(["--no-dirty-commits"], input=DummyInput())
_, 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.create") as MockCoder: with patch("aider.coders.Coder.create") as MockCoder:
main(["--dirty-commits"], input=DummyInput()) main(["--dirty-commits"], input=DummyInput())
_, kwargs = MockCoder.call_args _, kwargs = MockCoder.call_args
assert kwargs["dirty_commits"] is True assert kwargs["dirty_commits"] is True
@ -194,7 +194,7 @@ class TestMain(TestCase):
cwd_env.write_text("A=cwd\nB=cwd") cwd_env.write_text("A=cwd\nB=cwd")
named_env.write_text("A=named") named_env.write_text("A=named")
with patch('pathlib.Path.home', return_value=fake_home): with patch("pathlib.Path.home", return_value=fake_home):
main(["--yes", "--exit", "--env-file", str(named_env)]) main(["--yes", "--exit", "--env-file", str(named_env)])
self.assertEqual(os.environ["A"], "named") self.assertEqual(os.environ["A"], "named")
@ -209,7 +209,7 @@ class TestMain(TestCase):
with open(message_file_path, "w", encoding="utf-8") as message_file: with open(message_file_path, "w", encoding="utf-8") as message_file:
message_file.write(message_file_content) message_file.write(message_file_content)
with patch("aider.main.Coder.create") as MockCoder: with patch("aider.coders.Coder.create") as MockCoder:
MockCoder.return_value.run = MagicMock() MockCoder.return_value.run = MagicMock()
main( main(
["--yes", "--message-file", message_file_path], ["--yes", "--message-file", message_file_path],
@ -224,7 +224,7 @@ class TestMain(TestCase):
fname = "foo.py" fname = "foo.py"
with GitTemporaryDirectory(): with GitTemporaryDirectory():
with patch("aider.main.Coder.create") as MockCoder: # noqa: F841 with patch("aider.coders.Coder.create") as MockCoder: # noqa: F841
with patch("aider.main.InputOutput") as MockSend: with patch("aider.main.InputOutput") as MockSend:
def side_effect(*args, **kwargs): def side_effect(*args, **kwargs):