Ask for the region on the first screen and hand it to target setup

test/unit/test_region.py

@@ -0,0 +1,77 @@
+"""The configurator's region reaches the target setup as OMARCHY_REGION."""
+
+import json
+import os
+import sys
+import tempfile
+import types
+import unittest
+from pathlib import Path
+from unittest import mock
+
+ROOT = Path(__file__).resolve().parents[2]
+sys.path.insert(0, str(ROOT / "configs/airootfs/usr/share/omarchy-iso"))
+sys.modules.setdefault(
+    "orchestrator.archinstall_adapter",
+    types.ModuleType("orchestrator.archinstall_adapter"),
+)
+from orchestrator import context, phases_impl  # noqa: E402
+
+
+def load_context(root: Path, omarchy_install: dict | None) -> context.InstallContext:
+    config = {"kernels": ["linux-omarchy"]}
+    if omarchy_install is not None:
+        config["omarchy_install"] = omarchy_install
+    config_path = root / "config.json"
+    creds_path = root / "creds.json"
+    config_path.write_text(json.dumps(config))
+    creds_path.write_text(json.dumps({"users": [{"username": "test"}]}))
+    with mock.patch.dict(os.environ, {
+        "OMARCHY_INSTALL_CONFIG": str(config_path),
+        "OMARCHY_INSTALL_CREDS": str(creds_path),
+        "OMARCHY_INSTALL_STATE_DIR": str(root / "state"),
+    }, clear=True):
+        return context.InstallContext.from_env()
+
+
+class RegionTest(unittest.TestCase):
+    def test_context_reads_region(self):
+        for omarchy_install, expected in [
+            (None, "global"),
+            ({"mode": "full_disk"}, "global"),
+            ({"mode": "full_disk", "region": "global"}, "global"),
+            ({"mode": "full_disk", "region": "china"}, "china"),
+        ]:
+            with self.subTest(omarchy_install=omarchy_install), tempfile.TemporaryDirectory() as tmp:
+                ctx = load_context(Path(tmp), omarchy_install)
+                self.assertEqual(ctx.omarchy_install["region"], expected)
+
+    def test_context_rejects_unknown_region(self):
+        for region in ("China", "mars", ""):
+            with self.subTest(region=region), tempfile.TemporaryDirectory() as tmp:
+                with self.assertRaisesRegex(RuntimeError, "omarchy_install.region must be global or china"):
+                    load_context(Path(tmp), {"mode": "full_disk", "region": region})
+
+    def test_target_setup_receives_region(self):
+        for region in ("global", "china"):
+            with self.subTest(region=region), tempfile.TemporaryDirectory() as tmp:
+                root = Path(tmp)
+                ctx = types.SimpleNamespace(
+                    target=root / "mnt",
+                    log_path=root / "install.log",
+                    omarchy_install={"region": region},
+                    username="test",
+                    full_name="",
+                    email="",
+                    state={"omarchy_start_time": "now", "omarchy_start_epoch": 0,
+                           "omarchy_finalizer_header_written": True},
+                )
+                with mock.patch.object(phases_impl, "_prepare_target_setup"), \
+                        mock.patch.object(phases_impl.subprocess, "run") as run:
+                    phases_impl._run_target_setup_command(ctx, ["/usr/bin/omarchy-apply-system"])
+                chroot = next(call.args[0] for call in run.call_args_list if call.args[0][0] == "arch-chroot")
+                self.assertIn(f"OMARCHY_REGION={region}", chroot)
+
+
+if __name__ == "__main__":
+    unittest.main()