def generate_device_factory_request(storage, device) -> DeviceFactoryRequest:
"""Generate a device info for the given device.
:param storage: an instance of Blivet
:param device: a device
:return: a device factory request
"""
device_type = devicefactory.get_device_type(device)
if device_type is None:
raise UnsupportedDeviceError("Unsupported type of {}.".format(
device.name))
# Generate the device data.
request = DeviceFactoryRequest()
request.device_spec = device.name
request.device_name = getattr(device.raw_device, "lvname",
device.raw_device.name)
request.device_size = device.size.get_bytes()
request.device_type = device_type
request.reformat = not device.format.exists
request.format_type = device.format.type or ""
request.device_encrypted = isinstance(device, LUKSDevice)
request.luks_version = get_device_luks_version(device)
request.label = getattr(device.format, "label", "") or ""
request.mount_point = getattr(device.format, "mountpoint", "") or ""
request.device_raid_level = get_device_raid_level_name(device)
if hasattr(device, "req_disks") and not device.exists:
disks = device.req_disks
else:
disks = device.disks
request.disks = [d.name for d in disks]
if request.device_type not in CONTAINER_DEVICE_TYPES:
return request
# Generate the container data.
factory = devicefactory.get_device_factory(storage,
device_type=device_type,
device=device.raw_device)
container = factory.get_container()
if container:
set_container_data(request, container)
return request
def get_device_factory_arguments_test(self):
"""Test get_device_factory_arguments."""
dev1 = StorageDevice("dev1")
self._add_device(dev1)
dev2 = StorageDevice("dev2")
self._add_device(dev2)
dev3 = StorageDevice("dev3")
self._add_device(dev3)
request = DeviceFactoryRequest()
request.device_spec = "dev3"
request.disks = ["dev1", "dev2"]
request.device_name = "dev3"
request.device_type = devicefactory.DEVICE_TYPE_LVM_THINP
request.device_size = Size("10 GiB").get_bytes()
request.mount_point = "/"
request.format_type = "xfs"
request.label = "root"
request.device_encrypted = True
request.luks_version = "luks1"
request.device_raid_level = "raid1"
self.assertEqual(utils.get_device_factory_arguments(self.storage, request), {
"device": dev3,
"disks": [dev1, dev2],
"device_type": devicefactory.DEVICE_TYPE_LVM_THINP,
"device_name": "dev3",
"size": Size("10 GiB"),
"mountpoint": "/",
"fstype": "xfs",
"label": "root",
"encrypted": True,
"luks_version": "luks1",
"raid_level": raid.RAID1,
"container_name": None,
"container_size": devicefactory.SIZE_POLICY_AUTO,
"container_raid_level": None,
"container_encrypted": False
})
request = DeviceFactoryRequest()
request.device_spec = "dev3"
request.disks = ["dev1", "dev2"]
request.device_name = "dev3"
request.container_name = "container1"
request.container_size_policy = Size("10 GiB").get_bytes()
request.container_encrypted = True
request.container_raid_level = "raid1"
self.assertEqual(utils.get_device_factory_arguments(self.storage, request), {
"device": dev3,
"disks": [dev1, dev2],
"device_type": devicefactory.DEVICE_TYPE_LVM,
"device_name": "dev3",
"size": None,
"mountpoint": None,
"fstype": None,
"label": None,
"encrypted": False,
"luks_version": None,
"raid_level": None,
"container_name": "container1",
"container_size": Size("10 GiB"),
"container_raid_level": raid.RAID1,
"container_encrypted": True
})
def generate_container_data_test(self, proxy_getter):
"""Test GenerateContainerData."""
network_proxy = Mock()
network_proxy.Hostname = "localhost"
network_proxy.GetCurrentHostname.return_value = "localhost"
proxy_getter.return_value = network_proxy
pv1 = StorageDevice("pv1",
size=Size("1025 MiB"),
fmt=get_format("lvmpv"))
pv2 = StorageDevice("pv2",
size=Size("513 MiB"),
fmt=get_format("lvmpv"))
vg = LVMVolumeGroupDevice("testvg", parents=[pv1, pv2])
lv = LVMLogicalVolumeDevice("testlv",
size=Size("512 MiB"),
parents=[vg],
fmt=get_format("xfs"),
exists=False,
seg_type="raid1",
pvs=[pv1, pv2])
self._add_device(pv1)
self._add_device(pv2)
self._add_device(vg)
self._add_device(lv)
request = DeviceFactoryRequest()
request.device_spec = lv.name
request.device_type = DEVICE_TYPE_LVM
request = DeviceFactoryRequest.from_structure(
self.interface.GenerateContainerData(
DeviceFactoryRequest.to_structure(request)))
self.assertEqual(request.container_spec, "testvg")
self.assertEqual(request.container_name, "testvg")
self.assertEqual(request.container_encrypted, False)
self.assertEqual(request.container_raid_level, "")
self.assertEqual(request.container_size_policy,
Size("1.5 GiB").get_bytes())
request.device_type = DEVICE_TYPE_BTRFS
request = DeviceFactoryRequest.from_structure(
self.interface.GenerateContainerData(
DeviceFactoryRequest.to_structure(request)))
self.assertEqual(request.container_spec, "")
self.assertEqual(request.container_name, "anaconda")
self.assertEqual(request.container_encrypted, False)
self.assertEqual(request.container_raid_level, "single")
self.assertEqual(request.container_size_policy, 0)
request.device_type = DEVICE_TYPE_PARTITION
request = DeviceFactoryRequest.from_structure(
self.interface.GenerateContainerData(
DeviceFactoryRequest.to_structure(request)))
self.assertEqual(request.container_spec, "")
self.assertEqual(request.container_name, "")
self.assertEqual(request.container_encrypted, False)
self.assertEqual(request.container_raid_level, "")
self.assertEqual(request.container_size_policy, 0)
def generate_request_description_test(self):
"""Test generate_request_description."""
request = DeviceFactoryRequest()
request.device_spec = "dev3"
request.disks = ["dev1", "dev2"]
request.device_name = "dev3"
request.device_type = devicefactory.DEVICE_TYPE_LVM_THINP
request.device_size = Size("10 GiB").get_bytes()
request.mount_point = "/"
request.format_type = "xfs"
request.label = "root"
request.device_encrypted = True
request.luks_version = "luks1"
request.device_raid_level = "raid1"
expected = dedent("""
{
container-encrypted = False
container-name = ''
container-raid-level = ''
container-size-policy = 0
container-spec = ''
device-encrypted = True
device-name = 'dev3'
device-raid-level = 'raid1'
device-size = 10737418240
device-spec = 'dev3'
device-type = 5
disks = ['dev1', 'dev2']
format-type = 'xfs'
label = 'root'
luks-version = 'luks1'
mount-point = '/'
reformat = False
}
""").strip()
self.assertEqual(generate_request_description(request), expected)
original = copy.deepcopy(request)
self.assertEqual(generate_request_description(request, original),
expected)
request.device_name = "dev4"
request.disks = ["dev1"]
request.device_encrypted = False
expected = dedent("""
{
container-encrypted = False
container-name = ''
container-raid-level = ''
container-size-policy = 0
container-spec = ''
device-encrypted = True -> False
device-name = 'dev3' -> 'dev4'
device-raid-level = 'raid1'
device-size = 10737418240
device-spec = 'dev3'
device-type = 5
disks = ['dev1', 'dev2'] -> ['dev1']
format-type = 'xfs'
label = 'root'
luks-version = 'luks1'
mount-point = '/'
reformat = False
}
""").strip()
self.assertEqual(generate_request_description(request, original),
expected)