61 lines
2.0 KiB
Python
61 lines
2.0 KiB
Python
import pytest
|
|
import testinfra
|
|
|
|
# Use testinfra to get a handy function to run commands locally
|
|
check_output = testinfra.get_backend(
|
|
"local://"
|
|
).get_module("Command").check_output
|
|
|
|
|
|
@pytest.fixture
|
|
def TestinfraBackend(request):
|
|
docker_run = "docker run -d {}".format(request.param)
|
|
print docker_run
|
|
|
|
docker_id = check_output(docker_run)
|
|
check_output("docker exec %s sed -i 's/^gravity_spinup/#donotcurl/g' /usr/local/bin/gravity.sh", docker_id)
|
|
|
|
def teardown():
|
|
check_output("docker rm -f %s", docker_id)
|
|
|
|
# Destroy the container at the end of the fixture life
|
|
request.addfinalizer(teardown)
|
|
|
|
# Return a dynamic created backend
|
|
return testinfra.get_backend("docker://" + docker_id)
|
|
|
|
|
|
def pytest_generate_tests(metafunc):
|
|
if "TestinfraBackend" in metafunc.fixturenames:
|
|
|
|
mark_args = getattr(metafunc.function, "docker_args", None)
|
|
docker_args = []
|
|
if mark_args is not None:
|
|
docker_args = docker_args + list(mark_args.args)
|
|
|
|
mark_images = getattr(metafunc.function, "docker_images", None)
|
|
images = ['diginc/pi-hole:alpine', 'diginc/pi-hole:debian']
|
|
if mark_images is not None:
|
|
images = mark_images.args
|
|
|
|
mark_cmd = getattr(metafunc.function, "docker_cmd", None)
|
|
command = 'tail -f /dev/null'
|
|
if mark_cmd is not None:
|
|
command = " ".join(mark_cmd.args)
|
|
|
|
docker_run_args = []
|
|
for img in images:
|
|
docker_run_args.append('{} {} {}'.format(" ".join(docker_args),
|
|
img, command))
|
|
|
|
# If the test has a destructive marker, we scope TestinfraBackend
|
|
# at function level (i.e. executing for each test). If not we scope
|
|
# at session level (i.e. all tests will share the same container)
|
|
if getattr(metafunc.function, "persistent", None) is not None:
|
|
scope = "session"
|
|
else:
|
|
scope = "function"
|
|
|
|
metafunc.parametrize(
|
|
"TestinfraBackend", docker_run_args, indirect=True, scope=scope)
|