dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-24 22:34:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 9

feat: use GitTemporaryDirectory in test_cmd_add_read_only_file

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-08-12 09:01:32 -07:00
parent 42d16ebb99
commit 1e8be8a429
1 changed files with 31 additions and 33 deletions
Download patch file Download diff file Expand all files Collapse all files

64
tests/basic/test_commands.py
View file

@ -538,48 +538,46 @@ class TestCommands(TestCase):
self.assertEqual(coder.abs_fnames, set())
def test_cmd_add_read_only_file(self):
# Initialize the Commands and InputOutput objects
io = InputOutput(pretty=False, yes=True)
from aider.coders import Coder
with GitTemporaryDirectory():
# Initialize the Commands and InputOutput objects
io = InputOutput(pretty=False, yes=True)
from aider.coders import Coder
coder = Coder.create(self.GPT35, None, io)
commands = Commands(io, coder)
coder = Coder.create(self.GPT35, None, io)
commands = Commands(io, coder)
# Create a test file
test_file = Path("test_read_only.txt")
test_file.write_text("Test content")
# Create a test file
test_file = Path("test_read_only.txt")
test_file.write_text("Test content")
# Add the file as read-only
commands.cmd_read(str(test_file))
# Add the file as read-only
commands.cmd_read(str(test_file))
# Verify it's in abs_read_only_fnames
self.assertTrue(
any(
os.path.samefile(str(test_file.resolve()), fname)
for fname in coder.abs_read_only_fnames
# Verify it's in abs_read_only_fnames
self.assertTrue(
any(
os.path.samefile(str(test_file.resolve()), fname)
for fname in coder.abs_read_only_fnames
)
)
)
# Mock the repo to simulate a tracked file
coder.repo = mock.MagicMock()
coder.repo.is_tracked_file.return_value = True
# Mock the repo to simulate a tracked file
coder.repo = mock.MagicMock()
coder.repo.is_tracked_file.return_value = True
# Try to add the read-only file
commands.cmd_add(str(test_file))
# Try to add the read-only file
commands.cmd_add(str(test_file))
# Verify it's now in abs_fnames and not in abs_read_only_fnames
self.assertTrue(
any(os.path.samefile(str(test_file.resolve()), fname) for fname in coder.abs_fnames)
)
self.assertFalse(
any(
os.path.samefile(str(test_file.resolve()), fname)
for fname in coder.abs_read_only_fnames
# Verify it's now in abs_fnames and not in abs_read_only_fnames
self.assertTrue(
any(os.path.samefile(str(test_file.resolve()), fname) for fname in coder.abs_fnames)
)
self.assertFalse(
any(
os.path.samefile(str(test_file.resolve()), fname)
for fname in coder.abs_read_only_fnames
)
)
)
# Clean up
test_file.unlink()
def test_cmd_test_unbound_local_error(self):
with ChdirTemporaryDirectory():