test: cover the generated configs, the PXE API and PXE files

tests/test_pxe.py

@@ -1,9 +1,11 @@
 import pytest
 from django.conf import settings
+from django.contrib import messages
 from django.core.management import call_command
+from django.urls import reverse
 
-from mdb.models import Host, OperatingSystem
-from mdb.pxe import render_pxelinux_cfg
+from mdb.models import Host, Interface, OperatingSystem
+from mdb.pxe import get_pxe_filename, render_pxelinux_cfg
 
 
 @pytest.fixture
@@ -43,3 +45,85 @@ def test_render_pxeconfig_with_os_pxe_config(host):
 @pytest.mark.django_db
 def test_host_as_pxe_files(host):
     assert len(host.as_pxe_files()) == 1
+
+
+PXE_FILE = "01-f0-de-f1-58-be-1a"  # from the fixture interface's MAC address
+
+
+@pytest.fixture
+def tftp_root(settings, tmp_path):
+    settings.MDB_PXE_TFTP_ROOT = str(tmp_path)
+    return tmp_path
+
+
+def test_saving_installable_host_writes_pxe_file(host, tftp_root):
+    Interface.objects.update(pxe_filename="", dhcp_client=False)
+    host.save()
+
+    path = tftp_root / PXE_FILE
+    assert host.pxe_key in path.read_text()
+    assert path.stat().st_mode & 0o777 == 0o644
+    # The interface is set up to boot from the network.
+    interface = host.interface_set.get()
+    assert interface.pxe_filename == "pxelinux.0"
+    assert interface.dhcp_client
+
+
+def test_saving_host_keeps_custom_pxe_filename(host, tftp_root):
+    Interface.objects.update(pxe_filename="custom.efi")
+    host.save()
+    assert host.interface_set.get().pxe_filename == "custom.efi"
+
+
+def test_saving_non_installable_host_deletes_pxe_file(host, tftp_root):
+    (tftp_root / PXE_FILE).write_text("old")
+    host.pxe_installable = False
+    host.save()
+    assert not (tftp_root / PXE_FILE).exists()
+
+
+def test_saving_host_without_key_generates_one(host, tftp_root):
+    host.pxe_key = ""
+    host.save()
+    key = Host.objects.get(pk=host.pk).pxe_key
+    assert len(key) == 32
+    assert key in (tftp_root / PXE_FILE).read_text()
+
+
+def test_pxe_filename_by_ip_address(host):
+    interface = host.interface_set.get()  # 10.0.3.2
+    assert get_pxe_filename(interface, "ip_addr") == "0A000302"
+
+
+def test_admin_marks_hosts_installable(host, tftp_root, admin_client):
+    Host.objects.update(pxe_installable=False)
+
+    res = admin_client.post(
+        reverse("admin:mdb_host_changelist"),
+        {"action": "set_installable", "_selected_action": [host.pk]},
+        follow=True,
+    )
+
+    assert res.status_code == 200
+    assert Host.objects.get(pk=host.pk).pxe_installable
+    assert (tftp_root / PXE_FILE).exists()
+    assert not [m for m in res.context["messages"] if m.level == messages.WARNING]
+
+
+def test_admin_warns_about_interfaces_without_ip(host, tftp_root, admin_client):
+    Host.objects.update(pxe_installable=False)
+    Interface.objects.create(
+        name="eth1", macaddr="00:11:22:33:44:55", host=host, domain_id=1
+    )
+
+    res = admin_client.post(
+        reverse("admin:mdb_host_changelist"),
+        {"action": "set_installable", "_selected_action": [host.pk]},
+        follow=True,
+    )
+
+    warnings = [str(m) for m in res.context["messages"] if m.level == messages.WARNING]
+    assert warnings == [
+        "Host tut might not be installable since interface(s) eth1 has no IP set."
+    ]
+    assert Host.objects.get(pk=host.pk).pxe_installable