145 lines
4.4 KiB
Python
145 lines
4.4 KiB
Python
import pytest
|
|
import testinfra
|
|
|
|
WEB_SERVER = { 'alpine_amd64': 'nginx', 'debian_amd64': 'lighttpd' }
|
|
|
|
check_output = testinfra.get_backend(
|
|
"local://"
|
|
).get_module("Command").check_output
|
|
|
|
def DockerGeneric(request, args, image, cmd):
|
|
assert 'docker' in check_output('id'), "Are you in the docker group?"
|
|
if 'pi-hole' in image:
|
|
args += " --dns 127.0.0.1 -v /dev/null:/etc/.pihole/adlists.default -e PYTEST=\"True\""
|
|
docker_run = "docker run -d {} {} {}".format(args, image, cmd)
|
|
print docker_run
|
|
docker_id = check_output(docker_run)
|
|
|
|
def teardown():
|
|
check_output("docker logs {}".format(docker_id))
|
|
check_output("docker rm -f {}".format(docker_id))
|
|
request.addfinalizer(teardown)
|
|
|
|
docker_container = testinfra.get_backend("docker://" + docker_id)
|
|
docker_container.id = docker_id
|
|
|
|
def run_bash(self, command, *args, **kwargs):
|
|
cmd = self.get_command(command, *args)
|
|
if self.user is not None:
|
|
out = self.run_local(
|
|
"docker exec -u %s %s /bin/bash -c %s",
|
|
self.user, self.name, cmd)
|
|
else:
|
|
out = self.run_local(
|
|
"docker exec %s /bin/bash -c %s", self.name, cmd)
|
|
out.command = self.encode(cmd)
|
|
return out
|
|
|
|
funcType = type(docker_container.run)
|
|
docker_container.run = funcType(run_bash, docker_container, testinfra.backend.docker.DockerBackend)
|
|
return docker_container
|
|
|
|
@pytest.fixture
|
|
def Docker(request, args, image, cmd):
|
|
''' One-off Docker container run '''
|
|
return DockerGeneric(request, args, image, cmd)
|
|
|
|
@pytest.fixture(scope='module')
|
|
def DockerPersist(request, persist_args, persist_image, persist_cmd, Dig):
|
|
''' Persistent Docker container for multiple tests '''
|
|
persistent_container = DockerGeneric(request, persist_args, persist_image, persist_cmd)
|
|
''' attach a dig conatiner for lookups '''
|
|
persistent_container.dig = Dig(persistent_container.id)
|
|
return persistent_container
|
|
|
|
@pytest.fixture()
|
|
def args(request):
|
|
return '-e ServerIP="127.0.0.1" -e ServerIPv6="::1"'
|
|
|
|
@pytest.fixture(params=['amd64', 'armhf', 'aarch64'])
|
|
def arch(request):
|
|
return request.param
|
|
|
|
@pytest.fixture(params=['debian', 'alpine'])
|
|
def os(request):
|
|
return request.param
|
|
|
|
@pytest.fixture()
|
|
def tag(request, os, arch):
|
|
return '{}_{}'.format(os, arch)
|
|
|
|
@pytest.fixture
|
|
def webserver(request, tag):
|
|
return WEB_SERVER[tag]
|
|
|
|
@pytest.fixture()
|
|
def image(request, tag):
|
|
return 'pi-hole:{}'.format(tag)
|
|
|
|
@pytest.fixture()
|
|
def cmd(request):
|
|
return 'tail -f /dev/null'
|
|
|
|
@pytest.fixture(scope='module')
|
|
def persist_args(request):
|
|
return '-e ServerIP="127.0.0.1" -e ServerIPv6="::1"'
|
|
|
|
@pytest.fixture(scope='module', params=['alpine_amd64', 'debian_amd64'])
|
|
def persist_tag(request):
|
|
return request.param
|
|
|
|
@pytest.fixture(scope='module')
|
|
def persist_webserver(request, persist_tag):
|
|
return WEB_SERVER[persist_tag]
|
|
|
|
@pytest.fixture(scope='module')
|
|
def persist_image(request, persist_tag):
|
|
return 'pi-hole:{}'.format(persist_tag)
|
|
|
|
@pytest.fixture(scope='module')
|
|
def persist_cmd(request):
|
|
return 'tail -f /dev/null'
|
|
|
|
@pytest.fixture
|
|
def Slow():
|
|
"""
|
|
Run a slow check, check if the state is correct for `timeout` seconds.
|
|
"""
|
|
import time
|
|
def slow(check, timeout=20):
|
|
timeout_at = time.time() + timeout
|
|
while True:
|
|
try:
|
|
assert check()
|
|
except AssertionError, e:
|
|
if time.time() < timeout_at:
|
|
time.sleep(1)
|
|
else:
|
|
raise e
|
|
else:
|
|
return
|
|
return slow
|
|
|
|
@pytest.fixture(scope='module')
|
|
def Dig(request):
|
|
''' separate container to link to pi-hole and perform lookups '''
|
|
''' a docker pull is faster than running an install of dnsutils '''
|
|
def dig(docker_id):
|
|
args = '--link {}:test_pihole'.format(docker_id)
|
|
image = 'azukiapp/dig'
|
|
cmd = 'tail -f /dev/null'
|
|
dig_container = DockerGeneric(request, args, image, cmd)
|
|
return dig_container
|
|
return dig
|
|
|
|
'''
|
|
Persistent Docker container for testing service post start.sh
|
|
'''
|
|
@pytest.fixture
|
|
def RunningPiHole(DockerPersist, Slow, persist_webserver):
|
|
''' Persist a fully started docker-pi-hole to help speed up subsequent tests '''
|
|
Slow(lambda: DockerPersist.run('pgrep dnsmasq').rc == 0)
|
|
Slow(lambda: DockerPersist.run('pgrep {}'.format(persist_webserver) ).rc == 0)
|
|
return DockerPersist
|
|
|