From a30a27fa8ae69da0bafbb774c5393db203ce339b Mon Sep 17 00:00:00 2001
From: "Paul Gauthier (aider)" <aider@paulg.org>
Date: Tue, 13 Aug 2024 05:25:36 -0700
Subject: [PATCH] feat: add test for main --exit that confirms version_check is
 called

---
 tests/basic/test_main.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/tests/basic/test_main.py b/tests/basic/test_main.py
index f8322d2ef..b007ee35e 100644
--- a/tests/basic/test_main.py
+++ b/tests/basic/test_main.py
@@ -224,6 +224,11 @@ class TestMain(TestCase):
 
                     main(["--yes", fname, "--encoding", "iso-8859-15"])
 
+    def test_main_exit_calls_version_check(self):
+        with patch("aider.main.check_version") as mock_check_version:
+            main(["--exit"])
+            mock_check_version.assert_called_once()
+
     @patch("aider.main.InputOutput")
     @patch("aider.coders.base_coder.Coder.run")
     def test_main_message_adds_to_input_history(self, mock_run, MockInputOutput):