def _do_autopart(storage,
scheme,
requests,
encrypted=False,
luks_fmt_args=None):
"""Perform automatic partitioning.
:param storage: an instance of Blivet
:param scheme: a type of the partitioning scheme
:param requests: list of partitioning requests
:param encrypted: encrypt the scheduled partitions
:param luks_fmt_args: arguments for the LUKS format constructor
"""
log.debug("scheme: %s", scheme)
log.debug("requests:\n%s", "".join([str(p) for p in requests]))
log.debug("encrypted: %s", encrypted)
log.debug("storage.disks: %s", [d.name for d in storage.disks])
log.debug("storage.partitioned: %s",
[d.name for d in storage.partitioned if d.format.supported])
log.debug("all names: %s", [d.name for d in storage.devices])
log.debug("boot disk: %s",
getattr(storage.bootloader.stage1_disk, "name", None))
disks = get_candidate_disks(storage)
log.debug("candidate disks: %s", [d.name for d in disks])
# Schedule implicit partitions.
extra_disks = get_disks_for_implicit_partitions(
disks, scheme, requests)
devs = schedule_implicit_partitions(storage, extra_disks, scheme,
encrypted, luks_fmt_args)
# Schedule requested partitions.
devs = schedule_partitions(storage, disks, devs, scheme, requests,
encrypted, luks_fmt_args)
# run the autopart function to allocate and grow partitions
do_partitioning(storage, boot_disk=storage.bootloader.stage1_disk)
schedule_volumes(storage, devs, scheme, requests, encrypted)
# grow LVs
grow_lvm(storage)
# only newly added swaps should appear in the fstab
new_swaps = (dev for dev in storage.swaps if not dev.format.exists)
storage.set_fstab_swaps(new_swaps)
def test_get_disks_for_implicit_partitions(self):
"""Test the get_disks_for_implicit_partitions function."""
# The /boot partition always requires a slot.
requests = [
PartSpec(mountpoint="/boot", size=Size("1GiB")),
PartSpec(mountpoint="/",
size=Size("2GiB"),
max_size=Size("15GiB"),
grow=True,
btr=True,
lv=True,
thin=True,
encrypted=True),
PartSpec(fstype="swap", grow=False, lv=True, encrypted=True)
]
# No implicit partitions to schedule.
disk_1 = Mock()
disk_2 = Mock()
parted_disk_1 = disk_1.format.parted_disk
parted_disk_2 = disk_2.format.parted_disk
assert get_disks_for_implicit_partitions(
scheme=AUTOPART_TYPE_PLAIN,
disks=[disk_1, disk_2],
requests=requests
) == \
[]
# Extended partitions are supported by the first disk.
parted_disk_1.supportsFeature.return_value = True
parted_disk_1.maxPrimaryPartitionCount = 3
parted_disk_1.primaryPartitionCount = 3
parted_disk_2.supportsFeature.return_value = False
parted_disk_2.maxPrimaryPartitionCount = 3
parted_disk_2.primaryPartitionCount = 2
assert get_disks_for_implicit_partitions(
scheme=AUTOPART_TYPE_LVM_THINP,
disks=[disk_1, disk_2],
requests=requests
) == \
[disk_1, disk_2]
# Extended partitions are not supported by the first disk.
parted_disk_1.supportsFeature.return_value = False
parted_disk_1.maxPrimaryPartitionCount = 3
parted_disk_1.primaryPartitionCount = 2
assert get_disks_for_implicit_partitions(
scheme=AUTOPART_TYPE_LVM_THINP,
disks=[disk_1, disk_2],
requests=requests
) == \
[disk_2]
# Not empty slots for implicit partitions.
parted_disk_1.supportsFeature.return_value = False
parted_disk_1.maxPrimaryPartitionCount = 3
parted_disk_1.primaryPartitionCount = 3
assert get_disks_for_implicit_partitions(
scheme=AUTOPART_TYPE_LVM_THINP,
disks=[disk_1, disk_2],
requests=requests
) == \
[]