From 75a7a0043af1f1f9c2fb9624bbc23b6234f817de Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Mon, 19 Aug 2024 16:05:02 -0700 Subject: [PATCH] feat: Add test for --sonnet --cache-prompts --exit options --- tests/basic/test_main.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/tests/basic/test_main.py b/tests/basic/test_main.py index 91b8f86e5..55b79a050 100644 --- a/tests/basic/test_main.py +++ b/tests/basic/test_main.py @@ -533,3 +533,21 @@ 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(): + with patch("aider.coders.Coder.create") as MockCoder: + mock_coder = MagicMock() + MockCoder.return_value = mock_coder + + main( + ["--sonnet", "--cache-prompts", "--exit", "--yes"], + input=DummyInput(), + output=DummyOutput(), + ) + + MockCoder.assert_called_once() + _, kwargs = MockCoder.call_args + self.assertEqual(kwargs["main_model"].name, "gpt-4-1106-preview") + self.assertTrue(kwargs["cache_prompts"]) + self.assertTrue(mock_coder.add_cache_headers)