put/get archive implementation

Signed-off-by: Joffrey F <joffrey@docker.com>

1 files changed, 21 insertions, 0 deletions

diff --git a/tests/helpers.py b/tests/helpers.py
index 95692db..392be3b 100644
--- a/tests/helpers.py
+++ b/tests/helpers.py
@@ -1,5 +1,6 @@
 import os
 import os.path
+import tarfile
 import tempfile
 
 
@@ -14,3 +15,23 @@ def make_tree(dirs, files):
             f.write("content")
 
     return base
+
+
+def simple_tar(path):
+    f = tempfile.NamedTemporaryFile()
+    t = tarfile.open(mode='w', fileobj=f)
+
+    abs_path = os.path.abspath(path)
+    t.add(abs_path, arcname=os.path.basename(path), recursive=False)
+
+    t.close()
+    f.seek(0)
+    return f
+
+
+def untar_file(tardata, filename):
+    with tarfile.open(mode='r', fileobj=tardata) as t:
+        f = t.extractfile(filename)
+        result = f.read()
+        f.close()
+    return result