From d09e85adc2c0346c3701b881e6c57435fc7e9151 Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Fri, 18 Aug 2023 07:55:42 -0700 Subject: [PATCH] Also ensure commit msg only depends on diffs from the GPT edited file. --- tests/test_coder.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/tests/test_coder.py b/tests/test_coder.py index ad9f63e0e..f489b74ae 100644 --- a/tests/test_coder.py +++ b/tests/test_coder.py @@ -406,7 +406,10 @@ new self.assertEqual(num_commits, 1) def test_only_commit_gpt_edited_file(self): - """Only commit file that gpt edits, not other dirty files""" + """ + Only commit file that gpt edits, not other dirty files. + Also ensure commit msg only depends on diffs from the GPT edited file. + """ with GitTemporaryDirectory(): repo = git.Repo() @@ -441,9 +444,13 @@ TWO """ coder.partial_response_function_call = dict() + def mock_get_commit_message(diffs, context): + self.assertNotIn("one", diffs) + self.assertNotIn("ONE", diffs) + return "commit message" + coder.send = MagicMock(side_effect=mock_send) - coder.repo.get_commit_message = MagicMock() - coder.repo.get_commit_message.return_value = "commit message" + coder.repo.get_commit_message = MagicMock(side_effect=mock_get_commit_message) coder.run(with_message="hi")