from __future__ import annotations

import base64
import importlib.util
import sys
import unittest
from pathlib import Path
from unittest.mock import patch


MODULE_PATH = Path(__file__).resolve().parents[1] / "scripts" / "queue_post_merge_restore.py"


def load_module():
    # Preload common helper so the script can import it.
    common_path = MODULE_PATH.parent / "common.py"
    common_spec = importlib.util.spec_from_file_location("common", common_path)
    if common_spec is not None and common_spec.loader is not None:
        common_mod = importlib.util.module_from_spec(common_spec)
        sys.modules["common"] = common_mod
        common_spec.loader.exec_module(common_mod)

    module_name = "queue_post_merge_restore"
    spec = importlib.util.spec_from_file_location(module_name, MODULE_PATH)
    if spec is None or spec.loader is None:
        raise RuntimeError(f"Unable to load module from {MODULE_PATH}")
    module = importlib.util.module_from_spec(spec)
    sys.modules[module_name] = module
    spec.loader.exec_module(module)
    return module


def _marker(path: str) -> str:
    encoded = base64.urlsafe_b64encode(path.encode("utf-8")).decode("ascii").rstrip("=")
    return f"Automation marker: AUTO-CHANGE-TICKET:{encoded}"


class QueuePostMergeRestoreTests(unittest.TestCase):
    @classmethod
    def setUpClass(cls) -> None:
        cls.module = load_module()

    def test_ticket_path_from_content_decodes_marker(self) -> None:
        path = "tenant-state/intune/Device Configurations/macOS - WiFi TEST_macOSWiFiConfiguration__id.json"
        content = f"Header\n{_marker(path)}\nBody"
        self.assertEqual(self.module._ticket_path_from_content(content), path)

    def test_rejected_ticket_paths_uses_latest_decision(self) -> None:
        accepted_path = "tenant-state/intune/Settings Catalog/A.json"
        rejected_path = "tenant-state/intune/Settings Catalog/B.json"
        threads = [
            {
                "comments": [
                    {"id": 1, "parentCommentId": 0, "content": _marker(accepted_path)},
                    {"id": 2, "parentCommentId": 0, "content": "/reject"},
                    {"id": 3, "parentCommentId": 0, "content": "/accept"},
                ]
            },
            {
                "comments": [
                    {"id": 1, "parentCommentId": 0, "content": _marker(rejected_path)},
                    {"id": 2, "parentCommentId": 0, "content": "/accept"},
                    {"id": 3, "parentCommentId": 0, "content": "/reject"},
                ]
            },
        ]
        self.assertEqual(self.module._rejected_ticket_paths(threads), [rejected_path])

    def test_queue_restore_pipeline_includes_selective_params(self) -> None:
        captured: dict[str, object] = {}

        def _fake_request(url: str, headers: dict[str, str], method: str = "GET", body: dict | None = None):
            captured["url"] = url
            captured["method"] = method
            captured["body"] = body or {}
            return {"id": 123}

        with patch.object(self.module, "_request_json", side_effect=_fake_request):
            self.module._queue_restore_pipeline(
                collection_uri="https://dev.azure.com/org",
                project="proj",
                headers={"Authorization": "Bearer x"},
                definition_id=42,
                baseline_branch="main",
                include_entra_update=False,
                dry_run=False,
                update_assignments=True,
                remove_unmanaged=False,
                max_workers=10,
                exclude_csv="",
                restore_mode="selective",
                restore_paths_csv="tenant-state/intune/Device Configurations/macOS - WiFi TEST.json",
            )

        body = captured["body"]
        self.assertIsInstance(body, dict)
        template = body["templateParameters"]
        self.assertEqual(template["restoreMode"], "selective")
        self.assertIn("restorePathsCsv", template)


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