Add basic pledge/unveil sandbox on OpenBSD

[tinc] / test / integration / testlib / util.py

diff --git a/test/integration/testlib/util.py b/test/integration/testlib/util.py
index 8102bca..344958e 100755 (executable)
--- a/test/integration/testlib/util.py
+++ b/test/integration/testlib/util.py
@@ -7,6 +7,8 @@ import random
 import string
 import socket
 import typing as T
+import tempfile
+from pathlib import Path
 
 from . import check
 from .log import log
@@ -31,6 +33,23 @@ def random_port() -> int:
             log.debug("could not bind to random port %d", port, exc_info=ex)
 
 
+def temp_file(content: str) -> str:
+    """Create a temporary file and write text content into it."""
+    file = tempfile.mktemp()
+    with open(file, "w", encoding="utf-8") as f:
+        f.write(content)
+    return file
+
+
+def remove_file(path: T.Union[str, Path]) -> bool:
+    """Try to remove file without failing if it does not exist."""
+    try:
+        os.remove(path)
+        return True
+    except FileNotFoundError:
+        return False
+
+
 def random_string(k: int) -> str:
     """Generate a random alphanumeric string of length k."""
     return "".join(random.choices(_ALPHA_NUMERIC, k=k))