1 files changed, 67 insertions, 0 deletions
diff --git a/scripts/testing/tests/test_boot.py b/scripts/testing/tests/test_boot.py
new file mode 100644
index 0000000..78b4404
--- /dev/null
+++ b/scripts/testing/tests/test_boot.py
@@ -0,0 +1,67 @@
+# SPDX-License-Identifier: GPL-3.0-or-later
+"""Post-install checks: boot, initramfs, and filesystem config.
+
+Parity port of validate_zfs_config, validate_boot_config,
+validate_mkinitcpio_hooks, validate_initramfs_consolefont, validate_nvme_module.
+Filesystem/hardware-specific checks are gated on fixtures.
+"""
+
+import pytest
+
+
+@pytest.mark.attribution("archsetup")
+def test_bootloader_installed(host, zfs_root):
+    # A ZFS root boots via ZFSBootMenu (archangel installs the EFI binary under
+    # /efi/EFI/ZBM), so there is no GRUB; a non-ZFS root uses GRUB.
+    if zfs_root:
+        assert host.file("/efi/EFI/ZBM/zfsbootmenu.efi").exists, \
+            "ZFS root must have the ZFSBootMenu EFI binary"
+    else:
+        assert host.file("/boot/grub/grub.cfg").exists, \
+            "non-ZFS root must have a GRUB config"
+
+
+@pytest.mark.attribution("archsetup")
+def test_mkinitcpio_hooks(host, zfs_root):
+    hooks = host.run("grep '^HOOKS=' /etc/mkinitcpio.conf").stdout
+    if zfs_root:
+        # ZFS must use the udev hook; the systemd hook breaks a ZFS boot.
+        assert " udev" in hooks or "(udev" in hooks, "ZFS root must use the udev hook"
+        assert "systemd" not in hooks, "ZFS root must not use the systemd hook"
+    else:
+        # Non-ZFS: either hook is acceptable.
+        assert ("systemd" in hooks) or ("udev" in hooks)
+
+
+@pytest.mark.attribution("archsetup")
+def test_console_font_configured(host, zfs_root):
+    # archsetup sets FONT=ter-132n in /etc/vconsole.conf on every run.
+    assert host.file("/etc/vconsole.conf").contains("^FONT=ter-132n"), \
+        "archsetup should set FONT=ter-132n in /etc/vconsole.conf"
+    # On non-ZFS it also rebuilds the initramfs (mkinitcpio -P) so the font is
+    # baked in for early boot. On ZFS that rebuild is skipped (the busybox ZFS
+    # hook is incompatible with the systemd-hook switch), so the font applies at
+    # the vconsole layer once userspace starts, not inside the initramfs.
+    if zfs_root:
+        return
+    # Pick the main initramfs (this fleet runs linux-lts, so the name is
+    # initramfs-linux-lts.img, not initramfs-linux.img); skip the fallback image.
+    img = host.run(
+        "ls /boot/initramfs-*.img 2>/dev/null | grep -v fallback | head -1"
+    ).stdout.strip()
+    assert img, "no initramfs image found under /boot"
+    out = host.run("lsinitcpio %s 2>/dev/null | grep -cE 'consolefont.psf|ter-'" % img)
+    assert int((out.stdout.strip() or "0")) > 0, "console font not found in %s" % img
+
+
+def test_nvme_module_when_nvme_present(host, has_nvme):
+    if not has_nvme:
+        pytest.skip("no NVMe device present")
+    modules = host.run("grep '^MODULES=' /etc/mkinitcpio.conf").stdout
+    assert "nvme" in modules, "NVMe system should list nvme in mkinitcpio MODULES"
+
+
+def test_zfs_has_sanoid(host):
+    if not host.exists("zfs"):
+        pytest.skip("ZFS not installed (non-ZFS system)")
+    assert host.exists("sanoid"), "ZFS system should have sanoid installed"