feat(virtio-pmem): add boottime tests

Parametrize the boottime tests to also test boottime
from virtio-pmem device.

Signed-off-by: Egor Lazarchuk <[email protected]>

Author: ShadowCurse

tests/integration_tests/performance/test_boottime.py

@@ -95,18 +95,37 @@ def to_ms(v, unit):
 
 
 def launch_vm_with_boot_timer(
-    microvm_factory, guest_kernel_acpi, rootfs_rw, vcpu_count, mem_size_mib, pci_enabled
+    microvm_factory,
+    guest_kernel_acpi,
+    rootfs_rw,
+    vcpu_count,
+    mem_size_mib,
+    pci_enabled,
+    boot_from_pmem,
 ):
     """Launches a microVM with guest-timer and returns the reported metrics for it"""
-    vm = microvm_factory.build(guest_kernel_acpi, rootfs_rw, pci=pci_enabled)
+    vm = microvm_factory.build(
+        guest_kernel_acpi, rootfs_rw, pci=pci_enabled, monitor_memory=False
+    )
     vm.jailer.extra_args.update({"boot-timer": None})
     vm.spawn()
-    vm.basic_config(
-        vcpu_count=vcpu_count,
-        mem_size_mib=mem_size_mib,
-        boot_args=DEFAULT_BOOT_ARGS + " init=/usr/local/bin/init",
-        enable_entropy_device=True,
-    )
+    if not boot_from_pmem:
+        vm.basic_config(
+            vcpu_count=vcpu_count,
+            mem_size_mib=mem_size_mib,
+            boot_args=DEFAULT_BOOT_ARGS + " init=/usr/local/bin/init",
+            enable_entropy_device=True,
+        )
+    else:
+        vm.basic_config(
+            add_root_device=False,
+            vcpu_count=vcpu_count,
+            mem_size_mib=mem_size_mib,
+            boot_args=DEFAULT_BOOT_ARGS + " init=/usr/local/bin/init rootflags=dax",
+            enable_entropy_device=True,
+        )
+        vm.add_pmem("pmem", rootfs_rw, True, True)
+
     vm.add_net_iface()
     vm.start()
     vm.pin_threads(0)
@@ -119,21 +138,23 @@ def launch_vm_with_boot_timer(
 def test_boot_timer(microvm_factory, guest_kernel_acpi, rootfs, pci_enabled):
     """Tests that the boot timer device works"""
     launch_vm_with_boot_timer(
-        microvm_factory, guest_kernel_acpi, rootfs, 1, 128, pci_enabled
+        microvm_factory, guest_kernel_acpi, rootfs, 1, 128, pci_enabled, False
     )
 
 
 @pytest.mark.parametrize(
     "vcpu_count,mem_size_mib",
     [(1, 128), (1, 1024), (2, 2048), (4, 4096)],
 )
+@pytest.mark.parametrize("boot_from_pmem", [True, False], ids=["PmemBoot", "BlockBoot"])
 @pytest.mark.nonci
 def test_boottime(
     microvm_factory,
     guest_kernel_acpi,
     rootfs_rw,
     vcpu_count,
     mem_size_mib,
+    boot_from_pmem,
     pci_enabled,
     metrics,
 ):
@@ -147,12 +168,14 @@ def test_boottime(
             vcpu_count,
             mem_size_mib,
             pci_enabled,
+            boot_from_pmem,
         )
 
         if i == 0:
             metrics.set_dimensions(
                 {
                     "performance_test": "test_boottime",
+                    "boot_from_pmem": str(boot_from_pmem),
                     **vm.dimensions,
                 }
             )