feat(virtio-pmem): adjust integration tests with new config options

virtio-pmem introduces new field in config structure. Adjust
tests to expect it.

Signed-off-by: Egor Lazarchuk <[email protected]>

Author: ShadowCurse

tests/framework/microvm.py

@@ -989,7 +989,7 @@ def add_pmem(
             root_device=root_device,
             read_only=read_only,
         )
-        self.disks[id] = path_on_host
+        self.disks[pmem_id] = path_on_host
 
     def start(self):
         """Start the microvm.

tests/framework/vm_config.json

@@ -31,5 +31,6 @@
   "logger": null,
   "metrics": null,
   "mmds-config": null,
-  "entropy": null
+  "entropy": null,
+  "pmem": []
 }

tests/integration_tests/functional/test_api.py

@@ -1146,6 +1146,21 @@ def test_get_full_config_after_restoring_snapshot(microvm_factory, uvm_nano):
         }
     ]
 
+    uvm_nano.api.pmem.put(
+        id="pmem",
+        path_on_host="/" + uvm_nano.rootfs_file.name,
+        root_device=False,
+        read_only=False,
+    )
+    setup_cfg["pmem"] = [
+        {
+            "id": "pmem",
+            "path_on_host": "/" + uvm_nano.rootfs_file.name,
+            "root_device": False,
+            "read_only": False,
+        }
+    ]
+
     # Add a memory balloon device.
     uvm_nano.api.balloon.put(amount_mib=1, deflate_on_oom=True)
     setup_cfg["balloon"] = {
@@ -1257,6 +1272,21 @@ def test_get_full_config(uvm_plain):
         }
     ]
 
+    test_microvm.api.pmem.put(
+        id="pmem",
+        path_on_host="/" + test_microvm.rootfs_file.name,
+        root_device=False,
+        read_only=False,
+    )
+    expected_cfg["pmem"] = [
+        {
+            "id": "pmem",
+            "path_on_host": "/" + test_microvm.rootfs_file.name,
+            "root_device": False,
+            "read_only": False,
+        }
+    ]
+
     # Add a memory balloon device.
     test_microvm.api.balloon.put(amount_mib=1, deflate_on_oom=True)
     expected_cfg["balloon"] = {