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 == []