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

Use DummyInput in tests

Browse source
This commit is contained in:
Paul Gauthier 2023-07-05 14:20:28 -07:00
parent 9aaee49c51
commit b31dc2ac71
1 changed files with 11 additions and 33 deletions
Download patch file Download diff file Expand all files Collapse all files

44
tests/test_main.py
View file

@ -5,7 +5,7 @@ import tempfile
from unittest import TestCase
from unittest.mock import patch
from prompt_toolkit.input import create_pipe_input
from prompt_toolkit.input import DummyInput
from prompt_toolkit.output import DummyOutput
from aider.main import main
@ -27,75 +27,53 @@ class TestMain(TestCase):
self.patcher.stop()
def test_main_with_empty_dir_no_files_on_command(self):
with create_pipe_input() as pipe_input:
pipe_input.send_text("/exit\n")
try:
main([], input=pipe_input, output=DummyOutput())
except SystemExit:
pass
main([], input=DummyInput(), output=DummyOutput())
def test_main_with_empty_dir_new_file(self):
with create_pipe_input() as pipe_input:
pipe_input.send_text("/exit\n")
try:
main(["foo.txt"], input=pipe_input, output=DummyOutput())
except SystemExit:
pass
main(["foo.txt"], input=DummyInput(), output=DummyOutput())
self.assertTrue(os.path.exists("foo.txt"))
def test_main_with_empty_git_dir_new_file(self):
subprocess.run(["git", "init"])
subprocess.run(["git", "config", "user.email", "dummy@example.com"])
subprocess.run(["git", "config", "user.name", "Dummy User"])
with create_pipe_input() as pipe_input:
pipe_input.send_text("/exit\n")
try:
main(["--yes", "foo.txt"], input=pipe_input, output=DummyOutput())
except SystemExit:
pass
main(["--yes", "foo.txt"], input=DummyInput(), output=DummyOutput())
self.assertTrue(os.path.exists("foo.txt"))
def test_main_args(self):
with patch("aider.main.Coder.create") as MockCoder:
with create_pipe_input() as pipe_input:
main(["--no-auto-commits"], input=pipe_input)
main(["--no-auto-commits"], input=DummyInput())
_, kwargs = MockCoder.call_args
assert kwargs["auto_commits"] is False
with patch("aider.main.Coder.create") as MockCoder:
with create_pipe_input() as pipe_input:
main(["--auto-commits"], input=pipe_input)
main(["--auto-commits"], input=DummyInput())
_, kwargs = MockCoder.call_args
assert kwargs["auto_commits"] is True
with patch("aider.main.Coder.create") as MockCoder:
with create_pipe_input() as pipe_input:
main([], input=pipe_input)
main([], input=DummyInput())
_, 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.create") as MockCoder:
with create_pipe_input() as pipe_input:
main(["--no-pretty"], input=pipe_input)
main(["--no-pretty"], input=DummyInput())
_, kwargs = MockCoder.call_args
assert kwargs["pretty"] is False
with patch("aider.main.Coder.create") as MockCoder:
with create_pipe_input() as pipe_input:
main(["--pretty"], input=pipe_input)
main(["--pretty"], input=DummyInput())
_, kwargs = MockCoder.call_args
assert kwargs["pretty"] is True
with patch("aider.main.Coder.create") as MockCoder:
with create_pipe_input() as pipe_input:
main(["--no-dirty-commits"], input=pipe_input)
main(["--no-dirty-commits"], input=DummyInput())
_, kwargs = MockCoder.call_args
assert kwargs["dirty_commits"] is False
with patch("aider.main.Coder.create") as MockCoder:
with create_pipe_input() as pipe_input:
main(["--dirty-commits"], input=pipe_input)
main(["--dirty-commits"], input=DummyInput())
_, kwargs = MockCoder.call_args
assert kwargs["dirty_commits"] is True