@@ -142,7 +142,12 @@ def test_restore_latency(
142142@pytest .mark .parametrize ("uffd_handler" , [None , "on_demand" , "fault_all" ])
143143@pytest .mark .parametrize ("huge_pages" , HugePagesConfig )
144144def test_post_restore_latency (
145- microvm_factory , rootfs , guest_kernel_linux_5_10 , metrics , uffd_handler , huge_pages
145+ microvm_factory ,
146+ rootfs ,
147+ guest_kernel_linux_5_10 ,
148+ metrics ,
149+ uffd_handler ,
150+ huge_pages ,
146151):
147152 """Collects latency metric of post-restore memory accesses done inside the guest"""
148153 if huge_pages != HugePagesConfig .NONE and uffd_handler is None :
@@ -190,11 +195,20 @@ def test_post_restore_latency(
190195
191196@pytest .mark .nonci
192197@pytest .mark .parametrize ("huge_pages" , HugePagesConfig )
198+ @pytest .mark .parametrize (
199+ ("vcpus" , "mem" ), [(1 , 128 ), (1 , 1024 ), (2 , 2048 ), (3 , 4096 ), (4 , 6144 )]
200+ )
193201def test_population_latency (
194- microvm_factory , rootfs , guest_kernel_linux_5_10 , metrics , huge_pages
202+ microvm_factory ,
203+ rootfs ,
204+ guest_kernel_linux_5_10 ,
205+ metrics ,
206+ huge_pages ,
207+ vcpus ,
208+ mem ,
195209):
196210 """Collects population latency metrics (e.g. how long it takes UFFD handler to fault in all memory)"""
197- test_setup = SnapshotRestoreTest (mem = 128 , vcpus = 1 , huge_pages = huge_pages )
211+ test_setup = SnapshotRestoreTest (mem = mem , vcpus = vcpus , huge_pages = huge_pages )
198212 vm = test_setup .boot_vm (microvm_factory , guest_kernel_linux_5_10 , rootfs )
199213
200214 metrics .set_dimensions (
0 commit comments