def test_find_root_partition_none():
disk = Disk('disk', '/dev/sda', 20, [
Partition(10, '/dev/sda1', '/my', True, 'ext4', False),
Partition(10, '/dev/sda2', '', True, 'ext4', True)
])
assert disk.find_root_partition() is None
def test_find_root_partition_some():
disk = Disk('disk', '/dev/sda', 20, [
Partition(10, '/dev/sda1', '/', True, 'ext4', False),
Partition(10, '/dev/sda2', '', True, 'ext4', True)
])
assert disk.find_root_partition().device == '/dev/sda1'
def root_partition(self):
disks = self.lsblk.all_disks()
partition = Partition(0, 'unknown', '/', True, 'unknown', False)
boot_disk = next(
(d for d in disks if d.find_root_partition() is not None), None)
if boot_disk:
partition = boot_disk.find_root_partition()
parted_output = parted(boot_disk.device)
partition.extendable = has_unallocated_space_at_the_end(
parted_output)
return partition