edb4c397df
Upstream: https://github.com/YunoHost/yunohost @ 3a5f8bac8301c450897b96cbd43a4c7d3ba750fb But (José) : transformer tout le code en bions + ploxions du xerboxion. La carte de digestion vit au labo : /yunohost-digest.json
124 lines
3.4 KiB
Python
124 lines
3.4 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import os
|
|
import re
|
|
import shutil
|
|
import stat
|
|
import subprocess
|
|
import tempfile
|
|
import threading
|
|
from http.server import HTTPServer, SimpleHTTPRequestHandler
|
|
from pathlib import Path
|
|
from types import NoneType
|
|
from typing import Generator
|
|
|
|
import pytest
|
|
|
|
TEST_DIRS = {
|
|
"2": Path(__file__).parent / "test_helpers.v2.d",
|
|
"2.1": Path(__file__).parent / "test_helpers.v2.1.d",
|
|
}
|
|
|
|
|
|
def list_tests() -> list[tuple[str, str]]:
|
|
tests: list[tuple[str, str]] = []
|
|
for helpers_version, test_dir in TEST_DIRS.items():
|
|
for file in test_dir.glob("ynhtest_*.sh"):
|
|
file_testname = file.name.removeprefix("ynhtest_").removesuffix(".sh")
|
|
|
|
result = subprocess.check_output(
|
|
["bash", "-c", f"source {file}; declare -F"]
|
|
)
|
|
for line in result.decode("utf-8").splitlines():
|
|
if match := re.match(r"^declare -f ynhtest_(.*)$", line):
|
|
testfn = match.group(1)
|
|
tests.append((helpers_version, file_testname, testfn))
|
|
|
|
return tests
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def http_server() -> Generator[HTTPServer, None, None]:
|
|
tempdir = tempfile.mkdtemp()
|
|
|
|
class Handler(SimpleHTTPRequestHandler):
|
|
directory = tempdir
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, directory=self.directory, **kwargs)
|
|
|
|
class Server(HTTPServer):
|
|
def run(self):
|
|
try:
|
|
self.serve_forever()
|
|
except KeyboardInterrupt:
|
|
print("Closing HTTP server on keybord request...")
|
|
finally:
|
|
self.server_close()
|
|
|
|
host = "127.0.0.1"
|
|
port = 1312
|
|
server = Server((host, port), Handler)
|
|
thread = threading.Thread(None, server.run)
|
|
thread.start()
|
|
yield server
|
|
server.shutdown()
|
|
thread.join()
|
|
shutil.rmtree(tempdir)
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def var_www_tempdir() -> Generator[Path, None, None]:
|
|
tempdir = Path(tempfile.mkdtemp())
|
|
var_www = tempdir / "var" / "www"
|
|
var_www.mkdir(parents=True)
|
|
tempdir.chmod(stat.S_IROTH | stat.S_IXOTH)
|
|
yield tempdir
|
|
shutil.rmtree(tempdir)
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def ynhtest_app() -> Generator[Path, None, None]:
|
|
app_dir = Path("/etc/yunohost/apps/ynhtest")
|
|
app_dir.mkdir(parents=True)
|
|
settings_file = app_dir / "settings.yml"
|
|
settings_file.write_text("id: ynhtest\n")
|
|
yield app_dir
|
|
shutil.rmtree(app_dir)
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def ensure_user() -> None:
|
|
if subprocess.run(
|
|
["getent", "passwd", "ynhtest"],
|
|
stdout=subprocess.DEVNULL,
|
|
stderr=subprocess.DEVNULL,
|
|
).returncode:
|
|
subprocess.run(["useradd", "--system", "ynhtest"])
|
|
|
|
|
|
@pytest.mark.parametrize("version,file,func", list_tests())
|
|
def test_helpers(
|
|
version: str,
|
|
file: str,
|
|
func: str,
|
|
var_www_tempdir: Path,
|
|
http_server: HTTPServer,
|
|
ensure_user: NoneType,
|
|
) -> None:
|
|
wrapper_file = Path(__file__).parent / "test_helpers_wrapper.sh"
|
|
test_file = TEST_DIRS[version] / f"ynhtest_{file}.sh"
|
|
test_func = f"ynhtest_{func}"
|
|
|
|
test_env = os.environ | {
|
|
"HTTPSERVER_DIR": str(http_server.RequestHandlerClass.directory),
|
|
"HTTPSERVER_PORT": str(http_server.server_port),
|
|
"VAR_WWW": str(var_www_tempdir),
|
|
}
|
|
|
|
subprocess.check_call(
|
|
[wrapper_file, version, test_file, test_func],
|
|
env=test_env,
|
|
stderr=subprocess.STDOUT,
|
|
)
|