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

import macos_runtime


class MacRuntimeTests(unittest.TestCase):
    def inspect_fixture(self, dependency=None, sections=1, arches="arm64"):
        with tempfile.TemporaryDirectory(prefix="mac metadata ") as directory:
            worker = Path(directory) / "searchi-gemma-worker"
            worker.write_bytes(b"\xcf\xfa\xed\xfe" + b"metadata fixture, not executable")
            libraries = f"{worker}:\n\t/usr/lib/libSystem.B.dylib (compatibility version 1.0.0, current version 100.0.0)\n"
            libraries += "\t" + (dependency or "/System/Library/Frameworks/Metal.framework/Versions/A/Metal") + " (compatibility version 1.0.0, current version 100.0.0)\n"
            loads = "sectname __ggml_metallib\nsegname __DATA\naddr 0x1000\nsize 0x100\n" * sections
            with patch.object(macos_runtime.subprocess, "check_output", side_effect=[arches, libraries, loads]):
                return macos_runtime.inspect(worker)

    def test_static_system_metal_inventory_has_no_packaged_dylibs(self):
        result = self.inspect_fixture()
        self.assertEqual(result["architectures"], ["arm64"])
        self.assertEqual(result["libraries"], [])
        self.assertTrue(result["metal_embedded"])

    def test_external_library_and_missing_metal_are_rejected(self):
        for dependency in ("/private/build path/libggml.dylib", "/usr/lib/libc++.1.dylib"):
            with self.subTest(dependency=dependency), self.assertRaises(RuntimeError):
                self.inspect_fixture(dependency=dependency)

    def test_all_architectures_require_embedded_sources(self):
        with self.assertRaisesRegex(RuntimeError, "Each Mach-O architecture"):
            self.inspect_fixture(sections=0)
        with self.assertRaisesRegex(RuntimeError, "Each Mach-O architecture"):
            self.inspect_fixture(sections=1, arches="arm64 x86_64")
        self.assertEqual(self.inspect_fixture(sections=2, arches="arm64 x86_64")["architectures"], ["arm64", "x86_64"])

    def test_malformed_inventory_and_foreign_artifact_fail(self):
        with tempfile.TemporaryDirectory() as directory:
            worker = Path(directory) / "worker"
            worker.write_bytes(b"\xcf\xfa\xed\xfe")
            with self.assertRaises(RuntimeError):
                macos_runtime.validate_inventory({"architectures": [["arm64"]]}, worker)
            worker.write_bytes(b"MZxx")
            with self.assertRaisesRegex(RuntimeError, "Mach-O"):
                macos_runtime.validate_inventory({}, worker)


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