mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-25 14:55:00 +00:00
b4f78547bc
# Context:
USER: add setup and teardown which getcwd and chdir back
ASSISTANT: I will add `setUp` and `tearDown` methods to the `TestWholeFileCoder` class in the `tests/test_wholefile.py` file. The `setUp` method will store the current working directory, and the `tearDown` method will change the working directory back to the original one.
```python
tests/test_wholefile.py
<<<<<<< ORIGINAL
class TestWholeFileCoder(unittest.TestCase):
def test_update_files(self):
=======
class TestWholeFileCoder(unittest.TestCase):
def setUp(self):
self.original_cwd = os.getcwd()
def tearDown(self):
os.chdir(self.original_cwd)
def test_update_files(self):
>>>>>>> UPDATED
```
Let me know if you need any further changes.
42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
import os
|
|
from pathlib import Path
|
|
import tempfile
|
|
import unittest
|
|
|
|
from aider.coders.wholefile_coder import WholeFileCoder
|
|
from aider.io import InputOutput
|
|
|
|
class TestWholeFileCoder(unittest.TestCase):
|
|
def setUp(self):
|
|
self.original_cwd = os.getcwd()
|
|
|
|
def tearDown(self):
|
|
os.chdir(self.original_cwd)
|
|
|
|
def test_update_files(self):
|
|
with tempfile.TemporaryDirectory() as temp_dir:
|
|
# Create a sample file in the temporary directory
|
|
sample_file = os.path.join(temp_dir, "sample.txt")
|
|
with open(sample_file, "w") as f:
|
|
f.write("Original content\n")
|
|
|
|
# Initialize WholeFileCoder with the temporary directory
|
|
io = InputOutput(yes=yes)
|
|
coder = WholeFileCoder(root=temp_dir, io=io)
|
|
|
|
# Set the partial response content with the updated content
|
|
coder.partial_response_content = f"{sample_file}\n```\nUpdated content\n```"
|
|
|
|
# Call update_files method
|
|
edited_files = coder.update_files()
|
|
|
|
# Check if the sample file was updated
|
|
self.assertIn("sample.txt", edited_files)
|
|
|
|
# Check if the content of the sample file was updated
|
|
with open(sample_file, "r") as f:
|
|
updated_content = f.read()
|
|
self.assertEqual(updated_content, "Updated content\n")
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|