voculinux/tests/test_installer_cuda_diagnostics.py

"""Regression checks for installer CUDA diagnostics."""

import unittest
from pathlib import Path

INSTALLER = Path(__file__).resolve().parents[1] / "install.sh"
INSTALLER_MODULES = Path(__file__).resolve().parents[1] / "install.d"


def _installer_source() -> str:
    parts = [INSTALLER, *sorted(INSTALLER_MODULES.glob("*.sh"))]
    return "\n".join(path.read_text(encoding="utf-8") for path in parts)


class InstallerCudaDiagnosticsTests(unittest.TestCase):
    def test_installer_disables_user_site_packages(self) -> None:
        """Keep user-site packages out of the venv, activation script, and wrappers."""
        source = _installer_source()

        self.assertGreaterEqual(source.count("export PYTHONNOUSERSITE=1"), 4)
        self.assertIn('getattr(site, "ENABLE_USER_SITE", False)', source)

    def test_cuda_toolkit_validation_requires_complete_root(self) -> None:
        """CUDA roots must not be accepted when only a stale /usr/local/cuda exists."""
        source = _installer_source()

        self.assertIn("validate_cuda_toolkit_root()", source)
        self.assertIn("$CUDA_ROOT/bin/nvcc", source)
        self.assertIn("$CUDA_ROOT/include/cuda_runtime.h", source)
        self.assertIn("$CUDA_ROOT/lib64/libcudart.so*", source)
        self.assertIn("Ignoring incomplete CUDA toolkit root", source)

    def test_cuda_build_passes_explicit_cmake_toolkit_and_architecture(self) -> None:
        """CUDA builds should steer CMake away from stale defaults."""
        source = _installer_source()

        self.assertIn("-DCUDAToolkit_ROOT=$CUDA_ROOT", source)
        self.assertIn("-DCMAKE_CUDA_COMPILER=$CUDA_ROOT/bin/nvcc", source)
        self.assertIn("-DCMAKE_CUDA_ARCHITECTURES=$CUDA_ARCHS", source)
        self.assertIn("CUDA_CMAKE_ARGS=$(get_cuda_cmake_args", source)
        self.assertIn("GGML_CUDA=1", source)

    def test_backend_menu_does_not_promise_to_install_cuda_toolkit(self) -> None:
        """NVIDIA boxes used to say the CUDA toolkit would be installed.

        install.sh never installs it. Vulkan is tried first; CUDA is a fallback
        only when a complete toolkit is already on the machine.
        """
        source = _installer_source()
        self.assertNotIn("CUDA toolkit will be installed", source)
        self.assertIn("NVIDIA GPU detected (Vulkan)", source)
        self.assertIn('printf "  │     • %-*s│\\n" 54 "$RECOMMENDED_REASON"', source)

    def test_gpu_build_failures_print_pip_log_tail_before_cpu_fallback(self) -> None:
        """Backend failures should expose the real pip/CMake log."""
        source = _installer_source()

        self.assertIn("print_pip_log_tail()", source)
        self.assertIn("Vulkan build failed; checking for NVIDIA GPU to try CUDA", source)
        self.assertIn("CUDA build failed.", source)
        self.assertIn(
            "Continuing with CPU-only pywhispercpp; GPU acceleration is not active.",
            source,
        )

    def test_backend_verification_checks_gpu_libraries_and_cuda_linkage(self) -> None:
        """Do not report GPU support unless native backend libraries exist."""
        source = _installer_source()

        self.assertIn("verify_pywhispercpp_backend_install()", source)
        self.assertIn("libggml-vulkan.so", source)
        self.assertIn("libggml-cuda.so", source)
        self.assertIn("libcuda.so.1", source)
        self.assertIn("libcuda-[^]]+\\.so", source)

    def test_patchelf_remediation_is_attempted_for_bundled_libcuda(self) -> None:
        """Bundled libcuda should trigger patchelf relinking when available."""
        source = _installer_source()

        self.assertIn("patchelf", source)
        self.assertIn("--replace-needed", source)
        self.assertIn("readelf not found", source)

    def test_installer_includes_xsel_for_wayland_clipboard_fallback(self) -> None:
        """Fresh installs should include xsel for ydotool's layout-safe paste path."""
        source = _installer_source()

        required_lines = [
            "local APT_PACKAGES_UBUNTU=",
            "local APT_PACKAGES_DEBIAN_BASE=",
            "local DNF_PACKAGES=",
            "local PACMAN_PACKAGES=",
            "local ZYPPER_PACKAGES=",
            "local EMERGE_PACKAGES=",
            "local APK_PACKAGES=",
            "local XBPS_PACKAGES=",
            "local EOPKG_PACKAGES=",
        ]

        for name in required_lines:
            line = next(line for line in source.splitlines() if name in line)
            self.assertIn("xclip", line)
            self.assertIn("xsel", line)
            self.assertIn("wl-clipboard", line)

    def test_ubuntu_22_package_list_omits_util_linux_extra(self) -> None:
        """util-linux-extra is probed via apt-cache, not hardcoded in base package lists."""
        source = _installer_source()

        ubuntu_line = next(
            line for line in source.splitlines() if "local APT_PACKAGES_UBUNTU=" in line
        )
        debian_13_line = next(
            line for line in source.splitlines() if "local APT_PACKAGES_DEBIAN_13_PLUS=" in line
        )
        self.assertNotIn("util-linux-extra", ubuntu_line)
        self.assertNotIn("util-linux-extra", debian_13_line)
        self.assertIn("apt-cache show util-linux-extra", source)
        self.assertIn('APT_PACKAGES="$APT_PACKAGES util-linux-extra"', source)


if __name__ == "__main__":
    unittest.main()