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

test: update main tests to handle None input and add --exit flag

Browse source
This commit is contained in:
Paul Gauthier 2024-09-10 15:32:43 -07:00 • committed by Paul Gauthier (aider)
parent fa49ab09c4
commit 0a3b4147fa
1 changed files with 11 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

16
tests/basic/test_main.py
View file

@ -29,7 +29,7 @@ class TestMain(TestCase):
# Fake home directory prevents tests from using the real ~/.aider.conf.yml file:
self.homedir_obj = IgnorantTemporaryDirectory()
os.environ["HOME"] = self.homedir_obj.name
self.input_patcher = patch("builtins.input", return_value="")
self.input_patcher = patch("builtins.input", return_value=None)
self.mock_input = self.input_patcher.start()
def tearDown(self):
@ -274,7 +274,8 @@ class TestMain(TestCase):
def test_dark_mode_sets_code_theme(self):
# Mock InputOutput to capture the configuration
with patch("aider.main.InputOutput") as MockInputOutput:
main(["--dark-mode", "--no-git"], input=DummyInput(), output=DummyOutput())
MockInputOutput.return_value.get_input.return_value = None
main(["--dark-mode", "--no-git", "--exit"], input=DummyInput(), output=DummyOutput())
# Ensure InputOutput was called
MockInputOutput.assert_called_once()
# Check if the code_theme setting is for dark mode
@ -284,7 +285,8 @@ class TestMain(TestCase):
def test_light_mode_sets_code_theme(self):
# Mock InputOutput to capture the configuration
with patch("aider.main.InputOutput") as MockInputOutput:
main(["--light-mode", "--no-git"], input=DummyInput(), output=DummyOutput())
MockInputOutput.return_value.get_input.return_value = None
main(["--light-mode", "--no-git", "--exit"], input=DummyInput(), output=DummyOutput())
# Ensure InputOutput was called
MockInputOutput.assert_called_once()
# Check if the code_theme setting is for light mode
@ -299,8 +301,10 @@ class TestMain(TestCase):
def test_env_file_flag_sets_automatic_variable(self):
env_file_path = self.create_env_file(".env.test", "AIDER_DARK_MODE=True")
with patch("aider.main.InputOutput") as MockInputOutput:
MockInputOutput.return_value.get_input.return_value = None
MockInputOutput.return_value.get_input.confirm_ask = True
main(
["--env-file", str(env_file_path), "--no-git"],
["--env-file", str(env_file_path), "--no-git", "--exit"],
input=DummyInput(),
output=DummyOutput(),
)
@ -312,7 +316,9 @@ class TestMain(TestCase):
def test_default_env_file_sets_automatic_variable(self):
self.create_env_file(".env", "AIDER_DARK_MODE=True")
with patch("aider.main.InputOutput") as MockInputOutput:
main(["--no-git"], input=DummyInput(), output=DummyOutput())
MockInputOutput.return_value.get_input.return_value = None
MockInputOutput.return_value.get_input.confirm_ask = True
main(["--no-git", "--exit"], input=DummyInput(), output=DummyOutput())
# Ensure InputOutput was called
MockInputOutput.assert_called_once()
# Check if the color settings are for dark mode