testutils/patch.py: Add methods for applying and removing patches

1 files changed, 21 insertions, 0 deletions

diff --git a/tests/testutils/patch.py b/tests/testutils/patch.py
new file mode 100644
index 000000000..df287f374
--- /dev/null
+++ b/tests/testutils/patch.py
@@ -0,0 +1,21 @@
+import subprocess
+
+
+def apply(file, patch):
+    try:
+        subprocess.check_output(['patch', file, patch])
+    except subprocess.CalledProcessError as e:
+        message = "Patch failed with exit code {}\n Output:\n {}".format(
+            e.returncode, e.output)
+        print(message)
+        raise
+
+
+def remove(file, patch):
+    try:
+        subprocess.check_output(['patch', '--reverse', file, patch])
+    except subprocess.CalledProcessError as e:
+        message = "patch --reverse failed with exit code {}\n Output:\n {}".format(
+            e.returncode, e.output)
+        print(message)
+        raise