mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-21 12:55:00 +00:00
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`.
This commit is contained in:
parent
a9cc332e2d
commit
01963c28f8
1 changed files with 19 additions and 0 deletions
|
@ -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)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue