bot-bottle/tests/unit/test_wheel_install.py

"""Integration: build the wheel, install it into an isolated venv, and prove
the installed distribution is self-contained.

This is the boundary a source-tree existence test can't reach (issue #197
review): under an installed wheel the package lives in ``site-packages`` with
no repo root above it, so anything resolving Dockerfiles / nix / scripts from
``__file__``'s parents would break. Here we install for real and assert that
``bot-bottle doctor`` runs from the console script and that
``bot_bottle.resources`` stages a valid, repo-root-shaped build context.

It does NOT run `start` — building images needs a Docker/KVM host (CI). It
skips cleanly when the build/venv toolchain isn't available.
"""

from __future__ import annotations

import importlib.util
import subprocess
import sys
import tempfile
import unittest
from pathlib import Path

REPO_ROOT = Path(__file__).resolve().parents[2]


def _tooling_available() -> bool:
    return importlib.util.find_spec("build") is not None


@unittest.skipUnless(_tooling_available(), "python 'build' module not installed")
class TestWheelInstall(unittest.TestCase):
    _tmp: "tempfile.TemporaryDirectory[str]"
    venv_py: Path
    app_root: Path

    @classmethod
    def setUpClass(cls):
        cls._tmp = tempfile.TemporaryDirectory(  # pylint: disable=consider-using-with
            prefix="bb-wheel-")
        tmp = Path(cls._tmp.name)
        dist = tmp / "dist"

        build = subprocess.run(
            [sys.executable, "-m", "build", "--wheel", "--outdir", str(dist), str(REPO_ROOT)],
            capture_output=True, text=True, check=False,
        )
        if build.returncode != 0:
            raise unittest.SkipTest(f"wheel build unavailable:\n{build.stderr[-1500:]}")
        wheels = list(dist.glob("*.whl"))
        if not wheels:
            raise unittest.SkipTest("no wheel produced")

        venv = tmp / "venv"
        made = subprocess.run([sys.executable, "-m", "venv", str(venv)],
                              capture_output=True, text=True, check=False)
        if made.returncode != 0:
            raise unittest.SkipTest(f"venv unavailable:\n{made.stderr[-1500:]}")
        cls.venv_py = venv / "bin" / "python"

        install = subprocess.run(
            [str(cls.venv_py), "-m", "pip", "install", "--quiet", str(wheels[0])],
            capture_output=True, text=True, check=False,
        )
        if install.returncode != 0:
            raise unittest.SkipTest(f"pip install failed:\n{install.stderr[-1500:]}")

        # Isolate the staged build root the wheel writes under the app-data dir.
        cls.app_root = tmp / "appdata"

    @classmethod
    def tearDownClass(cls):
        cls._tmp.cleanup()

    def _run(self, tail: "list[str]") -> "subprocess.CompletedProcess[str]":
        """Run the installed venv's python with `tail` appended, from a neutral
        cwd so the source checkout isn't on sys.path — we must import the
        *installed* package, not the repo we built from."""
        env = {"BOT_BOTTLE_ROOT": str(self.app_root), "PATH": "/usr/bin:/bin"}
        return subprocess.run(
            [str(self.venv_py), *tail],
            capture_output=True, text=True, env=env, cwd=self._tmp.name, check=False,
        )

    def test_console_entry_point_installed(self):
        # The `bot-bottle` script the wheel declares must exist in the venv.
        script = self.venv_py.parent / "bot-bottle"
        self.assertTrue(script.is_file(), "bot-bottle console script not installed")

    def test_doctor_runs_from_installed_package(self):
        proc = self._run(["-m", "bot_bottle.cli", "doctor"])
        # doctor exits non-zero here (no backend), but it must RUN and report.
        self.assertIn("python", proc.stdout)
        self.assertIn(proc.returncode, (0, 1))

    def test_installed_wheel_is_self_contained(self):
        # From the installed layout (not a checkout), resources must resolve
        # Dockerfiles and stage a repo-root-shaped build context.
        script = (
            "import bot_bottle.resources as r\n"
            "assert not r.is_source_checkout(), 'should not look like a checkout'\n"
            "assert r.dockerfile('Dockerfile.gateway').is_file()\n"
            "assert r.nix_netpool_module().is_file()\n"
            "assert r.netpool_script().is_file()\n"
            "root = r.build_root()\n"
            "assert (root / 'bot_bottle' / '__init__.py').is_file(), 'no package in context'\n"
            "assert (root / 'pyproject.toml').is_file(), 'no pyproject in context'\n"
            "assert (root / 'Dockerfile.gateway').is_file(), 'no Dockerfile in context'\n"
            "print('SELF_CONTAINED_OK')\n"
        )
        proc = self._run(["-c", script])
        self.assertIn("SELF_CONTAINED_OK", proc.stdout, msg=proc.stderr)


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