24 lines
777 B
Python
24 lines
777 B
Python
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")
|