test-framework: add detection for partitions on devices.

Signed-off-by: Kamil Gierszewski <kamil.gierszewski@huawei.com>

Author: Kamil Gierszewski

storage_devices/disk.py

@@ -1,6 +1,6 @@
 #
 # Copyright(c) 2019-2022 Intel Corporation
-# Copyright(c) 2024 Huawei Technologies Co., Ltd.
+# Copyright(c) 2024-2025 Huawei Technologies Co., Ltd.
 # SPDX-License-Identifier: BSD-3-Clause
 #
 
@@ -118,7 +118,7 @@ def __init__(
         self.serial_number = serial_number
         self.block_size = Unit(block_size)
         self.device_id = self.get_device_id()
-        self.partitions = []
+        self.partitions = self.discover_partitions()
         self.pci_address = None
 
     @classmethod
@@ -139,6 +139,9 @@ def resolve_type(cls, disk_path):
             )
         return recognized_types[0]
 
+    def discover_partitions(self) -> list:
+        return disk_tools.discover_partition(self)
+
     def create_partitions(self, sizes: [], partition_table_type=PartitionTable.gpt):
         disk_tools.create_partitions(self, sizes, partition_table_type)
 
@@ -154,7 +157,7 @@ def umount_all_partitions(self):
 
     def remove_partitions(self):
         for part in self.partitions:
-            if is_mounted(part.path):
+            if is_mounted(part.device_id):
                 part.unmount()
         if disk_tools.remove_partitions(self):
             self.partitions.clear()

storage_devices/partition.py

@@ -1,10 +1,12 @@
 #
 # Copyright(c) 2019-2021 Intel Corporation
+# Copyright(c) 2025 Huawei Technologies Co., Ltd.
 # SPDX-License-Identifier: BSD-3-Clause
 #
 
 from storage_devices.device import Device
 from test_tools.disk_tools import get_partition_path
+from test_tools.fs_tools import readlink
 from type_def.size import Size
 
 
@@ -16,7 +18,11 @@ def __init__(self, parent_dev, type, number, begin: Size, end: Size):
         self.type = type
         self.begin = begin
         self.end = end
+        self.device_id = self.get_device_id()
 
     def __str__(self):
         return f"\tsystem path: {self.path}, size: {self.size}, type: {self.type}, " \
             f"parent device: {self.parent_device.path}\n"
+
+    def get_device_id(self):
+        return readlink(self.path).split('/')[-1]

test_tools/disk_tools.py

@@ -57,6 +57,57 @@ def get_partition_path(parent_dev, number):
     return f'{parent_dev}{id_separator}{number}'
 
 
+def discover_partition(device) -> list:
+    cmd = (
+        f"lsblk -ln -o NAME,TYPE {device.path} "
+        + r"""| awk '$2 == "part" { print "/dev/" $1 }'"""
+    )
+
+    output = TestRun.executor.run(cmd).stdout
+    partition_list = [line for line in output.split("\n") if line]
+
+    # needs to be placed here (circular dependency)
+    from storage_devices.partition import Partition
+    partitions_on_device_list = []
+
+    # the only tool "that I found out" that returns a value for the partition number field when the
+    # partition is mounted is udevadm.
+    for partition in partition_list:
+        cmd = f"lsblk -n --bytes -o START,SIZE,LOG-SEC {partition}"
+        output = TestRun.executor.run(cmd)
+        if output.exit_code != 0:
+            TestRun.LOGGER.warning(
+                "Warning: Partition detection failed. "
+                "The issue might be caused by an outdated version of `lsblk`. "
+                "Please ensure you are using a version that supports the required "
+                "features (version 2.34 or higher).")
+            return []
+
+        cmd = f"udevadm info {partition} " + "-q property | grep PARTN= | awk -F= '{print $2}'"
+        partition_number = int(TestRun.executor.run(cmd).stdout)
+
+        partition_start, partition_size, partition_leg_sec = output.stdout.split()
+        part_number = int(partition_number)
+        begin = Size(int(partition_start), Unit.Byte)
+        end = Size(
+            (
+                (int(partition_size) / int(partition_leg_sec) + int(partition_start) - 1)
+            ),
+            Unit.Byte,
+        )
+        partitions_on_device_list.append(
+            Partition(
+                parent_dev=device,
+                type=PartitionType.logical,
+                number=part_number,
+                begin=begin,
+                end=end,
+            )
+        )
+
+    return partitions_on_device_list
+
+
 def remove_parition(device, part_number):
     TestRun.LOGGER.info(f"Removing part {part_number} from {device.path}")
     cmd = f'parted --script {device.path} rm {part_number}'