class DiskInitializationModuleTestCase(unittest.TestCase): """Test the disk initialization module.""" def setUp(self): """Set up the module.""" self.disk_init_module = DiskInitializationModule() def storage_property_test(self): """Test the storage property.""" with self.assertRaises(UnavailableStorageError): self.assertIsNotNone(self.disk_init_module.storage) storage = Mock() self.disk_init_module.on_storage_changed(storage) self.assertEqual(self.disk_init_module.storage, storage) def setup_kickstart_test(self): """Test setup_kickstart with storage.""" storage = create_storage() data = Mock() self.disk_init_module.on_storage_changed(storage) self.disk_init_module.set_initialization_mode(InitializationMode.CLEAR_NONE) self.disk_init_module.setup_kickstart(data) self.assertEqual(data.clearpart.type, CLEARPART_TYPE_NONE) self.assertEqual(data.clearpart.devices, []) self.assertEqual(data.clearpart.drives, [])
class DiskInitializationModuleTestCase(unittest.TestCase): """Test the disk initialization module.""" def setUp(self): """Set up the module.""" self.disk_init_module = DiskInitializationModule() def test_storage_property(self): """Test the storage property.""" with pytest.raises(UnavailableStorageError): assert self.disk_init_module.storage is not None storage = Mock() self.disk_init_module.on_storage_changed(storage) assert self.disk_init_module.storage == storage def test_setup_kickstart(self): """Test setup_kickstart with storage.""" storage = create_storage() data = Mock() self.disk_init_module.on_storage_changed(storage) self.disk_init_module.set_initialization_mode( InitializationMode.CLEAR_NONE) self.disk_init_module.setup_kickstart(data) assert data.clearpart.type == CLEARPART_TYPE_NONE assert data.clearpart.devices == [] assert data.clearpart.drives == []
class DiskInitializationModuleTestCase(unittest.TestCase): """Test the disk initialization module.""" def setUp(self): """Set up the module.""" self.disk_init_module = DiskInitializationModule() def storage_property_test(self): """Test the storage property.""" with self.assertRaises(UnavailableStorageError): self.assertIsNotNone(self.disk_init_module.storage) storage = Mock() self.disk_init_module.on_storage_reset(storage) self.assertEqual(self.disk_init_module.storage, storage) def setup_kickstart_test(self): """Test setup_kickstart with storage.""" storage = create_storage() data = Mock() self.disk_init_module.on_storage_reset(storage) self.disk_init_module.set_initialization_mode(InitializationMode.CLEAR_NONE) self.disk_init_module.setup_kickstart(data) self.assertEqual(data.clearpart.type, CLEARPART_TYPE_NONE) self.assertEqual(data.clearpart.devices, []) self.assertEqual(data.clearpart.drives, [])
class DiskInitializationInterfaceTestCase(unittest.TestCase): """Test DBus interface of the disk initialization module.""" def setUp(self): """Set up the module.""" self.disk_init_module = DiskInitializationModule() self.disk_init_interface = DiskInitializationInterface(self.disk_init_module) def on_partitioning_changed_test(self): """Smoke test for on_partitioning_changed callback.""" mode_changed_callback = Mock() devices_changed_callback = Mock() drives_changed_callback = Mock() self.disk_init_module.set_initialization_mode(InitializationMode.CLEAR_NONE) self.disk_init_module.initialization_mode_changed.connect(mode_changed_callback) self.disk_init_module.devices_to_clear_changed.connect(devices_changed_callback) self.disk_init_module.drives_to_clear_changed.connect(drives_changed_callback) self.disk_init_module.on_partitioning_changed(create_storage()) mode_changed_callback.called_once_with(InitializationMode.CLEAR_NONE) drives_changed_callback.called_once_with([]) devices_changed_callback.called_once_with([]) def _test_dbus_property(self, *args, **kwargs): check_dbus_property( self, DISK_INITIALIZATION, self.disk_init_interface, *args, **kwargs ) def default_disk_label_property_test(self): """Test the default disk label property.""" self._test_dbus_property( "DefaultDiskLabel", "msdos" ) def format_unrecognized_enabled_property_test(self): """Test the can format unrecognized property.""" self._test_dbus_property( "FormatUnrecognizedEnabled", False ) def can_initialize_label_property_test(self): """Test the can initialize label property.""" self._test_dbus_property( "InitializeLabelsEnabled", False ) def format_ldl_enabled_property_test(self): """Test the can format LDL property.""" self._test_dbus_property( "FormatLDLEnabled", True ) def initialization_mode_property_test(self): """Test the type to clear property.""" self._test_dbus_property( "InitializationMode", CLEAR_PARTITIONS_LINUX ) def devices_to_clear_property_test(self): """Test the devices to clear property.""" self._test_dbus_property( "DevicesToClear", ["sda2", "sda3", "sdb1"] ) def drives_to_clear_property_test(self): """Test the drives to clear property.""" self._test_dbus_property( "DrivesToClear", ["sda", "sdb"] )