# 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_grub_config_exists(host):
    assert host.file("/boot/grub/grub.cfg").exists


@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_in_initramfs(host):
    # 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"