mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-21 21:04:59 +00:00
65c708b0c4
# Context:
USER: add a test for cmd_add that passes in 'foo.txt' and 'bar.txt' and ensures they have both been created after the test. run the test in a tmp dir.
ASSISTANT: To add the test for `cmd_add`, we need to modify the `tests/test_commands.py` file. Please provide the content of the `tests/test_commands.py` file so I can suggest the necessary changes.
USER: it's empty
ASSISTANT: In that case, I will create a new test function for `cmd_add` in the `tests/test_commands.py` file. Here's the suggested change:
```python
tests/test_commands.py
<<<<<<< ORIGINAL
=======
import os
import tempfile
import unittest
from unittest.mock import MagicMock
from aider.commands import Commands
from aider.io import IO
from aider.coder import Coder
class TestCommands(unittest.TestCase):
def test_cmd_add(self):
with tempfile.TemporaryDirectory() as tmpdir:
io = IO()
coder = Coder(root=tmpdir)
commands = Commands(io, coder)
# Mock the Confirm.ask method to return True for creating files
with unittest.mock.patch("rich.prompt.Confirm.ask", return_value=True):
commands.cmd_add("foo.txt bar.txt")
foo_path = os.path.join(tmpdir, "foo.txt")
bar_path = os.path.join(tmpdir, "bar.txt")
self.assertTrue(os.path.exists(foo_path), "foo.txt should be created")
self.assertTrue(os.path.exists(bar_path), "bar.txt should be created")
if __name__ == "__main__":
unittest.main()
>>>>>>> UPDATED
```
This will create a new test function `test_cmd_add` in the `tests/test_commands.py` file that tests the `cmd_add` method with 'foo.txt' and 'bar.txt' as input. The test will run in a temporary directory and ensure that both files have been created after the test.
27 lines
933 B
Python
27 lines
933 B
Python
import os
|
|
import tempfile
|
|
import unittest
|
|
from unittest.mock import MagicMock
|
|
from aider.commands import Commands
|
|
from aider.io import IO
|
|
from aider.coder import Coder
|
|
|
|
class TestCommands(unittest.TestCase):
|
|
def test_cmd_add(self):
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
io = IO()
|
|
coder = Coder(root=tmpdir)
|
|
commands = Commands(io, coder)
|
|
|
|
# Mock the Confirm.ask method to return True for creating files
|
|
with unittest.mock.patch("rich.prompt.Confirm.ask", return_value=True):
|
|
commands.cmd_add("foo.txt bar.txt")
|
|
|
|
foo_path = os.path.join(tmpdir, "foo.txt")
|
|
bar_path = os.path.join(tmpdir, "bar.txt")
|
|
|
|
self.assertTrue(os.path.exists(foo_path), "foo.txt should be created")
|
|
self.assertTrue(os.path.exists(bar_path), "bar.txt should be created")
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|