130 lines
3.9 KiB
Python
130 lines
3.9 KiB
Python
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, Interface, OperatingSystem
|
|
from mdb.pxe import get_pxe_filename, render_pxelinux_cfg
|
|
|
|
|
|
@pytest.fixture
|
|
def db_with_fixtures(db):
|
|
call_command("loaddata", "test_data", verbosity=0)
|
|
|
|
|
|
@pytest.fixture
|
|
def host(db_with_fixtures):
|
|
return Host.objects.first()
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_render_pxeconfig(host):
|
|
interface = host.interface_set.exclude(ip4address__isnull=True).first()
|
|
config = render_pxelinux_cfg(interface, host)
|
|
|
|
assert settings.MDB_PXE_INITRD in config
|
|
assert settings.MDB_PXE_KERNEL in config
|
|
assert settings.MDB_PXE_PRESEED_URL in config
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_render_pxeconfig_with_os_pxe_config(host):
|
|
_os = OperatingSystem.objects.get(pk=2) # OS with all pxe vars set
|
|
Host.objects.filter(pk=host.pk).update(operating_system=_os)
|
|
host = Host.objects.first()
|
|
|
|
interface = host.interface_set.exclude(ip4address__isnull=True).first()
|
|
config = render_pxelinux_cfg(interface, host)
|
|
|
|
assert _os.pxe_initrd in config
|
|
assert _os.pxe_kernel in config
|
|
assert _os.pxe_preseed_config_url in config
|
|
|
|
|
|
@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
|