1 files changed, 48 insertions, 0 deletions
diff --git a/minishell_test/sandbox.py b/minishell_test/sandbox.py
new file mode 100644
index 0000000..bd49d1e
--- /dev/null
+++ b/minishell_test/sandbox.py
@@ -0,0 +1,48 @@
+# ############################################################################ #
+#                                                                              #
+#                                                         :::      ::::::::    #
+#    sandbox.py                                         :+:      :+:    :+:    #
+#                                                     +:+ +:+         +:+      #
+#    By: charles <me@cacharle.xyz>                  +#+  +:+       +#+         #
+#                                                 +#+#+#+#+#+   +#+            #
+#    Created: 2020/09/11 13:48:07 by charles           #+#    #+#              #
+#    Updated: 2021/01/31 03:59:30 by charles          ###   ########.fr        #
+#                                                                              #
+# ############################################################################ #
+
+import os
+import glob
+import shutil
+import subprocess
+from contextlib import contextmanager
+
+import config
+
+
+def create():
+    """Create a new sandbox directory"""
+    try:
+        os.mkdir(config.SANDBOX_PATH)
+    except OSError:
+        pass
+
+
+def remove():
+    """Remove the sandbox directory
+       Brute force rm -rf if clean removal doesn't work due to permissions.
+    """
+    try:
+        shutil.rmtree(config.SANDBOX_PATH)
+    except PermissionError:
+        subprocess.run(["chmod", "777", *glob.glob(config.SANDBOX_PATH + "/*")], check=True)
+        subprocess.run(["rm", "-rf", config.SANDBOX_PATH], check=True)
+    except FileNotFoundError:
+        pass
+
+
+@contextmanager
+def context():
+    """Sandbox context manager"""
+    create()
+    yield
+    remove()