Files
computer-playbook/tests/unit/cli/mirror/test_resolver.py
T
Environment TestandClaude Sonnet 4.6 aa33256cde feat(mirror): split resolver output, move images to defaults, add lookup plugin
- resolver: output both applications: (config/main.yml compose images) and
  images: (defaults/main.yml images) keyed by source_file
- image_discovery: scan defaults/main.yml instead of vars/main.yml for images:
- mirror_overrides: write mirror URLs to applications.*.compose.services.*
  (mirror_policy-aware) and images_overrides.*.* (always merged, if_missing)
- roles/sys-ctl-hlth-csp, roles/test-e2e-playwright: move images: from
  vars/main.yml to defaults/main.yml for correct Ansible variable precedence
- plugins/lookup/image.py: new Ansible lookup plugin that prefers
  images_overrides over role-local images defaults
- docs: add artefact/image.md and artefact/mirror.md as SPOTs; add
  code/role/image.md contributor guide; link from ci.md
- tests: update test_util, test_mirror_overrides; add test_resolver,
  test_publish (404/401 coverage), test_image (lookup plugin)

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
2026-04-10 11:17:42 +02:00

127 lines
4.1 KiB
Python

from __future__ import annotations
import io
import unittest
from unittest.mock import patch
import yaml
import cli.mirror.resolver.__main__ as resolver_main
from cli.mirror.model import ImageRef
class TestResolverOutputFormat(unittest.TestCase):
"""resolver preserves compose output and adds a separate images key."""
def _run_resolver(
self, images: list[ImageRef], extra_argv: list[str] | None = None
) -> dict:
argv = [
"resolver",
"--ghcr-namespace",
"acme",
"--ghcr-repository",
"myrepo",
]
if extra_argv:
argv += extra_argv
buf = io.StringIO()
with (
patch("cli.mirror.resolver.__main__.iter_role_images", return_value=images),
patch("sys.argv", argv),
patch("sys.stdout", buf),
):
result = resolver_main.main()
self.assertEqual(result, 0)
return yaml.safe_load(buf.getvalue()) or {}
def test_compose_images_go_to_applications_key(self) -> None:
image = ImageRef(
role="web-app-nextcloud",
service="app",
name="nextcloud",
version="31.0.0",
source="docker.io/library/nextcloud:31.0.0",
registry="docker.io",
source_file="config/main.yml",
)
out = self._run_resolver([image])
self.assertIn("applications", out)
self.assertIn("images", out)
svc = out["applications"]["web-app-nextcloud"]["compose"]["services"]["app"]
self.assertEqual(svc["version"], "31.0.0")
self.assertIn("ghcr.io/acme/myrepo/mirror/docker.io/nextcloud", svc["image"])
self.assertEqual(out["images"], {})
def test_defaults_images_go_to_images_key(self) -> None:
image = ImageRef(
role="test-e2e-playwright",
service="playwright",
name="playwright",
version="v1.58.2-noble",
source="mcr.microsoft.com/playwright:v1.58.2-noble",
registry="mcr.microsoft.com",
source_file="defaults/main.yml",
)
out = self._run_resolver([image])
self.assertEqual(out["applications"], {})
svc = out["images"]["test-e2e-playwright"]["playwright"]
self.assertEqual(svc["version"], "v1.58.2-noble")
self.assertIn(
"ghcr.io/acme/myrepo/mirror/mcr.microsoft.com/playwright",
svc["image"],
)
def test_multiple_services_same_role_stay_grouped_per_source_type(self) -> None:
images = [
ImageRef(
role="web-app-nextcloud",
service="app",
name="nextcloud",
version="31.0.0",
source="docker.io/library/nextcloud:31.0.0",
registry="docker.io",
source_file="config/main.yml",
),
ImageRef(
role="web-app-nextcloud",
service="proxy",
name="nginx",
version="alpine",
source="docker.io/library/nginx:alpine",
registry="docker.io",
source_file="config/main.yml",
),
ImageRef(
role="web-app-nextcloud",
service="backup",
name="restic",
version="latest",
source="docker.io/restic/restic:latest",
registry="docker.io",
source_file="defaults/main.yml",
),
]
out = self._run_resolver(images)
self.assertIn(
"app", out["applications"]["web-app-nextcloud"]["compose"]["services"]
)
self.assertIn(
"proxy", out["applications"]["web-app-nextcloud"]["compose"]["services"]
)
self.assertIn("backup", out["images"]["web-app-nextcloud"])
def test_empty_images_yield_both_empty_top_level_keys(self) -> None:
out = self._run_resolver([])
self.assertEqual(out["applications"], {})
self.assertEqual(out["images"], {})
if __name__ == "__main__": # pragma: no cover
unittest.main()