Prepare inspected static Metal runtime inventory for macOS staging

tools/embeddinggemma2/test_macos_runtime.py

@@ -0,0 +1,50 @@
+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()