Added tests for set-root

1 files changed, 24 insertions, 0 deletions

diff --git a/libbe/tests.py b/libbe/tests.py
new file mode 100644
index 0000000..ecd6ed5
--- /dev/null
+++ b/libbe/tests.py
@@ -0,0 +1,24 @@
+import tempfile
+import shutil
+import os
+import os.path
+cleanable = []
+def clean_up():
+    global cleanable
+    tmp = cleanable
+    tmp.reverse()
+    for obj in tmp:
+        obj.clean_up()
+    cleanable = []
+
+class Dir:
+    def __init__(self):
+        self.name = tempfile.mkdtemp(prefix="testdir")
+        cleanable.append(self)
+    def clean_up(self):
+        shutil.rmtree(self.name)
+
+def arch_dir():
+    dir = Dir()
+    os.mkdir(os.path.join(dir.name, "{arch}"))
+    return dir