from __future__ import annotations

from pathlib import Path

import pytest

from pyro_mcp.vm_profiles import get_profile, list_profiles, resolve_artifacts


def test_list_profiles_includes_expected_entries() -> None:
    profiles = list_profiles()
    names = {str(entry["name"]) for entry in profiles}
    assert {"debian-base", "debian-git", "debian-build"} <= names


def test_get_profile_rejects_unknown() -> None:
    with pytest.raises(ValueError, match="unknown profile"):
        get_profile("does-not-exist")


def test_resolve_artifacts() -> None:
    artifacts = resolve_artifacts(Path("/tmp/artifacts"), "debian-git")
    assert str(artifacts.kernel_image).endswith("/debian-git/vmlinux")
    assert str(artifacts.rootfs_image).endswith("/debian-git/rootfs.ext4")