mirror of
https://github.com/bunkerity/bunkerweb
synced 2026-05-24 09:28:37 +00:00
77 lines
No EOL
3.2 KiB
Python
77 lines
No EOL
3.2 KiB
Python
from Test import Test
|
|
from os.path import isdir, join, isfile
|
|
from os import chown, walk, getenv, listdir
|
|
from shutil import copytree
|
|
from traceback import format_exc
|
|
from subprocess import run
|
|
|
|
class DockerTest(Test) :
|
|
|
|
def __init__(self, name, timeout, tests) :
|
|
super().__init__(name, "docker", timeout, tests)
|
|
self.__domains = {
|
|
r"www\.example\.com": getenv("TEST_DOMAIN1"),
|
|
r"auth\.example\.com": getenv("TEST_DOMAIN1"),
|
|
r"app1\.example\.com": getenv("TEST_DOMAIN1_1"),
|
|
r"app2\.example\.com": getenv("TEST_DOMAIN1_2"),
|
|
r"app3\.example\.com": getenv("TEST_DOMAIN1_3")
|
|
}
|
|
|
|
def init() :
|
|
try :
|
|
if not Test.init() :
|
|
return False
|
|
for root, dirs, files in walk("/tmp/bw-data") :
|
|
for name in dirs + files :
|
|
chown(join(root, name), 101, 101)
|
|
except :
|
|
self._log("exception while running DockerTest.init()\n" + format_exc(), error=True)
|
|
return False
|
|
return True
|
|
|
|
def _setup_test(self) :
|
|
try :
|
|
super()._setup_test()
|
|
test = "/tmp/tests/" + self._name
|
|
compose = "/tmp/tests/" + self._name + "/docker-compose.yml"
|
|
example_data = "./examples/" + self._name + "/bw-data"
|
|
self._replace_in_file(compose, r"bunkerity/bunkerweb:.*$", "10.20.1.1:5000/bw-tests:latest")
|
|
self._replace_in_file(compose, r"\./bw\-data:/", "/tmp/bw-data:/")
|
|
self._replace_in_file(compose, r"\- bw_data:/", "/tmp/bw-data:/")
|
|
for ex_domain, test_domain in self.__domains.items() :
|
|
self._replace_in_files(test, ex_domain, test_domain)
|
|
self._rename(test, ex_domain, test_domain)
|
|
setup = test + "/setup-docker.sh"
|
|
if isfile(setup) :
|
|
run("./docker-setup.sh", cwd=test, shell=True, check=True)
|
|
if isdir(example_data) :
|
|
for cp_dir in listdir(example_data) :
|
|
if isdir(join(example_data, cp_dir)) :
|
|
copytree(join(example_data, cp_dir), join("/tmp/bw-data", cp_dir))
|
|
cmd = "docker-compose pull"
|
|
proc = run(cmd.split(" "), shell=True, cwd=test)
|
|
if proc.returncode != 0 :
|
|
raise("docker-compose pull failed")
|
|
cmd = "docker-compose up -d"
|
|
proc = run(cmd.split(" "), shell=True, cwd=test)
|
|
if proc.returncode != 0 :
|
|
raise("docker-compose up failed")
|
|
except :
|
|
self._log("exception while running DockerTest._setup_test()\n" + format_exc(), error=True)
|
|
return False
|
|
return True
|
|
|
|
|
|
def _cleanup_test(self) :
|
|
try :
|
|
test = "/tmp/tests/" + self._name
|
|
cmd = "docker-compose down -v"
|
|
proc = run(cmd.split(" "), shell=True, cwd=test)
|
|
if proc.returncode != 0 :
|
|
raise("docker-compose down failed")
|
|
super()._cleanup_test()
|
|
except :
|
|
self._log("exception while running DockerTest._setup_test()\n" + format_exc(), error=True)
|
|
return False
|
|
return True
|
|
|