From b70e0bd1f6ea0b19570031d19d679044755093e6 Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Tue, 26 Nov 2024 07:07:11 -0800 Subject: [PATCH] test: update invalid edit format test to check return code instead of SystemExit --- tests/basic/test_main.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/tests/basic/test_main.py b/tests/basic/test_main.py index 92ba2d332..772ea04d2 100644 --- a/tests/basic/test_main.py +++ b/tests/basic/test_main.py @@ -670,12 +670,12 @@ class TestMain(TestCase): def test_invalid_edit_format(self): with GitTemporaryDirectory(): with patch("aider.io.InputOutput.offer_url") as mock_offer_url: - with self.assertRaises(SystemExit): - main( - ["--edit-format", "not-a-real-format", "--exit", "--yes"], - input=DummyInput(), - output=DummyOutput(), - ) + result = main( + ["--edit-format", "not-a-real-format", "--exit", "--yes"], + input=DummyInput(), + output=DummyOutput(), + ) + self.assertEqual(result, 1) # main() should return 1 on error mock_offer_url.assert_called_once() args, _ = mock_offer_url.call_args self.assertEqual(args[0], "https://aider.chat/docs/edit-formats.html")