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

Refactor coder.get_tracked_files to coder.repo.get_tracked_files in test_coder.py and add test_get_commit_message_no_strip_unmatched_quotes in test_repo.py.

Browse source
This commit is contained in:
Paul Gauthier 2023-07-21 14:36:26 -03:00
parent e9090f8d61
commit 1fcb090ff6
2 changed files with 23 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

28
tests/test_coder.py
View file

@ -53,7 +53,7 @@ class TestCoder(unittest.TestCase):
# Mock the git repo
mock = MagicMock()
mock.return_value = set(["file1.txt", "file2.py"])
coder.get_tracked_files = mock
coder.repo.get_tracked_files = mock
# Call the check_for_file_mentions method
coder.check_for_file_mentions("Please check file1.txt and file2.py")
@ -97,7 +97,7 @@ class TestCoder(unittest.TestCase):
mock = MagicMock()
mock.return_value = set(["file1.txt", "file2.py"])
coder.get_tracked_files = mock
coder.repo.get_tracked_files = mock
# Call the check_for_file_mentions method
coder.check_for_file_mentions("Please check file1.txt and file2.py")
@ -128,7 +128,7 @@ class TestCoder(unittest.TestCase):
mock = MagicMock()
mock.return_value = set([str(fname), str(other_fname)])
coder.get_tracked_files = mock
coder.repo.get_tracked_files = mock
# Call the check_for_file_mentions method
coder.check_for_file_mentions(f"Please check {fname}!")
@ -146,7 +146,7 @@ class TestCoder(unittest.TestCase):
mock = MagicMock()
mock.return_value = set([str(fname)])
coder.get_tracked_files = mock
coder.repo.get_tracked_files = mock
dump(fname)
# Call the check_for_file_mentions method
@ -154,26 +154,6 @@ class TestCoder(unittest.TestCase):
self.assertEqual(coder.abs_fnames, set([str(fname.resolve())]))
def test_get_commit_message_no_strip_unmatched_quotes(self):
# Mock the IO object
mock_io = MagicMock()
# Initialize the Coder object with the mocked IO and mocked repo
coder = Coder.create(models.GPT4, None, mock_io)
# Mock the send method to set partial_response_content and return False
def mock_send(*args, **kwargs):
coder.partial_response_content = 'a good "commit message"'
return False
coder.send = MagicMock(side_effect=mock_send)
# Call the get_commit_message method with dummy diff and context
result = coder.get_commit_message("dummy diff", "dummy context")
# Assert that the returned message is the expected one
self.assertEqual(result, 'a good "commit message"')
def test_run_with_file_deletion(self):
# Create a few temporary files

19
tests/test_repo.py
View file

@ -51,6 +51,25 @@ class TestRepo(unittest.TestCase):
# Assert that the returned message is the expected one
self.assertEqual(result, "a good commit message")
@patch("aider.repo.send_with_retries")
def test_get_commit_message_no_strip_unmatched_quotes(self, mock_send):
# Set the return value of the mocked function
mock_response = MagicMock()
mock_response.choices = [MagicMock()]
mock_response.choices[0].message.content = 'a good "commit message"'
mock_send.return_value = (
None,
mock_response
)
repo = AiderRepo(InputOutput(), None)
# Call the get_commit_message method with dummy diff and context
result = repo.get_commit_message("dummy diff", "dummy context")
# Assert that the returned message is the expected one
self.assertEqual(result, 'a good "commit message"')
def test_get_tracked_files(self):
# Create a temporary directory
tempdir = Path(tempfile.mkdtemp())