def get_default_class_test(self):
"""Test get_default_class."""
cls = BootLoaderFactory.get_default_class()
self.assertEqual(cls, None)
BootLoaderFactory.set_default_class(EXTLINUX)
cls = BootLoaderFactory.get_default_class()
self.assertEqual(cls, EXTLINUX)
def test_get_default_class(self):
"""Test get_default_class."""
cls = BootLoaderFactory.get_default_class()
assert cls is None
BootLoaderFactory.set_default_class(EXTLINUX)
cls = BootLoaderFactory.get_default_class()
assert cls == EXTLINUX
def set_default_type(self, default_type):
"""Set the default type of the boot loader.
:param default_type: an instance of BootloaderType
"""
# Set up the bootloader factory.
cls = BootLoaderFactory.get_class_by_name(default_type.value)
BootLoaderFactory.set_default_class(cls)
# Set up the property.
self._default_type = default_type
log.debug("The default type is set to '%s'.", default_type)
def __exit__(self, *exc):
from pyanaconda.modules.storage.bootloader import BootLoaderFactory
BootLoaderFactory.set_default_class(None)
return False
def __enter__(self):
from pyanaconda.modules.storage.bootloader import BootLoaderFactory
BootLoaderFactory.set_default_class(self._default_type)
return self