144 lines
4.7 KiB
Python
144 lines
4.7 KiB
Python
"""Tests for flow.commands.bootstrap helpers and schema behavior."""
|
|
|
|
import os
|
|
|
|
import pytest
|
|
|
|
from flow.commands.bootstrap import (
|
|
_ensure_required_variables,
|
|
_get_profiles,
|
|
_normalize_profile_package_entry,
|
|
_resolve_package_manager,
|
|
_resolve_package_spec,
|
|
_resolve_pkg_source_name,
|
|
)
|
|
from flow.core.config import AppConfig, FlowContext
|
|
from flow.core.console import ConsoleLogger
|
|
from flow.core.platform import PlatformInfo
|
|
|
|
|
|
@pytest.fixture
|
|
def ctx():
|
|
return FlowContext(
|
|
config=AppConfig(),
|
|
manifest={
|
|
"packages": [
|
|
{
|
|
"name": "fd",
|
|
"type": "pkg",
|
|
"sources": {"apt": "fd-find", "dnf": "fd-find", "brew": "fd"},
|
|
},
|
|
{
|
|
"name": "neovim",
|
|
"type": "binary",
|
|
"source": "github:neovim/neovim",
|
|
"version": "0.10.4",
|
|
"asset-pattern": "nvim-{{os}}-{{arch}}.tar.gz",
|
|
"platform-map": {"linux-x64": {"os": "linux", "arch": "x64"}},
|
|
"install": {"bin": ["bin/nvim"]},
|
|
},
|
|
]
|
|
},
|
|
platform=PlatformInfo(os="linux", arch="x64", platform="linux-x64"),
|
|
console=ConsoleLogger(),
|
|
)
|
|
|
|
|
|
def test_get_profiles_from_manifest(ctx):
|
|
ctx.manifest = {"profiles": {"linux": {"os": "linux"}}}
|
|
assert "linux" in _get_profiles(ctx)
|
|
|
|
|
|
def test_get_profiles_rejects_environments(ctx):
|
|
ctx.manifest = {"environments": {"legacy": {"os": "linux"}}}
|
|
with pytest.raises(RuntimeError, match="no longer supported"):
|
|
_get_profiles(ctx)
|
|
|
|
|
|
def test_resolve_package_manager_explicit_value(ctx):
|
|
assert _resolve_package_manager(ctx, {"os": "linux", "package-manager": "dnf"}) == "dnf"
|
|
|
|
|
|
def test_resolve_package_manager_linux_auto_apt(monkeypatch, ctx):
|
|
monkeypatch.setattr("flow.commands.bootstrap.shutil.which", lambda name: "/usr/bin/apt" if name == "apt" else None)
|
|
assert _resolve_package_manager(ctx, {"os": "linux"}) == "apt"
|
|
|
|
|
|
def test_resolve_package_manager_linux_auto_dnf(monkeypatch, ctx):
|
|
monkeypatch.setattr("flow.commands.bootstrap.shutil.which", lambda name: "/usr/bin/dnf" if name == "dnf" else None)
|
|
assert _resolve_package_manager(ctx, {"os": "linux"}) == "dnf"
|
|
|
|
|
|
def test_resolve_package_manager_requires_os(ctx):
|
|
with pytest.raises(RuntimeError, match="must be set"):
|
|
_resolve_package_manager(ctx, {})
|
|
|
|
|
|
def test_normalize_package_entry_string():
|
|
assert _normalize_profile_package_entry("git") == {"name": "git"}
|
|
|
|
|
|
def test_normalize_package_entry_type_prefix():
|
|
assert _normalize_profile_package_entry("cask/wezterm") == {"name": "wezterm", "type": "cask"}
|
|
|
|
|
|
def test_normalize_package_entry_object():
|
|
out = _normalize_profile_package_entry({"name": "docker", "allow_sudo": True})
|
|
assert out["name"] == "docker"
|
|
assert out["allow_sudo"] is True
|
|
|
|
|
|
def test_resolve_package_spec_uses_catalog_type(ctx):
|
|
catalog = {
|
|
"fd": {
|
|
"name": "fd",
|
|
"type": "pkg",
|
|
"sources": {"apt": "fd-find"},
|
|
}
|
|
}
|
|
resolved = _resolve_package_spec(catalog, {"name": "fd"})
|
|
assert resolved["type"] == "pkg"
|
|
assert resolved["sources"]["apt"] == "fd-find"
|
|
|
|
|
|
def test_resolve_package_spec_defaults_to_pkg(ctx):
|
|
resolved = _resolve_package_spec({}, {"name": "git"})
|
|
assert resolved["type"] == "pkg"
|
|
|
|
|
|
def test_resolve_package_spec_profile_override(ctx):
|
|
catalog = {
|
|
"neovim": {
|
|
"name": "neovim",
|
|
"type": "binary",
|
|
"version": "0.10.4",
|
|
}
|
|
}
|
|
resolved = _resolve_package_spec(catalog, {"name": "neovim", "post-install": "echo ok"})
|
|
assert resolved["type"] == "binary"
|
|
assert resolved["post-install"] == "echo ok"
|
|
|
|
|
|
def test_resolve_pkg_source_name_with_mapping(ctx):
|
|
spec = {"name": "fd", "sources": {"apt": "fd-find", "dnf": "fd-find", "brew": "fd"}}
|
|
assert _resolve_pkg_source_name(spec, "apt") == "fd-find"
|
|
assert _resolve_pkg_source_name(spec, "dnf") == "fd-find"
|
|
assert _resolve_pkg_source_name(spec, "brew") == "fd"
|
|
|
|
|
|
def test_resolve_pkg_source_name_fallback_to_name(ctx):
|
|
spec = {"name": "ripgrep", "sources": {"apt": "ripgrep"}}
|
|
assert _resolve_pkg_source_name(spec, "dnf") == "ripgrep"
|
|
|
|
|
|
def test_ensure_required_variables_missing_raises():
|
|
with pytest.raises(RuntimeError, match="Missing required environment variables"):
|
|
_ensure_required_variables({"requires": ["USER_EMAIL", "TARGET_HOSTNAME"]}, {"USER_EMAIL": "a@b"})
|
|
|
|
|
|
def test_ensure_required_variables_accepts_vars(monkeypatch):
|
|
env = dict(os.environ)
|
|
env["USER_EMAIL"] = "a@b"
|
|
env["TARGET_HOSTNAME"] = "devbox"
|
|
_ensure_required_variables({"requires": ["USER_EMAIL", "TARGET_HOSTNAME"]}, env)
|