More test coverage

This commit is contained in:
Miguel Jacq 2026-06-21 13:37:37 +10:00
parent 528176ad82
commit fc120f02a5
Signed by: mig5
GPG key ID: 03906B4110AAD3B8
3 changed files with 156 additions and 4 deletions

View file

@ -28,5 +28,6 @@ def get_enroll_version() -> str:
for dist in [*dist_names, "enroll"]: for dist in [*dist_names, "enroll"]:
try: try:
return version(dist) return version(dist)
except Exception: except Exception: # nosec B112
return "unknown" continue
return "unknown"

View file

@ -141,3 +141,107 @@ def test_container_images_collector_records_unpullable_tagged_images(
assert result.images[0]["pull_ref"] is None assert result.images[0]["pull_ref"] is None
assert "exact digest-pinned pull cannot be rendered" in result.images[0]["notes"][0] assert "exact digest-pinned pull cannot be rendered" in result.images[0]["notes"][0]
def test_container_images_collector_notes_list_exceptions(monkeypatch, tmp_path):
from enroll.harvest_collectors import container_images as ci
from enroll.harvest_collectors.container_images import ContainerImagesCollector
monkeypatch.setattr(
ci.shutil,
"which",
lambda cmd: f"/usr/bin/{cmd}" if cmd == "docker" else None,
)
def boom(_argv, *, timeout=20):
raise RuntimeError("socket unavailable")
monkeypatch.setattr(ci, "_run_command", boom)
result = ContainerImagesCollector(_context(tmp_path)).collect()
assert result.images == []
assert "Failed to list docker images" in result.notes[0]
def test_container_images_collector_notes_list_nonzero_without_detail(
monkeypatch, tmp_path
):
import subprocess
from enroll.harvest_collectors import container_images as ci
from enroll.harvest_collectors.container_images import ContainerImagesCollector
monkeypatch.setattr(
ci.shutil,
"which",
lambda cmd: f"/usr/bin/{cmd}" if cmd == "podman" else None,
)
monkeypatch.setattr(
ci,
"_run_command",
lambda argv, *, timeout=20: subprocess.CompletedProcess(argv, 42, "", ""),
)
result = ContainerImagesCollector(_context(tmp_path)).collect()
assert result.images == []
assert "exit 42" in result.notes[0]
def test_container_images_collector_notes_bad_inspect_json(monkeypatch, tmp_path):
import subprocess
from enroll.harvest_collectors import container_images as ci
from enroll.harvest_collectors.container_images import ContainerImagesCollector
image_id = "sha256:" + "d" * 64
monkeypatch.setattr(
ci.shutil,
"which",
lambda cmd: f"/usr/bin/{cmd}" if cmd == "docker" else None,
)
def fake_run(argv, *, timeout=20):
if argv[:4] == ["/usr/bin/docker", "image", "ls", "-q"]:
return subprocess.CompletedProcess(argv, 0, image_id + "\n", "")
if argv[:3] == ["/usr/bin/docker", "image", "inspect"]:
return subprocess.CompletedProcess(argv, 0, "not json", "")
raise AssertionError(argv)
monkeypatch.setattr(ci, "_run_command", fake_run)
result = ContainerImagesCollector(_context(tmp_path)).collect()
assert result.images == []
assert "Failed to parse docker image inspect JSON" in result.notes[0]
def test_container_images_collector_notes_unexpected_inspect_shape(
monkeypatch, tmp_path
):
import subprocess
from enroll.harvest_collectors import container_images as ci
from enroll.harvest_collectors.container_images import ContainerImagesCollector
image_id = "sha256:" + "e" * 64
monkeypatch.setattr(
ci.shutil,
"which",
lambda cmd: f"/usr/bin/{cmd}" if cmd == "docker" else None,
)
def fake_run(argv, *, timeout=20):
if argv[:4] == ["/usr/bin/docker", "image", "ls", "-q"]:
return subprocess.CompletedProcess(argv, 0, image_id + "\n", "")
if argv[:3] == ["/usr/bin/docker", "image", "inspect"]:
return subprocess.CompletedProcess(argv, 0, '{"not":"a-list"}', "")
raise AssertionError(argv)
monkeypatch.setattr(ci, "_run_command", fake_run)
result = ContainerImagesCollector(_context(tmp_path)).collect()
assert result.images == []
assert "Unexpected docker image inspect JSON shape" in result.notes[0]

View file

@ -1,7 +1,6 @@
from __future__ import annotations from __future__ import annotations
# The version module is hard to test fully because it uses importlib.metadata import importlib.metadata as metadata
# which is difficult to mock. We'll test what we can.
def test_get_enroll_version_returns_string(): def test_get_enroll_version_returns_string():
@ -10,3 +9,51 @@ def test_get_enroll_version_returns_string():
result = get_enroll_version() result = get_enroll_version()
assert isinstance(result, str) assert isinstance(result, str)
assert len(result) > 0 assert len(result) > 0
def test_get_enroll_version_falls_back_after_bad_mapped_dist(monkeypatch):
from enroll.version import get_enroll_version
monkeypatch.setattr(
metadata,
"packages_distributions",
lambda: {"enroll": ["wrong-dist", "enroll"]},
)
def fake_version(name):
if name == "enroll":
return "0.7.0"
raise metadata.PackageNotFoundError(name)
monkeypatch.setattr(metadata, "version", fake_version)
assert get_enroll_version() == "0.7.0"
def test_get_enroll_version_uses_default_when_mapping_fails(monkeypatch):
from enroll.version import get_enroll_version
def broken_mapping():
raise RuntimeError("metadata unavailable")
monkeypatch.setattr(metadata, "packages_distributions", broken_mapping)
monkeypatch.setattr(
metadata,
"version",
lambda name: "0.7.0" if name == "enroll" else "bad",
)
assert get_enroll_version() == "0.7.0"
def test_get_enroll_version_returns_unknown_when_all_lookups_fail(monkeypatch):
from enroll.version import get_enroll_version
monkeypatch.setattr(metadata, "packages_distributions", lambda: {"enroll": ["bad"]})
def missing(_name):
raise metadata.PackageNotFoundError(_name)
monkeypatch.setattr(metadata, "version", missing)
assert get_enroll_version() == "unknown"