def create_partitioning(method: PartitioningMethod):
"""Create a partitioning module.
:param method: a partitioning method
:return: a partitioning module
"""
if method is PartitioningMethod.AUTOMATIC:
from pyanaconda.modules.storage.partitioning.automatic.automatic_module import \
AutoPartitioningModule
return AutoPartitioningModule()
if method is PartitioningMethod.MANUAL:
from pyanaconda.modules.storage.partitioning.manual.manual_module import \
ManualPartitioningModule
return ManualPartitioningModule()
if method is PartitioningMethod.CUSTOM:
from pyanaconda.modules.storage.partitioning.custom.custom_module import \
CustomPartitioningModule
return CustomPartitioningModule()
if method is PartitioningMethod.INTERACTIVE:
from pyanaconda.modules.storage.partitioning.interactive.interactive_module import \
InteractivePartitioningModule
return InteractivePartitioningModule()
if method is PartitioningMethod.BLIVET:
from pyanaconda.modules.storage.partitioning.blivet.blivet_module import \
BlivetPartitioningModule
return BlivetPartitioningModule()
raise ValueError("Unknown partitioning method: {}".format(method))
def setUp(self):
"""Set up the module."""
self.module = BlivetPartitioningModule()
self.interface = BlivetPartitioningInterface(self.module)