refactor(test): set Microvm.uffd_handler from restore_from_snapshot

It's a bit weird to have this attribute set from outside the microvm
class.

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

Author: roypat

tests/framework/microvm.py

@@ -986,7 +986,7 @@ def restore_from_snapshot(
         jailed_snapshot = snapshot.copy_to_chroot(Path(self.chroot()))
 
         if uffd_handler_name:
-            spawn_pf_handler(
+            self.uffd_handler = spawn_pf_handler(
                 self,
                 uffd_handler(uffd_handler_name, binary_dir=self.fc_binary_path.parent),
                 jailed_snapshot,

tests/framework/utils_uffd.py

@@ -94,7 +94,6 @@ def spawn_pf_handler(vm, handler_path, jailed_snapshot):
         handler_name, SOCKET_PATH, jailed_snapshot, vm.chroot(), "uffd.log"
     )
     uffd_handler.spawn(vm.jailer.uid, vm.jailer.gid)
-    vm.uffd_handler = uffd_handler
 
     return uffd_handler