import shutil
import tempfile
import unittest
from pathlib import Path

from plugins.lookup.application_gid import LookupModule
from utils.cache.yaml import dump_yaml


class TestApplicationGidLookup(unittest.TestCase):
    def setUp(self):
        self.temp_dir = tempfile.mkdtemp()
        self.roles_dir = str(Path(self.temp_dir) / "roles")
        Path(self.roles_dir).mkdir()

        self.application_ids = [
            "web-app-nextcloud",
            "web-app-moodle",
            "web-app-wordpress",
            "web-app-taiga",
        ]

        for application_id in self.application_ids:
            role_path = str(Path(self.roles_dir) / application_id / "meta")
            Path(role_path).mkdir(parents=True)
            dump_yaml(
                str(Path(role_path) / "services.yml"),
                {application_id: {"title": application_id}},
            )

        self.lookup = LookupModule()

    def tearDown(self):
        shutil.rmtree(self.temp_dir)

    def test_gid_lookup(self):
        expected_order = sorted(self.application_ids)
        for i, application_id in enumerate(expected_order):
            result = self.lookup.run([application_id], roles_dir=self.roles_dir)
            self.assertEqual(result, [10000 + i])

    def test_custom_base_gid(self):
        result = self.lookup.run(
            ["web-app-taiga"],
            roles_dir=self.roles_dir,
            base_gid=20000,
        )
        self.assertEqual(result, [20002])  # 2nd index in sorted list

    def test_application_id_not_found(self):
        with self.assertRaises(Exception) as context:
            self.lookup.run(["unknownapp"], roles_dir=self.roles_dir)
        self.assertIn("Application ID 'unknownapp' not found", str(context.exception))


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