computer-playbook/tests/integration/infrastructure/services/test_frontend_spot.py

import unittest
from pathlib import Path

from utils.cache.files import read_text
from utils.roles.applications.services.registry import (
    build_service_registry_from_roles_dir,
    load_applications_from_roles_dir,
    ordered_primary_service_entries,
)
from utils.roles.mapping import ROLE_FILE_TASKS_MAIN

from . import PROJECT_ROOT


class TestFrontendServiceSpot(unittest.TestCase):
    def setUp(self):
        self.repo_root = PROJECT_ROOT
        self.roles_dir = self.repo_root / "roles"
        self.applications = load_applications_from_roles_dir(self.roles_dir)
        self.service_registry = build_service_registry_from_roles_dir(self.roles_dir)
        self.ordered = ordered_primary_service_entries(
            self.service_registry,
            self.roles_dir,
        )

    def _index(self, role_name: str) -> int:
        for index, entry in enumerate(self.ordered):
            if entry["role"] == role_name:
                return index
        self.fail(f"Role {role_name} not found in ordered service registry")
        return -1

    def test_bucket_order_is_monotonic(self):
        bucket_order = {
            "universal": 0,
            "workstation": 1,
            "server": 2,
            "web-svc": 3,
            "web-app": 4,
        }
        actual = [bucket_order[entry["bucket"]] for entry in self.ordered]
        self.assertEqual(actual, sorted(actual))

    def test_dashboard_is_loaded_after_matomo(self):
        self.assertLess(self._index("web-app-matomo"), self._index("web-app-dashboard"))

    def test_keycloak_respects_run_after_dependencies(self):
        self.assertLess(self._index("web-app-matomo"), self._index("web-app-keycloak"))
        self.assertLess(
            self._index("web-app-stalwart"), self._index("web-app-keycloak")
        )
        self.assertLess(self._index("web-app-mailu"), self._index("web-app-keycloak"))

    def test_front_proxy_falls_back_to_canonical_port_and_domain(self):
        content = read_text(
            str(Path(f"roles/sys-stk-front-proxy/{ROLE_FILE_TASKS_MAIN}"))
        )
        self.assertIn(
            "lookup('config', application_id,",
            content,
        )
        self.assertIn(
            ".ports.local.http",
            content,
        )
        self.assertIn(
            "domain | default(lookup('domain', application_id))",
            content,
        )


if __name__ == "__main__":
    unittest.main()