mirror of
https://github.com/kevinveenbirkenbach/computer-playbook.git
synced 2026-11-08 19:56:53 +00:00
kevinveenbirkenbach
5e2a5fcc31
# Conflicts: # scripts/lint/mermaid.sh # tests/unit/python/utils/docker/test_image_discovery_lifecycle.py # utils/docker/image/discovery.py
86 lines
2.7 KiB
Python
86 lines
2.7 KiB
Python
from __future__ import annotations
|
|
|
|
import os
|
|
import tempfile
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
from utils.docker.image.discovery import iter_role_images, role_lifecycle
|
|
from utils.roles.mapping import ROLE_FILE_META_SERVICES
|
|
|
|
SUPPORTED = """---
|
|
demo:
|
|
image: libretranslate/libretranslate
|
|
version: v1.9.6
|
|
lifecycle: stable
|
|
helper:
|
|
image: alpine
|
|
version: "3.22"
|
|
"""
|
|
|
|
RETIRED = """---
|
|
demo:
|
|
image: quay.io/minio/minio
|
|
version: RELEASE.2025-09-07T16-13-09Z
|
|
lifecycle: eol
|
|
client:
|
|
image: quay.io/minio/mc
|
|
version: RELEASE.2025-08-13T08-35-41Z
|
|
"""
|
|
|
|
UNSTAGED = """---
|
|
demo:
|
|
image: libretranslate/libretranslate
|
|
version: v1.9.6
|
|
"""
|
|
|
|
|
|
class TestImageDiscoveryLifecycle(unittest.TestCase):
|
|
def _images(self, services_yaml: str, *, relative: bool = False) -> set[str]:
|
|
"""Discover the images of one role written into a temporary repository.
|
|
|
|
Args:
|
|
services_yaml: the role's ``meta/services.yml``.
|
|
relative: pass the repository root relative to the working directory.
|
|
"""
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
services = Path(tmp) / "roles" / "web-app-demo" / ROLE_FILE_META_SERVICES
|
|
services.parent.mkdir(parents=True)
|
|
services.write_text(services_yaml, encoding="utf-8")
|
|
root = Path(os.path.relpath(tmp)) if relative else Path(tmp)
|
|
return {ref.service for ref in iter_role_images(root)}
|
|
|
|
def test_a_supported_role_yields_every_service(self) -> None:
|
|
self.assertEqual({"demo", "helper"}, self._images(SUPPORTED))
|
|
|
|
def test_a_retired_role_yields_none_of_its_services(self) -> None:
|
|
"""Its upstream may withdraw the images; no deploy can reach them."""
|
|
self.assertEqual(set(), self._images(RETIRED))
|
|
|
|
def test_the_sibling_of_a_retired_service_goes_with_it(self) -> None:
|
|
"""Only the primary entry carries the stage, and it speaks for the role."""
|
|
self.assertNotIn("client", self._images(RETIRED))
|
|
|
|
def test_a_role_that_names_no_stage_is_kept(self) -> None:
|
|
self.assertEqual({"demo"}, self._images(UNSTAGED))
|
|
|
|
def test_a_relative_root_still_leaves_a_retired_role_out(self) -> None:
|
|
self.assertEqual(set(), self._images(RETIRED, relative=True))
|
|
|
|
|
|
class TestRoleLifecycle(unittest.TestCase):
|
|
def test_it_reads_the_entry_that_carries_the_field(self) -> None:
|
|
self.assertEqual(
|
|
"eol", role_lifecycle({"a": {"image": "x"}, "b": {"lifecycle": "eol"}})
|
|
)
|
|
|
|
def test_it_is_empty_when_no_entry_carries_one(self) -> None:
|
|
self.assertEqual("", role_lifecycle({"a": {"image": "x"}}))
|
|
|
|
def test_a_non_mapping_entry_does_not_break_it(self) -> None:
|
|
self.assertEqual("", role_lifecycle({"a": "scalar"}))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|