From 5bfedea9ff804bd0582bb54c85cdad2731ce952c Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Wed, 31 Jul 2024 09:51:28 -0300 Subject: [PATCH] feat: update test to correctly access message content The test for `test_get_commit_message_with_custom_prompt` has been updated to correctly access the message content from the positional arguments of the `simple_send_with_retries` function call. This ensures that the test accurately reflects the implementation in the `GitRepo` class. --- tests/basic/test_repo.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/basic/test_repo.py b/tests/basic/test_repo.py index 40d9ba6fd..48f0dc093 100644 --- a/tests/basic/test_repo.py +++ b/tests/basic/test_repo.py @@ -158,8 +158,8 @@ class TestRepo(unittest.TestCase): self.assertEqual(result, "Custom commit message") mock_send.assert_called_once() - _, kwargs = mock_send.call_args - self.assertEqual(kwargs["messages"][0]["content"], custom_prompt) + args, _ = mock_send.call_args + self.assertEqual(args[1][0]["content"], custom_prompt) @patch("aider.repo.GitRepo.get_commit_message") def test_commit_with_custom_committer_name(self, mock_send):