test(pytest): bootstrap function test

Author: tai271828

.gitignore

@@ -0,0 +1,2 @@
+__pycache__
+dist

README.md

@@ -47,7 +47,8 @@ Today `ubuntu-server-netboot` needs to run on Ubuntu or another Debian derivativ
 
 This script is tested with Ubuntu 18.04 ("bionic beaver") and above.
 
-## Contribution
+## Contribution and Development
 
 Please report bugs to [this github issue tracker](https://github.com/dannf/ubuntu-server-netboot/issues). The github templates including "Issue" and "Pull requests" are originally forked from [this "cookiecutter" templates for python](https://github.com/Lee-W/cookiecutter-python-template).
 
+Place `pytest` to cover the basic test sets.

conftest.py

@@ -0,0 +1,5 @@
+import sys
+
+
+# we would like to use the distro-info provided by system
+sys.path.append("/usr/lib/python3/dist-packages")

poetry.lock

@@ -8,6 +8,7 @@ authors = ["dann frazier <[email protected]>", "Taihsiang Ho (tai271828) <taihsia
 python = "^3.6.5"
 
 [tool.poetry.dev-dependencies]
+pytest = "^6.2.4"
 
 [tool.poetry.scripts]
 ubuntu-server-netboot = "usn.usn:ubuntu_server_netboot"

pyproject.toml

@@ -0,0 +1,29 @@
+set menu_color_normal=white/black
+set menu_color_highlight=black/light-gray
+if background_color 44,0,30,0; then
+  clear
+fi
+
+insmod gzio
+
+set timeout=30
+menuentry "Install Ubuntu Server" {
+	set gfxpayload=keep
+	linux	/casper/vmlinuz   quiet splash root=/dev/ram0 ramdisk_size=1500000 ip=dhcp url=http://cdimage.ubuntu.com/ubuntu/releases/20.04.2/release/ubuntu-20.04.2-live-server-arm64.iso autoinstall "ds=nocloud-net;s=http://12.34.56.78/" ---
+	initrd	/casper/initrd
+}
+menuentry 'Boot from next volume' {
+	exit 1
+}
+menuentry 'UEFI Firmware Settings' {
+	fwsetup
+}
+submenu 'Boot and Install with the HWE kernel' {
+menuentry "Install Ubuntu Server" {
+	set gfxpayload=keep
+	linux	/casper/hwe-vmlinuz   quiet splash root=/dev/ram0 ramdisk_size=1500000 ip=dhcp url=http://cdimage.ubuntu.com/ubuntu/releases/20.04.2/release/ubuntu-20.04.2-live-server-arm64.iso autoinstall "ds=nocloud-net;s=http://12.34.56.78/" ---
+	initrd	/casper/hwe-initrd
+}
+}
+
+

usn/tests/__init__.py

@@ -0,0 +1,27 @@
+set menu_color_normal=white/black
+set menu_color_highlight=black/light-gray
+if background_color 44,0,30,0; then
+  clear
+fi
+
+insmod gzio
+
+set timeout=30
+menuentry "Install Ubuntu Server" {
+	set gfxpayload=keep
+	linux	/casper/vmlinuz   quiet splash ---
+	initrd	/casper/initrd
+}
+menuentry 'Boot from next volume' {
+	exit 1
+}
+menuentry 'UEFI Firmware Settings' {
+	fwsetup
+}
+submenu 'Boot and Install with the HWE kernel' {
+menuentry "Install Ubuntu Server" {
+	set gfxpayload=keep
+	linux	/casper/hwe-vmlinuz   quiet splash ---
+	initrd	/casper/hwe-initrd
+}
+}

usn/tests/data/grub_expected.cfg

@@ -0,0 +1,50 @@
+import pytest
+
+from pkg_resources import resource_filename
+from usn import ubuntu_server_netboot
+
+
+class TestGrubCfg:
+    @pytest.fixture(scope="class")
+    def grub_cfg_orig_filename(self):
+        resource = "tests/data/grub_ubuntu-20.04.2-live-server-arm64.cfg"
+        grub_cfg_orig_filename = resource_filename("usn", resource)
+
+        return grub_cfg_orig_filename
+
+    @pytest.fixture(scope="class")
+    def grub_cfg_expected_filename(self):
+        resource = "tests/data/grub_expected.cfg"
+        grub_cfg_expected_filename = resource_filename("usn", resource)
+
+        return grub_cfg_expected_filename
+
+    def test_grub_cfg_ubuntu_20_4_2_live_server_arm64(
+        self, grub_cfg_orig_filename, grub_cfg_expected_filename
+    ):
+        """
+        Check the generated grub.cfg content
+
+        Compare the grub.cfg generated with the original grub.cfg extracted from the iso ubuntu 20.04.2 live server
+        arm64.
+        """
+        with open(grub_cfg_orig_filename, "r", encoding="utf-8") as grub_cfg_orig_f:
+            bootloader_cfg = ubuntu_server_netboot.GrubConfig(grub_cfg_orig_f.read())
+
+            url = "http://cdimage.ubuntu.com/ubuntu/releases/20.04.2/release/ubuntu-20.04.2-live-server-arm64.iso"
+            autoinstall_url = "http://12.34.56.78/"
+            extra_args = ""
+            ubuntu_server_netboot.setup_kernel_params(
+                bootloader_cfg, url, autoinstall_url, extra_args
+            )
+            cfg_modified = bootloader_cfg.cfg
+
+        with open(
+            grub_cfg_expected_filename, "r", encoding="utf-8"
+        ) as grub_cfg_expected_f:
+            bootloader_cfg_expected = ubuntu_server_netboot.GrubConfig(
+                grub_cfg_expected_f.read()
+            )
+            cfg_expected = bootloader_cfg_expected.cfg
+
+        assert cfg_modified == cfg_expected