47 lines
1.5 KiB
Python
47 lines
1.5 KiB
Python
import tempfile
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
from nlprog.patching import PatchError, apply_patch_to_root, summarize_patch
|
|
|
|
|
|
class PatchingTests(unittest.TestCase):
|
|
def test_add_update_delete_file(self):
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
root = Path(tmp)
|
|
add_patch = """*** Begin Patch
|
|
*** Add File: notes.txt
|
|
+hello
|
|
*** End Patch"""
|
|
self.assertIn("Add notes.txt", summarize_patch(add_patch))
|
|
self.assertEqual(apply_patch_to_root(root, add_patch, set()), ["Added notes.txt"])
|
|
self.assertEqual((root / "notes.txt").read_text(encoding="utf-8"), "hello\n")
|
|
|
|
update_patch = """*** Begin Patch
|
|
*** Update File: notes.txt
|
|
@@
|
|
-hello
|
|
+hello world
|
|
*** End Patch"""
|
|
self.assertEqual(apply_patch_to_root(root, update_patch, set()), ["Updated notes.txt"])
|
|
self.assertEqual((root / "notes.txt").read_text(encoding="utf-8"), "hello world\n")
|
|
|
|
delete_patch = """*** Begin Patch
|
|
*** Delete File: notes.txt
|
|
*** End Patch"""
|
|
self.assertEqual(apply_patch_to_root(root, delete_patch, set()), ["Deleted notes.txt"])
|
|
self.assertFalse((root / "notes.txt").exists())
|
|
|
|
def test_protected_path_is_rejected(self):
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
patch = """*** Begin Patch
|
|
*** Add File: __pycache__/blocked.txt
|
|
+x
|
|
*** End Patch"""
|
|
with self.assertRaises(PatchError):
|
|
apply_patch_to_root(Path(tmp), patch, {"__pycache__"})
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|