feat(cpu-feauter-x86_64): add after snapshot 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.py

@@ -20,6 +20,7 @@
 import framework.utils_cpuid as cpuid_utils
 from framework import utils
 from framework.defs import SUPPORTED_HOST_KERNELS
+from framework.microvm import SnapshotType
 from framework.properties import global_props
 from framework.utils_cpu_templates import SUPPORTED_CPU_TEMPLATES
 
@@ -628,7 +629,7 @@ def test_cpu_cpuid_restore(microvm_factory, guest_kernel, msr_cpu_template):
     PLATFORM != "x86_64", reason="CPU features are masked only on x86_64."
 )
 @pytest.mark.parametrize("cpu_template", ["T2", "T2S", "C3"])
-def test_cpu_template(test_microvm_with_api, cpu_template):
+def test_cpu_template(test_microvm_with_api, cpu_template, microvm_factory):
     """
     Test masked and enabled cpu features against the expected template.
 
@@ -657,6 +658,13 @@ def test_cpu_template(test_microvm_with_api, cpu_template):
     check_masked_features(test_microvm, cpu_template)
     check_enabled_features(test_microvm, cpu_template)
 
+    snapshot = test_microvm.make_snapshot(SnapshotType.FULL)
+    restored_vm = microvm_factory.build()
+    restored_vm.spawn()
+    restored_vm.restore_from_snapshot(snapshot, resume=True)
+    check_masked_features(restored_vm, cpu_template)
+    check_enabled_features(restored_vm, cpu_template)
+
 
 def check_masked_features(test_microvm, cpu_template):
     """Verify the masked features of the given template."""