Update storage.py

microsoft · Nov 20, 2024 · 1df3cac · 1df3cac
1 parent e01fe43
commit 1df3cac
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/microsoft/testsuites/core/storage.py b/microsoft/testsuites/core/storage.py
@@ -321,11 +321,13 @@ def verify_nvme_disk_controller_type(self, node: RemoteNode) -> None:
     )
     def verify_os_partition_identifier(self, log: Logger, node: RemoteNode) -> None:
         # get information of root disk from blkid
-        os_partition = (
-            node.features[Disk]
-            .get_partition_with_mount_point(self.os_disk_mount_point)
-            .name
+        os_disk_partition = node.features[Disk].get_partition_with_mount_point(
+            self.os_disk_mount_point
         )
+        if not os_disk_partition:
+            raise LisaException("Failed to get os disk partition")
+
+        os_partition = os_disk_partition.name
         os_partition_info = node.tools[Blkid].get_partition_info_by_name(os_partition)
 
         # check if cvm