From 3e81cb1c6b3506f21525108daf41b6bc1971116d Mon Sep 17 00:00:00 2001 From: Your Name Date: Tue, 28 Nov 2023 01:52:35 +0100 Subject: [PATCH] Fixes #374 - test_main.py += unit test for --message flag. Fixes https://github.com/paul-gauthier/aider/issues/374 --- tests/test_main.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/tests/test_main.py b/tests/test_main.py index d16e1672d..c02204217 100644 --- a/tests/test_main.py +++ b/tests/test_main.py @@ -196,3 +196,13 @@ class TestMain(TestCase): MockSend.side_effect = side_effect main(["--yes", fname, "--encoding", "iso-8859-15"]) + @patch("aider.main.InputOutput") + @patch("openai.ChatCompletion.create") + def test_main_message_adds_to_input_history(self, mock_chat_completion, MockInputOutput): + test_message = "test message" + mock_io_instance = MockInputOutput.return_value + mock_chat_completion.return_value = MagicMock(choices=[{"text": "mocked response"}]) + + main(["--message", test_message], input=DummyInput(), output=DummyOutput()) + + mock_io_instance.add_to_input_history.assert_called_once_with(test_message)