import unittest
from pathlib import Path

from plugins.filter.invokable_paths import get_invokable_paths
from utils.cache.yaml import load_yaml_any
from utils.roles.categories import categories_file
from utils.roles.mapping import ROLE_FILE_VARS_MAIN


class TestSysRolesApplicationId(unittest.TestCase):
    """
    Integration tests for sys-* roles based on categories.yml prefixes:
    For each actual sys-* directory under roles/:
      - If dash-joined prefix is in invokable_paths -> vars/main.yml must exist and contain application_id.
      - Otherwise (non-invokable or undeclared) -> if vars/main.yml exists, it must NOT contain application_id.
    """

    @classmethod
    def setUpClass(cls):
        from . import PROJECT_ROOT

        cls.base_dir = str(PROJECT_ROOT)
        cat_file = str(categories_file())
        cls.invokable_prefixes = set(get_invokable_paths(cat_file))
        roles_dir = Path(cls.base_dir) / "roles"
        cls.actual_dirs = [
            str(p)
            for p in roles_dir.iterdir()
            if p.is_dir() and p.name.startswith("sys-")
        ]

    def test_sys_roles_application_id(self):
        for role_dir in self.actual_dirs:
            name = Path(role_dir).name
            prefix = f"sys-{name.removeprefix('sys-')}"
            vars_file = str(Path(role_dir) / ROLE_FILE_VARS_MAIN)
            if prefix in self.invokable_prefixes:
                self.assertTrue(
                    Path(vars_file).is_file(),
                    f"Missing vars/main.yml for invokable role {prefix}",
                )
                content = load_yaml_any(vars_file) or {}
                self.assertIn(
                    "application_id",
                    content,
                    f"Expected 'application_id' in {vars_file} for invokable role {prefix}",
                )
            else:
                if not Path(vars_file).is_file():
                    continue
                content = load_yaml_any(vars_file) or {}
                self.assertNotIn(
                    "application_id",
                    content,
                    f"Unexpected 'application_id' in {vars_file} for non-invokable role {prefix}",
                )


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