Add a way to simply restore a snapshot in tests

Signed-off-by: Pablo Barbáchano <[email protected]>

Author: pb8o

tests/framework/microvm.py

@@ -856,6 +856,36 @@ def pause_to_snapshot(
             response.status_code
         ), response.text
 
+    def restore_from_snapshot(
+        self,
+        *,
+        snapshot_mem: Path,
+        snapshot_vmstate: Path,
+        snapshot_disks: list[Path],
+        snapshot_is_diff: bool = False,
+    ):
+        """
+        Restores a snapshot, and resumes the microvm
+        """
+
+        # Hardlink all the snapshot files into the microvm jail.
+        jailed_mem = self.create_jailed_resource(snapshot_mem)
+        jailed_vmstate = self.create_jailed_resource(snapshot_vmstate)
+
+        assert len(snapshot_disks) > 0, "Snapshot requires at least one disk."
+        jailed_disks = []
+        for disk in snapshot_disks:
+            jailed_disks.append(self.create_jailed_resource(disk))
+
+        response = self.snapshot.load(
+            mem_file_path=jailed_mem,
+            snapshot_path=jailed_vmstate,
+            diff=snapshot_is_diff,
+            resume=True,
+        )
+        assert response.ok
+        return True
+
     def start_console_logger(self, log_fifo):
         """
         Start a thread that monitors the microVM console.