test_all.py

#!/usr/bin/env python3
"""Deprecated compatibility shim for the historical test_all.py runner."""

from __future__ import annotations

import argparse
import os
import sys
from pathlib import Path


SOURCE_ROOT = Path(__file__).resolve().parent / "src"
if str(SOURCE_ROOT) not in sys.path:
    sys.path.insert(0, str(SOURCE_ROOT))

from mn_system_tests.catalog import suite_catalog  # noqa: E402
from mn_system_tests.cli import main as mn_test_main  # noqa: E402


def build_legacy_parser() -> argparse.ArgumentParser:
    parser = argparse.ArgumentParser(
        description="Deprecated test_all.py compatibility interface.",
    )
    parser.add_argument("--fast", action="store_true")
    parser.add_argument("--changed", action="store_true")
    parser.add_argument("--unit", action="store_true")
    parser.add_argument("--contracts", action="store_true")
    parser.add_argument("--integration", action="store_true")
    parser.add_argument("--e2e", action="store_true")
    parser.add_argument("--regression", action="store_true")
    parser.add_argument("--blueprints", action="store_true")
    parser.add_argument("--benchmarks", action="store_true")
    parser.add_argument("--performance", action="store_true")
    parser.add_argument("--runtime-e2e", action="store_true")
    parser.add_argument("--redis-ha", action="store_true")
    parser.add_argument("--redis-ha-remote-host")
    parser.add_argument("--redis-ha-local-ip")
    parser.add_argument("--redis-ha-remote-ip")
    parser.add_argument("--shared-storage", "--syncthing", action="store_true")
    parser.add_argument("--shared-storage-remote-host", "--syncthing-remote-host")
    parser.add_argument("--cluster-models", "--model-routing", action="store_true")
    parser.add_argument("--cluster-models-remote-ssh")
    parser.add_argument("--cluster-models-remote-host")
    parser.add_argument("--cluster-model")
    parser.add_argument("--cluster-model-expect-gpu", action="store_true")
    parser.add_argument("--cluster-job-controls", action="store_true")
    parser.add_argument("--llm-gateway-live", action="store_true")
    parser.add_argument("--llm-gateway-remote-node")
    parser.add_argument("--security", action="store_true")
    parser.add_argument("--strict-security", action="store_true")
    parser.add_argument("--skip-node", action="store_true")
    parser.add_argument("--skip-core", action="store_true")
    parser.add_argument("--skip-blueprints", action="store_true")
    parser.add_argument("--live", action="store_true")
    parser.add_argument("--results-dir")
    parser.add_argument("--mn-set-folder", default="../")
    parser.add_argument("--remote-ssh-command")
    parser.add_argument("--remote-mn-set-folder")
    return parser


def translate_legacy_args(argv: list[str]) -> list[str]:
    args = build_legacy_parser().parse_args(argv)
    translated = ["--mn-set-folder", args.mn_set_folder]
    selected: list[str] = []

    if args.fast:
        selected = ["contracts", "benchmarks", "installer"]
        if not args.skip_blueprints:
            selected.append("blueprints.local")
    elif args.changed:
        translated.append("--changed")
    elif args.unit:
        translated.extend(["--level", "unit"])
    elif args.contracts:
        selected = ["contracts"]
    elif args.integration:
        selected = ["mn-system-tests.integration"]
    elif args.e2e:
        selected = ["MirrorNeuron.e2e", "mn-system-tests.e2e"]
    elif args.regression:
        selected = ["MirrorNeuron.regression"]
    elif args.blueprints:
        selected = ["blueprints.local"]
    elif args.benchmarks:
        selected = ["benchmarks"]
    elif args.performance:
        selected = ["performance"]
    elif args.runtime_e2e:
        selected = ["MirrorNeuron.e2e"]
    elif args.redis_ha:
        selected = ["redis-ha.local"]
        if args.redis_ha_remote_host:
            selected.append("redis-ha.two-box")
            translated.extend(["--enable", "two-box"])
    elif args.shared_storage:
        selected = ["cluster.shared-storage"]
        translated.extend(["--enable", "two-box"])
    elif args.cluster_models:
        selected = ["cluster.models"]
        translated.extend(["--enable", "two-box"])
    elif args.cluster_job_controls:
        selected = ["cluster.job-controls"]
        translated.extend(["--enable", "two-box", "--enable", "destructive"])
    elif args.llm_gateway_live:
        selected = ["cluster.llm-live"]
        translated.extend(
            [
                "--enable",
                "two-box",
                "--enable",
                "host-secret",
                "--enable",
                "gpu",
            ]
        )
    elif args.skip_node or args.skip_core or args.skip_blueprints:
        for suite in suite_catalog():
            if args.skip_node and suite.component in {
                "mn-web-ui",
                "otterdesk-desktop-app",
            }:
                continue
            if args.skip_core and suite.component == "MirrorNeuron":
                continue
            if args.skip_blueprints and suite.component in {
                "mn-blueprints",
            }:
                continue
            selected.append(suite.id)

    if args.security and "security.dependencies" not in selected:
        selected.append("security.dependencies")

    # Keep global options ahead of selectors. ``mn_test`` exports its default
    # remote workspace to every child process, including this deprecated shim;
    # emitting that inherited setting after ``--suite`` made the translated
    # legacy invocation depend on the runner's environment and broke callers
    # that consume the canonical selector tail.
    if args.results_dir:
        translated.extend(["--results-dir", args.results_dir])
    remote_ssh = (
        args.remote_ssh_command
        or args.cluster_models_remote_ssh
        or _legacy_remote_ssh(args)
    )
    remote_folder = (
        args.remote_mn_set_folder
        or os.environ.get("MN_CLUSTER_REMOTE_MN_SET_FOLDER")
    )
    if remote_ssh:
        translated.extend(["--remote-ssh-command", remote_ssh])
    if remote_folder:
        translated.extend(["--remote-mn-set-folder", remote_folder])

    for suite_id in selected:
        translated.extend(["--suite", suite_id])

    _set_legacy_environment(args)
    return translated


def _legacy_remote_ssh(args: argparse.Namespace) -> str | None:
    host = (
        args.shared_storage_remote_host
        or args.cluster_models_remote_host
        or args.redis_ha_remote_host
    )
    return f"ssh {host}" if host else None


def _set_legacy_environment(args: argparse.Namespace) -> None:
    values = {
        "MN_CLUSTER_REMOTE_HOST": (
            args.shared_storage_remote_host
            or args.cluster_models_remote_host
            or args.redis_ha_remote_host
        ),
        "MN_CLUSTER_REMOTE_SSH": args.cluster_models_remote_ssh,
        "MN_CLUSTER_REMOTE_MODEL": args.cluster_model,
        "MN_LIVE_LLM_REMOTE_NODE": args.llm_gateway_remote_node,
        "MN_REDIS_HA_LOCAL_IP": args.redis_ha_local_ip,
        "MN_REDIS_HA_REMOTE_IP": args.redis_ha_remote_ip,
        "MN_CLUSTER_REMOTE_MN_SET_FOLDER": args.remote_mn_set_folder,
    }
    for name, value in values.items():
        if value:
            os.environ[name] = value
    if args.cluster_model_expect_gpu:
        os.environ["MN_CLUSTER_MODEL_EXPECT_GPU"] = "1"


def main(argv: list[str] | None = None) -> int:
    print(
        "DEPRECATION: test_all.py will be removed after one release; use mn_test.",
        file=sys.stderr,
    )
    return mn_test_main(translate_legacy_args(list(argv if argv is not None else sys.argv[1:])))


if __name__ == "__main__":
    raise SystemExit(main())