feat: Add patch for InputOutput in test_main_exit_calls_version_check

This commit is contained in:
Paul Gauthier (aider) 2024-08-13 05:29:18 -07:00
parent fcf758527a
commit cef421dfc0

View file

@ -226,9 +226,11 @@ class TestMain(TestCase):
def test_main_exit_calls_version_check(self):
with GitTemporaryDirectory():
with patch("aider.main.check_version") as mock_check_version:
with patch("aider.main.check_version") as mock_check_version, \
patch("aider.main.InputOutput") as mock_input_output:
main(["--exit"], input=DummyInput(), output=DummyOutput())
mock_check_version.assert_called_once()
mock_input_output.assert_called_once()
@patch("aider.main.InputOutput")
@patch("aider.coders.base_coder.Coder.run")