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)