Public Access
107 lines
3.3 KiB
Python
107 lines
3.3 KiB
Python
#!/usr/bin/env python3
|
|
"""Run all molecule scenarios on all supported OS platforms.
|
|
|
|
Replaces the previous ``scripts/molecule_all.sh`` with a tested Python equivalent.
|
|
Sequential execution — CI uses the parallel matrix instead.
|
|
|
|
Usage::
|
|
|
|
python3 -m devx.molecule.molecule_all
|
|
python3 -m devx.molecule.molecule_all --bin .venv/bin
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
import subprocess # nosec B404
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
import click
|
|
|
|
from devx.molecule.platforms import PLATFORMS
|
|
|
|
DEFAULT_ROLES_ROOT = Path("ansible/roles")
|
|
|
|
|
|
def _default_role_dir() -> Path:
|
|
"""Auto-discover the single role directory with molecule scenarios."""
|
|
roles_root = DEFAULT_ROLES_ROOT
|
|
if not roles_root.is_dir():
|
|
return roles_root / "gitea_runner" # sensible default for error message
|
|
role_dirs = sorted(d for d in roles_root.iterdir() if (d / "molecule").is_dir())
|
|
if role_dirs:
|
|
return role_dirs[0]
|
|
return roles_root / "role" # will produce a clear error
|
|
|
|
|
|
SCENARIOS = ["default", "multi-instance", "lifecycle", "template-content", "deregister", "update"]
|
|
|
|
|
|
def _run_molecule(molecule_bin: str, scenario: str, role_dir: Path, env: dict[str, str]) -> int:
|
|
"""Run a single molecule scenario. Returns the exit code."""
|
|
cmd = [molecule_bin, "test"]
|
|
if scenario != "default":
|
|
cmd.extend(["-s", scenario])
|
|
|
|
click.echo(f"--- Scenario: {scenario} ---")
|
|
result = subprocess.run( # nosec B603
|
|
cmd,
|
|
cwd=str(role_dir),
|
|
env=env,
|
|
)
|
|
return result.returncode
|
|
|
|
|
|
def _run_platform(
|
|
molecule_bin: str,
|
|
platform: dict[str, str],
|
|
role_dir: Path,
|
|
scenarios: list[str],
|
|
base_env: dict[str, str],
|
|
) -> int:
|
|
"""Run all scenarios for a single platform. Returns the first non-zero exit code."""
|
|
env = dict(base_env)
|
|
env["MOLECULE_PLATFORM_NAME"] = platform["name"]
|
|
env["MOLECULE_PLATFORM_IMAGE"] = platform["image"]
|
|
if platform.get("command"):
|
|
env["MOLECULE_PLATFORM_COMMAND"] = platform["command"]
|
|
else:
|
|
env.pop("MOLECULE_PLATFORM_COMMAND", None)
|
|
|
|
click.echo(f"=== Platform: {platform['name']} ===")
|
|
for scenario in scenarios:
|
|
rc = _run_molecule(molecule_bin, scenario, role_dir, env)
|
|
if rc != 0:
|
|
return rc
|
|
return 0
|
|
|
|
|
|
@click.command()
|
|
@click.option("--bin", "bin_dir", default=".venv/bin", help="Path to the virtualenv bin directory.")
|
|
def main(bin_dir: str) -> None:
|
|
"""Run all molecule scenarios on all supported OS platforms sequentially."""
|
|
molecule_bin = str(Path(bin_dir) / "molecule")
|
|
if not Path(molecule_bin).exists():
|
|
raise click.ClickException(f"molecule not found at {molecule_bin}. Run 'make setup' first.")
|
|
|
|
role_dir = _default_role_dir()
|
|
if not role_dir.exists():
|
|
raise click.ClickException(f"Role directory not found: {role_dir}")
|
|
|
|
base_env = dict(os.environ)
|
|
base_env["ANSIBLE_ALLOW_BROKEN_CONDITIONALS"] = "true"
|
|
base_env["ANSIBLE_INJECT_INVOCATION"] = "1"
|
|
|
|
for platform in PLATFORMS:
|
|
rc = _run_platform(molecule_bin, platform, role_dir, SCENARIOS, base_env)
|
|
if rc != 0:
|
|
click.echo(f"FAILED on platform {platform['name']}", err=True)
|
|
sys.exit(rc)
|
|
|
|
click.echo("All molecule scenarios passed on all platforms.")
|
|
|
|
|
|
if __name__ == "__main__": # pragma: no cover
|
|
main() # pragma: no cover
|