Initial import of NLProg

tests/test_patching.py

@@ -0,0 +1,46 @@
+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()