Add a new zip sourcezip

This is equivalent to the tar source, but for Zip archives.

1 files changed, 40 insertions, 0 deletions

diff --git a/tests/testutils/repo/zip.py b/tests/testutils/repo/zip.py
new file mode 100644
index 000000000..47c402421
--- /dev/null
+++ b/tests/testutils/repo/zip.py
@@ -0,0 +1,40 @@
+import hashlib
+import os
+import zipfile
+
+from buildstream.utils import sha256sum
+
+from .repo import Repo
+
+
+class Zip(Repo):
+
+    def create(self, directory):
+        archive = os.path.join(self.repo, 'file.zip')
+
+        old_dir = os.getcwd()
+        os.chdir(directory)
+        with zipfile.ZipFile(archive, "w") as zip:
+            for root, dirs, files in os.walk('.'):
+                names = dirs + files
+                names = [os.path.join(root, name) for name in names]
+
+                for name in names:
+                    zip.write(name)
+
+        os.chdir(old_dir)
+
+        return sha256sum(archive)
+
+    def source_config(self, ref=None):
+        archive = os.path.join(self.repo, 'file.zip')
+        config = {
+            'kind': 'zip',
+            'url': 'file://' + archive,
+            'directory': '',
+            'base-dir': ''
+        }
+        if ref is not None:
+            config['ref'] = ref
+
+        return config