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

refactor: Consolidate subprocess.call mock in test_pipe_editor

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-11-23 14:23:26 -08:00
parent 14522dbbcd
commit e63df83091
1 changed files with 14 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

28
tests/basic/test_editor.py
View file

@ -99,6 +99,7 @@ def test_pipe_editor():
patch("aider.editor.write_temp_file") as mock_write, patch("aider.editor.write_temp_file") as mock_write,
patch("builtins.open") as mock_open, patch("builtins.open") as mock_open,
patch("os.remove") as mock_remove, patch("os.remove") as mock_remove,
patch("subprocess.call") as mock_subprocess,
): ):
# Setup mocks # Setup mocks
mock_write.return_value = "temp.txt" mock_write.return_value = "temp.txt"
@ -106,22 +107,21 @@ def test_pipe_editor():
mock_file.__enter__.return_value.read.return_value = modified_content mock_file.__enter__.return_value.read.return_value = modified_content
mock_open.return_value = mock_file mock_open.return_value = mock_file
with patch("subprocess.call") as mock_subprocess: # Test with default editor
# Test with default editor result = pipe_editor(test_content)
result = pipe_editor(test_content) assert result == modified_content
assert result == modified_content mock_write.assert_called_with(test_content, None)
mock_write.assert_called_with(test_content, None) mock_subprocess.assert_called()
mock_subprocess.assert_called()
# Test with custom editor # Test with custom editor
result = pipe_editor(test_content, editor="code") result = pipe_editor(test_content, editor="code")
assert result == modified_content assert result == modified_content
mock_subprocess.assert_called() mock_subprocess.assert_called()
# Test with suffix # Test with suffix
result = pipe_editor(test_content, suffix="md") result = pipe_editor(test_content, suffix="md")
assert result == modified_content assert result == modified_content
mock_write.assert_called_with(test_content, "md") mock_write.assert_called_with(test_content, "md")
# Test cleanup on permission error # Test cleanup on permission error
mock_remove.side_effect = PermissionError mock_remove.side_effect = PermissionError