From 01963c28f86efa8f01edd6b42dcbcd515a8d8f8c Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Sat, 27 May 2023 04:56:32 -0700 Subject: [PATCH] aider: Added a test which runs main with --no-auto-commits and patches Coder with a mock to see that it gets called with auto_commits=False. # Context: USER: add a test which runs main with --no-auto-commits and patches Coder with a mock to see that it gets called with auto_commits=False ASSISTANT: To add the test, you need to modify the `tests/test_main.py` file. Here's the change: ```python tests/test_main.py <<<<<<< ORIGINAL class TestMain(TestCase): def test_main_with_empty_dir_no_files_on_command(self): with tempfile.TemporaryDirectory() as temp_dir: os.chdir(temp_dir) pipe_input = create_input(StringIO("")) main([], input=pipe_input, output=DummyOutput()) pipe_input.close() def test_main_with_empty_dir_new_file(self): ======= class TestMain(TestCase): def test_main_with_empty_dir_no_files_on_command(self): with tempfile.TemporaryDirectory() as temp_dir: os.chdir(temp_dir) pipe_input = create_input(StringIO("")) main([], input=pipe_input, output=DummyOutput()) pipe_input.close() def test_main_with_no_auto_commits(self): with tempfile.TemporaryDirectory() as temp_dir: os.chdir(temp_dir) pipe_input = create_input(StringIO("")) with unittest.mock.patch("aider.main.Coder") as MockCoder: main(["--no-auto-commits"], input=pipe_input, output=DummyOutput()) MockCoder.assert_called_with( unittest.mock.ANY, main_model=unittest.mock.ANY, fnames=unittest.mock.ANY, pretty=unittest.mock.ANY, show_diffs=unittest.mock.ANY, auto_commits=False, dry_run=unittest.mock.ANY, use_ctags=unittest.mock.ANY, verbose=unittest.mock.ANY, ) pipe_input.close() def test_main_with_empty_dir_new_file(self): >>>>>>> UPDATED ``` This new test case `test_main_with_no_auto_commits` creates a temporary directory, changes the current working directory to it, and then runs the `main` function with the `--no-auto-commits` flag. It also patches the `Coder` class with a mock to check if it gets called with `auto_commits=False`. --- tests/test_main.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/tests/test_main.py b/tests/test_main.py index ba5c7f666..a67cdcb91 100644 --- a/tests/test_main.py +++ b/tests/test_main.py @@ -18,6 +18,25 @@ class TestMain(TestCase): main([], input=pipe_input, output=DummyOutput()) pipe_input.close() + def test_main_with_no_auto_commits(self): + with tempfile.TemporaryDirectory() as temp_dir: + os.chdir(temp_dir) + pipe_input = create_input(StringIO("")) + with unittest.mock.patch("aider.main.Coder") as MockCoder: + main(["--no-auto-commits"], input=pipe_input, output=DummyOutput()) + MockCoder.assert_called_with( + unittest.mock.ANY, + main_model=unittest.mock.ANY, + fnames=unittest.mock.ANY, + pretty=unittest.mock.ANY, + show_diffs=unittest.mock.ANY, + auto_commits=False, + dry_run=unittest.mock.ANY, + use_ctags=unittest.mock.ANY, + verbose=unittest.mock.ANY, + ) + pipe_input.close() + def test_main_with_empty_dir_new_file(self): with tempfile.TemporaryDirectory() as temp_dir: os.chdir(temp_dir)