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