def reset_device_test(self): """Test ResetDevice.""" 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=True, size=Size("6 GiB") ) dev3.original_format = get_format("ext4") dev3.format = get_format("xfs") self.module.on_storage_changed(create_storage()) self.module.storage.devicetree._add_device(dev1) self.module.storage.devicetree._add_device(dev2) self.module.storage.devicetree._add_device(dev3) with self.assertRaises(StorageConfigurationError): self.interface.ResetDevice("dev1") self.assertIn(dev1, self.module.storage.devices) self.assertIn(dev2, self.module.storage.devices) self.assertIn(dev3, self.module.storage.devices) self.assertEqual(dev3.format.type, "xfs") self.interface.ResetDevice("dev2") self.assertNotIn(dev1, self.module.storage.devices) self.assertNotIn(dev2, self.module.storage.devices) self.assertIn(dev3, self.module.storage.devices) self.assertEqual(dev3.format.type, "xfs") self.interface.ResetDevice("dev3") self.assertNotIn(dev1, self.module.storage.devices) self.assertNotIn(dev2, self.module.storage.devices) self.assertIn(dev3, self.module.storage.devices) self.assertEqual(dev3.format.type, "ext4")
def test_reset_device(self): """Test ResetDevice.""" 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=True, size=Size("6 GiB")) dev3.original_format = get_format("ext4") dev3.format = get_format("xfs") self.module.on_storage_changed(create_storage()) self.module.storage.devicetree._add_device(dev1) self.module.storage.devicetree._add_device(dev2) self.module.storage.devicetree._add_device(dev3) with pytest.raises(StorageConfigurationError): self.interface.ResetDevice("dev1") assert dev1 in self.module.storage.devices assert dev2 in self.module.storage.devices assert dev3 in self.module.storage.devices assert dev3.format.type == "xfs" self.interface.ResetDevice("dev2") assert dev1 not in self.module.storage.devices assert dev2 not in self.module.storage.devices assert dev3 in self.module.storage.devices assert dev3.format.type == "xfs" self.interface.ResetDevice("dev3") assert dev1 not in self.module.storage.devices assert dev2 not in self.module.storage.devices assert dev3 in self.module.storage.devices assert dev3.format.type == "ext4"