feat(cpu-features-aarch64): add after snaphot test

Add verification of cpu features after loading a snapshot
of a vm with cpu-template applied.

Signed-off-by: Egor Lazarchuk <yegorlz@amazon.co.uk>

Author: ShadowCurse

tests/integration_tests/functional/test_cpu_features_aarch64.py

@@ -8,6 +8,7 @@
 import pytest
 
 import framework.utils_cpuid as cpuid_utils
+from framework.microvm import SnapshotType
 from framework.utils_cpuid import CpuModel
 
 PLATFORM = platform.machine()
@@ -108,6 +109,12 @@ def test_cpu_features_with_static_template(
     guest_kv = re.search(r"vmlinux-(\d+\.\d+)", guest_kernel.name).group(1)
     _check_cpu_features_arm(vm, guest_kv, "v1n1")
 
+    snapshot = vm.make_snapshot(SnapshotType.FULL)
+    restored_vm = microvm_factory.build()
+    restored_vm.spawn()
+    restored_vm.restore_from_snapshot(snapshot, resume=True)
+    _check_cpu_features_arm(restored_vm, guest_kv, "v1n1")
+
 
 @pytest.mark.skipif(
     PLATFORM != "aarch64",
@@ -128,3 +135,9 @@ def test_cpu_features_with_custom_template(
     vm.start()
     guest_kv = re.search(r"vmlinux-(\d+\.\d+)", guest_kernel.name).group(1)
     _check_cpu_features_arm(vm, guest_kv, custom_cpu_template["name"])
+
+    snapshot = vm.make_snapshot(SnapshotType.FULL)
+    restored_vm = microvm_factory.build()
+    restored_vm.spawn()
+    restored_vm.restore_from_snapshot(snapshot, resume=True)
+    _check_cpu_features_arm(restored_vm, guest_kv, custom_cpu_template["name"])