tests.testutils: add ls_zip() helper
authorMarkus Lehtonen <markus.lehtonen@linux.intel.com>
Thu, 29 Aug 2013 07:37:56 +0000 (10:37 +0300)
committerMarkus Lehtonen <markus.lehtonen@linux.intel.com>
Thu, 5 Jun 2014 11:20:04 +0000 (14:20 +0300)
For listing zip contents.

Signed-off-by: Markus Lehtonen <markus.lehtonen@linux.intel.com>
tests/testutils.py

index 96a164d..ce0c20c 100644 (file)
@@ -8,6 +8,7 @@ import subprocess
 import tarfile
 import tempfile
 import unittest
+import zipfile
 
 import gbp.log
 import gbp.deb.git
@@ -135,3 +136,13 @@ def ls_tar(tarball):
     finally:
         shutil.rmtree(tmpdir)
 
+def ls_zip(archive):
+    """List the contents of zip file"""
+    tmpdir = tempfile.mkdtemp()
+    try:
+        zipobj = zipfile.ZipFile(archive, 'r')
+        zipobj.extractall(tmpdir)
+        return ls_dir(tmpdir)
+    finally:
+        shutil.rmtree(tmpdir)
+