61 lines
2.1 KiB
Python
61 lines
2.1 KiB
Python
"""Reusable functions and classes for different types of integration tests.
|
|
|
|
For example ``Archive`` can be used to check the contents of distribution built
|
|
with setuptools, and ``run`` will always try to be as verbose as possible to
|
|
facilitate debugging.
|
|
"""
|
|
import os
|
|
import subprocess
|
|
import tarfile
|
|
from zipfile import ZipFile
|
|
|
|
|
|
def run(cmd, env=None):
|
|
r = subprocess.run(
|
|
cmd,
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE,
|
|
universal_newlines=True,
|
|
env={**os.environ, **(env or {})}
|
|
# ^-- allow overwriting instead of discarding the current env
|
|
)
|
|
|
|
out = r.stdout + "\n" + r.stderr
|
|
# pytest omits stdout/err by default, if the test fails they help debugging
|
|
print("~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~")
|
|
print(f"Command: {cmd}\nreturn code: {r.returncode}\n\n{out}")
|
|
|
|
if r.returncode == 0:
|
|
return out
|
|
raise subprocess.CalledProcessError(r.returncode, cmd, r.stdout, r.stderr)
|
|
|
|
|
|
class Archive:
|
|
"""Compatibility layer for ZipFile/Info and TarFile/Info"""
|
|
def __init__(self, filename):
|
|
self._filename = filename
|
|
if filename.endswith("tar.gz"):
|
|
self._obj = tarfile.open(filename, "r:gz")
|
|
elif filename.endswith("zip"):
|
|
self._obj = ZipFile(filename)
|
|
else:
|
|
raise ValueError(f"{filename} doesn't seem to be a zip or tar.gz")
|
|
|
|
def __iter__(self):
|
|
if hasattr(self._obj, "infolist"):
|
|
return iter(self._obj.infolist())
|
|
return iter(self._obj)
|
|
|
|
def get_name(self, zip_or_tar_info):
|
|
if hasattr(zip_or_tar_info, "filename"):
|
|
return zip_or_tar_info.filename
|
|
return zip_or_tar_info.name
|
|
|
|
def get_content(self, zip_or_tar_info):
|
|
if hasattr(self._obj, "extractfile"):
|
|
content = self._obj.extractfile(zip_or_tar_info)
|
|
if content is None:
|
|
msg = f"Invalid {zip_or_tar_info.name} in {self._filename}"
|
|
raise ValueError(msg)
|
|
return str(content.read(), "utf-8")
|
|
return str(self._obj.read(zip_or_tar_info), "utf-8")
|