dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-25 14:55:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

Refactor test_get_commit_message to use the correct import and handle the return value of mock_send correctly.

Browse source
This commit is contained in:
Paul Gauthier 2023-07-21 14:27:31 -03:00
parent 6f3b498228
commit a789de9433
1 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
tests/test_repo.py
View file

@ -15,10 +15,13 @@ from tests.utils import GitTemporaryDirectory
class TestRepo(unittest.TestCase): class TestRepo(unittest.TestCase):
@patch("aider.sendchat.send_with_retries") @patch("aider.repo.send_with_retries")
def test_get_commit_message(self, mock_send): def test_get_commit_message(self, mock_send):
# Set the return value of the mocked function # Set the return value of the mocked function
mock_send.return_value = "a good commit message" mock_send.return_value = (
None,
"a good commit message"
)
repo = AiderRepo(InputOutput(), None) repo = AiderRepo(InputOutput(), None)
# Call the get_commit_message method with dummy diff and context # Call the get_commit_message method with dummy diff and context