From 3424cda63cef0907dd3f33b6804d689b75e97bca Mon Sep 17 00:00:00 2001
From: "Paul Gauthier (aider)" <aider@paulg.org>
Date: Mon, 19 Aug 2024 16:06:34 -0700
Subject: [PATCH] feat: Add test for main() with --sonnet and --cache-prompts
 options

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

diff --git a/tests/basic/test_main.py b/tests/basic/test_main.py
index 91b8f86e5..eadc0b1b8 100644
--- a/tests/basic/test_main.py
+++ b/tests/basic/test_main.py
@@ -533,3 +533,15 @@ class TestMain(TestCase):
                 self.assertEqual(
                     call_kwargs.get("refresh"), "files"
                 )  # Check the 'refresh' keyword argument
+
+    def test_sonnet_and_cache_prompts_options(self):
+        with GitTemporaryDirectory():
+            coder = main(
+                ["--sonnet", "--cache-prompts", "--exit", "--yes"],
+                input=DummyInput(),
+                output=DummyOutput(),
+                return_coder=True,
+            )
+
+            self.assertTrue(coder.add_cache_headers)
+            self.assertEqual(coder.main_model.name, "gpt-4-1106-preview")