test: run snapshot creation performance test with both full and diff

Diff snapshots create sparse files, so they should be a lot faster than
full snapshots.

Signed-off-by: Patrick Roy <[email protected]>

Author: roypat

tests/integration_tests/performance/test_snapshot.py

@@ -11,7 +11,7 @@
 import pytest
 
 import host_tools.drive as drive_tools
-from framework.microvm import HugePagesConfig, Microvm
+from framework.microvm import HugePagesConfig, Microvm, SnapshotType
 
 USEC_IN_MSEC = 1000
 NS_IN_MSEC = 1_000_000
@@ -257,21 +257,30 @@ def test_snapshot_create_latency(
     guest_kernel_linux_5_10,
     rootfs,
     metrics,
+    snapshot_type,
 ):
     """Measure the latency of creating a Full snapshot"""
 
     vm = microvm_factory.build(guest_kernel_linux_5_10, rootfs, monitor_memory=False)
     vm.spawn()
-    vm.basic_config(vcpu_count=2, mem_size_mib=512)
+    vm.basic_config(
+        vcpu_count=2,
+        mem_size_mib=512,
+        track_dirty_pages=snapshot_type == SnapshotType.DIFF,
+    )
     vm.start()
     vm.pin_threads(0)
 
     metrics.set_dimensions(
-        {**vm.dimensions, "performance_test": "test_snapshot_create_latency"}
+        {
+            **vm.dimensions,
+            "performance_test": "test_snapshot_create_latency",
+            "snapshot_type": snapshot_type.value,
+        }
     )
 
     for _ in range(ITERATIONS):
-        vm.snapshot_full()
+        vm.make_snapshot(snapshot_type)
         fc_metrics = vm.flush_metrics()
 
         value = fc_metrics["latencies_us"]["full_create_snapshot"] / USEC_IN_MSEC