cloud-build/cloud_build/image_tests/__init__.py
2019-11-01 15:30:53 +03:00

43 lines
959 B
Python

import contextlib
import os
import shutil
import subprocess
import tempfile
from .lxd import test_lxd
from .docker import test_docker
@contextlib.contextmanager
def pushtmpd():
previous_dir = os.getcwd()
tmpdir = tempfile.mkdtemp()
try:
os.chdir(tmpdir)
yield tmpdir
finally:
os.chdir(previous_dir)
shutil.rmtree(tmpdir)
def test(method, image, branch, arch):
if arch not in ['x86_64', 'i586']:
return True
with pushtmpd() as tmpdir:
image = shutil.copy(image, tmpdir)
image = os.path.basename(image)
if method == 'lxd':
commands = test_lxd(image)
elif method == 'docker':
commands = test_docker(image)
else:
raise Exception(f'Undefined test method {method}')
for command in commands:
rc = subprocess.call(command, shell=True)
if rc:
return False
return True