tests/testutils: add directories arg to ls_* functions

With this you can exclude directories from the listing returned by the
ls_tar(), ls_zip() and ls_dir() functions.

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

1 files changed, 8 insertions, 7 deletions

diff --git a/tests/testutils.py b/tests/testutils.py
index ce0c20c8..6331e12d 100644
--- a/tests/testutils.py
+++ b/tests/testutils.py
@@ -115,34 +115,35 @@ def get_dch_default_urgency():
             shutil.rmtree(tempdir)
     return urgency
 
-def ls_dir(directory):
+def ls_dir(directory, directories=True):
     """List the contents of directory, recurse to subdirectories"""
     contents = set()
     for root, dirs, files in os.walk(directory):
         prefix = ''
         if root != directory:
             prefix = os.path.relpath(root, directory) + '/'
-        contents.update(['%s%s' % (prefix, fname) for fname in files] +
-                        ['%s%s' % (prefix, dname) for dname in dirs])
+        contents.update(['%s%s' % (prefix, fname) for fname in files])
+        if directories:
+            contents.update(['%s%s' % (prefix, dname) for dname in dirs])
     return contents
 
-def ls_tar(tarball):
+def ls_tar(tarball, directories=True):
     """List the contents of tar archive"""
     tmpdir = tempfile.mkdtemp()
     try:
         tarobj = tarfile.open(tarball, 'r')
         tarobj.extractall(tmpdir)
-        return ls_dir(tmpdir)
+        return ls_dir(tmpdir, directories)
     finally:
         shutil.rmtree(tmpdir)
 
-def ls_zip(archive):
+def ls_zip(archive, directories=True):
     """List the contents of zip file"""
     tmpdir = tempfile.mkdtemp()
     try:
         zipobj = zipfile.ZipFile(archive, 'r')
         zipobj.extractall(tmpdir)
-        return ls_dir(tmpdir)
+        return ls_dir(tmpdir, directories)
     finally:
         shutil.rmtree(tmpdir)