computer-playbook/tests/unit/python/utils/domains/test_list.py

import importlib
import tempfile
import unittest
from pathlib import Path
from unittest.mock import patch

from ansible.errors import AnsibleError

from utils.cache.yaml import dump_yaml_str
from utils.roles.mapping import (
    ROLE_FILE_META_DOMAINS,
    ROLE_FILE_META_SERVER,
    ROLE_FILE_VARS_MAIN,
)

domain_list = importlib.import_module("utils.domains.list")


class TestDomainList(unittest.TestCase):
    def write_role(self, roles_dir: Path, role_name: str, app_id: str, config: dict):
        role_dir = roles_dir / role_name
        (role_dir / "vars").mkdir(parents=True, exist_ok=True)
        (role_dir / "meta").mkdir(parents=True, exist_ok=True)
        (role_dir / ROLE_FILE_VARS_MAIN).write_text(
            dump_yaml_str({"application_id": app_id}),
            encoding="utf-8",
        )
        server_payload = (
            dict(config.get("server", {})) if isinstance(config, dict) else {}
        )
        domains_payload = server_payload.pop("domains", {})
        (role_dir / ROLE_FILE_META_SERVER).write_text(
            dump_yaml_str(server_payload),
            encoding="utf-8",
        )
        (role_dir / ROLE_FILE_META_DOMAINS).write_text(
            dump_yaml_str(domains_payload),
            encoding="utf-8",
        )

    def test_list_application_domains_renders_and_flattens_supported_shapes(self):
        with tempfile.TemporaryDirectory() as tmp:
            roles_dir = Path(tmp) / "roles"
            roles_dir.mkdir()

            self.write_role(
                roles_dir,
                "web-app-dashboard",
                "web-app-dashboard",
                {
                    "server": {
                        "domains": {
                            "canonical": ["dashboard.{{ DOMAIN_PRIMARY }}"],
                            "aliases": ["www.dashboard.{{ DOMAIN_PRIMARY }}"],
                        }
                    }
                },
            )
            self.write_role(
                roles_dir,
                "web-app-minio",
                "web-app-minio",
                {
                    "server": {
                        "domains": {
                            "canonical": {
                                "api": "api.s3.{{ DOMAIN_PRIMARY }}",
                                "console": "console.s3.{{ DOMAIN_PRIMARY }}",
                            },
                            "aliases": [],
                        }
                    }
                },
            )

            with patch.object(domain_list, "ROLES_DIR", roles_dir):
                domains = domain_list.list_application_domains("infinito.test")

            self.assertEqual(
                domains,
                sorted(
                    [
                        "api.s3.infinito.test",
                        "console.s3.infinito.test",
                        "dashboard.infinito.test",
                        "test.infinito.test",
                    ]
                ),
            )

    def test_list_application_domains_can_include_aliases_and_www_variants(self):
        with tempfile.TemporaryDirectory() as tmp:
            roles_dir = Path(tmp) / "roles"
            roles_dir.mkdir()

            self.write_role(
                roles_dir,
                "web-app-dashboard",
                "web-app-dashboard",
                {
                    "server": {
                        "domains": {
                            "canonical": ["dashboard.{{ DOMAIN_PRIMARY }}"],
                            "aliases": ["www.dashboard.{{ DOMAIN_PRIMARY }}"],
                        }
                    }
                },
            )

            with patch.object(domain_list, "ROLES_DIR", roles_dir):
                domains = domain_list.list_application_domains(
                    "infinito.test",
                    include_aliases=True,
                    include_www=True,
                )

            self.assertEqual(
                domains,
                [
                    "dashboard.infinito.test",
                    "test.infinito.test",
                    "www.dashboard.infinito.test",
                    "www.test.infinito.test",
                ],
            )

    def test_list_application_domains_includes_derived_test_domain_without_roles(self):
        with tempfile.TemporaryDirectory() as tmp:
            roles_dir = Path(tmp) / "roles"
            roles_dir.mkdir()

            with patch.object(domain_list, "ROLES_DIR", roles_dir):
                domains = domain_list.list_application_domains("infinito.test")

            self.assertEqual(domains, ["test.infinito.test"])

    def test_list_application_domains_raises_on_collisions(self):
        with tempfile.TemporaryDirectory() as tmp:
            roles_dir = Path(tmp) / "roles"
            roles_dir.mkdir()

            shared_config = {
                "server": {
                    "domains": {
                        "canonical": ["same.{{ DOMAIN_PRIMARY }}"],
                        "aliases": [],
                    }
                }
            }
            self.write_role(roles_dir, "app-a", "app-a", shared_config)
            self.write_role(roles_dir, "app-b", "app-b", shared_config)

            with (
                patch.object(domain_list, "ROLES_DIR", roles_dir),
                self.assertRaises(AnsibleError),
            ):
                domain_list.list_application_domains("infinito.test")


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