More test coverage
This commit is contained in:
parent
528176ad82
commit
fc120f02a5
3 changed files with 156 additions and 4 deletions
|
|
@ -141,3 +141,107 @@ def test_container_images_collector_records_unpullable_tagged_images(
|
|||
|
||||
assert result.images[0]["pull_ref"] is None
|
||||
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]
|
||||
|
|
|
|||
|
|
@ -1,7 +1,6 @@
|
|||
from __future__ import annotations
|
||||
|
||||
# The version module is hard to test fully because it uses importlib.metadata
|
||||
# which is difficult to mock. We'll test what we can.
|
||||
import importlib.metadata as metadata
|
||||
|
||||
|
||||
def test_get_enroll_version_returns_string():
|
||||
|
|
@ -10,3 +9,51 @@ def test_get_enroll_version_returns_string():
|
|||
result = get_enroll_version()
|
||||
assert isinstance(result, str)
|
||||
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"
|
||||
|
|
|
|||
Reference in a new issue