from __future__ import annotations

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 = """---
engine:
  image: libretranslate/libretranslate
  version: v1.9.6
  lifecycle: stable
helper:
  image: alpine
  version: "3.22"
"""

RETIRED = """---
store:
  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 = """---
engine:
  image: libretranslate/libretranslate
  version: v1.9.6
"""


class TestImageDiscoveryLifecycle(unittest.TestCase):
    def _images(self, services_yaml: str) -> set[str]:
        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")
            return {ref.service for ref in iter_role_images(Path(tmp))}

    def test_a_supported_role_yields_every_service(self) -> None:
        self.assertEqual({"engine", "helper"}, self._images(SUPPORTED))

    def test_a_retired_role_yields_nothing(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({"engine"}, self._images(UNSTAGED))


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()