def shrink_device_test(self):
"""Test ShrinkDevice."""
self.module.on_storage_reset(create_storage())
sda1 = StorageDevice("sda1",
exists=False,
size=Size("10 GiB"),
fmt=get_format("ext4"))
self.module.storage.devicetree._add_device(sda1)
def resize_device(device, size):
device.size = size
self.module.storage.resize_device = resize_device
sda1.protected = True
with self.assertRaises(ProtectedDeviceError):
self.interface.ShrinkDevice("sda1", Size("3 GiB").get_bytes())
sda1.protected = False
self.interface.ShrinkDevice("sda1", Size("3 GiB").get_bytes())
self.assertEqual(sda1.size, Size("3 GiB"))
self.interface.ShrinkDevice("sda1", Size("5 GiB").get_bytes())
self.assertEqual(sda1.size, Size("3 GiB"))
def test_remove_device(self):
"""Test RemoveDevice."""
self.module.on_storage_changed(create_storage())
dev1 = StorageDevice(
"dev1",
exists=False,
size=Size("15 GiB"),
fmt=get_format("disklabel")
)
dev2 = StorageDevice(
"dev2",
exists=False,
parents=[dev1],
size=Size("6 GiB"),
fmt=get_format("ext4")
)
dev3 = StorageDevice(
"dev3",
exists=False,
parents=[dev1],
size=Size("9 GiB"),
fmt=get_format("ext4")
)
self.module.storage.devicetree._add_device(dev1)
self.module.storage.devicetree._add_device(dev2)
self.module.storage.devicetree._add_device(dev3)
dev1.protected = True
with pytest.raises(ProtectedDeviceError):
self.interface.RemoveDevice("dev1")
assert dev1 in self.module.storage.devices
assert dev2 in self.module.storage.devices
assert dev3 in self.module.storage.devices
dev1.protected = False
dev2.protected = True
self.interface.RemoveDevice("dev1")
assert dev1 in self.module.storage.devices
assert dev2 in self.module.storage.devices
assert dev3 not in self.module.storage.devices
dev2.protected = False
self.interface.RemoveDevice("dev1")
assert dev1 not in self.module.storage.devices
assert dev2 not in self.module.storage.devices
assert dev3 not in self.module.storage.devices
def test_gather_unusable_requests(self):
"""Test GatherRequests with unusable devices."""
self.module.on_storage_changed(create_storage())
# Add device with no size.
self._add_device(StorageDevice(
"dev1",
size=Size(0)
))
assert self.interface.GatherRequests() == []
# Add protected device.
device = StorageDevice(
"dev2",
size=Size("1 GiB")
)
device.protected = True
self._add_device(device)
assert self.interface.GatherRequests() == []
# Add unselected disk.
self._add_device(DiskDevice(
"dev3",
size=Size("1 GiB")
))
self.module.on_selected_disks_changed(["dev1", "dev2"])
assert self.interface.GatherRequests() == []
def remove_device_test(self):
"""Test RemoveDevice."""
self.module.on_storage_reset(create_storage())
dev1 = StorageDevice("dev1",
exists=False,
size=Size("15 GiB"),
fmt=get_format("disklabel"))
dev2 = StorageDevice("dev2",
exists=False,
parents=[dev1],
size=Size("6 GiB"),
fmt=get_format("ext4"))
dev3 = StorageDevice("dev3",
exists=False,
parents=[dev1],
size=Size("9 GiB"),
fmt=get_format("ext4"))
self.module.storage.devicetree._add_device(dev1)
self.module.storage.devicetree._add_device(dev2)
self.module.storage.devicetree._add_device(dev3)
dev1.protected = True
with self.assertRaises(ProtectedDeviceError):
self.interface.RemoveDevice("dev1")
self.assertIn(dev1, self.module.storage.devices)
self.assertIn(dev2, self.module.storage.devices)
self.assertIn(dev3, self.module.storage.devices)
dev1.protected = False
dev2.protected = True
self.interface.RemoveDevice("dev1")
self.assertIn(dev1, self.module.storage.devices)
self.assertIn(dev2, self.module.storage.devices)
self.assertNotIn(dev3, self.module.storage.devices)
dev2.protected = False
self.interface.RemoveDevice("dev1")
self.assertNotIn(dev1, self.module.storage.devices)
self.assertNotIn(dev2, self.module.storage.devices)
self.assertNotIn(dev3, self.module.storage.devices)
