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

Improvements: flag -f (not -mf); unitest improvements.

Browse source
This commit is contained in:
Your Name 2023-12-05 23:49:02 +01:00
parent 13ac5f0b60
commit 4692a689fe
2 changed files with 5 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

5
aider/main.py
View file

@ -382,7 +382,7 @@ def main(argv=None, input=None, output=None, force_git_root=None):
) )
other_group.add_argument( other_group.add_argument(
"--message-file", "--message-file",
"-mf", "-f",
metavar="MESSAGE_FILE", metavar="MESSAGE_FILE",
help="Specify a file containing the message to send GPT, process reply, then exit (disables chat mode)", help="Specify a file containing the message to send GPT, process reply, then exit (disables chat mode)",
) )
@ -571,8 +571,7 @@ def main(argv=None, input=None, output=None, force_git_root=None):
coder.run(with_message=args.message) coder.run(with_message=args.message)
elif args.message_file: elif args.message_file:
try: try:
with open(args.message_file, 'r', encoding='utf-8') as file: message_from_file = io.read_text(args.message_file)
message_from_file = file.read()
io.tool_output() io.tool_output()
coder.run(with_message=message_from_file) coder.run(with_message=message_from_file)
except FileNotFoundError: except FileNotFoundError:

7
tests/test_main.py
View file

@ -182,16 +182,15 @@ class TestMain(TestCase):
_, kwargs = MockCoder.call_args _, kwargs = MockCoder.call_args
assert kwargs["dirty_commits"] is True assert kwargs["dirty_commits"] is True
@patch('aider.main.InputOutput.confirm_ask', return_value=True) def test_message_file_flag(self):
def test_message_file_flag(self, mock_confirm):
message_file_content = "This is a test message from a file." message_file_content = "This is a test message from a file."
message_file_path = "test_message.txt" message_file_path = tempfile.mktemp()
with open(message_file_path, 'w', encoding='utf-8') as message_file: with open(message_file_path, 'w', encoding='utf-8') as message_file:
message_file.write(message_file_content) message_file.write(message_file_content)
with patch("aider.main.Coder.create") as MockCoder: with patch("aider.main.Coder.create") as MockCoder:
MockCoder.return_value.run = MagicMock() MockCoder.return_value.run = MagicMock()
main(["--message-file", message_file_path], input=DummyInput(), output=DummyOutput()) main(["--yes", "--message-file", message_file_path], input=DummyInput(), output=DummyOutput())
MockCoder.return_value.run.assert_called_once_with(with_message=message_file_content) MockCoder.return_value.run.assert_called_once_with(with_message=message_file_content)
os.remove(message_file_path) os.remove(message_file_path)