def test_target_size(self):
with sparsetmpfile("targetsizetest", Size("10 MiB")) as disk_file:
disk = DiskFile(disk_file)
disk.format = get_format("disklabel", device=disk.path, label_type="msdos")
grain_size = Size(disk.format.alignment.grainSize)
sector_size = Size(disk.format.parted_device.sectorSize)
start = int(grain_size)
orig_size = Size("6 MiB")
end = start + int(orig_size / sector_size) - 1
disk.format.add_partition(start, end)
partition = disk.format.parted_disk.getPartitionBySector(start)
self.assertNotEqual(partition, None)
self.assertEqual(orig_size, Size(partition.getLength(unit='B')))
device = PartitionDevice(os.path.basename(partition.path),
size=orig_size)
device.disk = disk
device.exists = True
device.parted_partition = partition
device.format = get_format("ext4", device=device.path)
device.format.exists = True
# grain size should be 1 MiB
device.format._min_instance_size = Size("2 MiB") + (grain_size / 2)
device.format._resizable = True
# Make sure things are as expected to begin with.
self.assertEqual(device.size, orig_size)
self.assertEqual(device.min_size, Size("3 MiB"))
# start sector's at 1 MiB
self.assertEqual(device.max_size, Size("9 MiB"))
# ValueError if not Size
with six.assertRaisesRegex(self, ValueError,
"new size must.*type Size"):
device.target_size = 22
self.assertEqual(device.target_size, orig_size)
# ValueError if size smaller than min_size
with six.assertRaisesRegex(self, ValueError,
"size.*smaller than the minimum"):
device.target_size = Size("1 MiB")
self.assertEqual(device.target_size, orig_size)
# ValueError if size larger than max_size
with six.assertRaisesRegex(self, ValueError,
"size.*larger than the maximum"):
device.target_size = Size("11 MiB")
self.assertEqual(device.target_size, orig_size)
# ValueError if unaligned
with six.assertRaisesRegex(self, ValueError, "new size.*not.*aligned"):
device.target_size = Size("3.1 MiB")
self.assertEqual(device.target_size, orig_size)
# successfully set a new target size
new_target = device.max_size
device.target_size = new_target
self.assertEqual(device.target_size, new_target)
self.assertEqual(device.size, new_target)
parted_size = Size(device.parted_partition.getLength(unit='B'))
self.assertEqual(parted_size, device.target_size)
# reset target size to original size
device.target_size = orig_size
self.assertEqual(device.target_size, orig_size)
self.assertEqual(device.size, orig_size)
parted_size = Size(device.parted_partition.getLength(unit='B'))
self.assertEqual(parted_size, device.target_size)
