def create_source(source_type: SourceType): """Create a source module. :param source_type: a source type :return: a source module """ if source_type == SourceType.LIVE_OS_IMAGE: from pyanaconda.modules.payloads.source.live_os.live_os import LiveOSSourceModule return LiveOSSourceModule() elif source_type == SourceType.LIVE_IMAGE: from pyanaconda.modules.payloads.source.live_image.live_image import \ LiveImageSourceModule return LiveImageSourceModule() elif source_type == SourceType.LIVE_TAR: from pyanaconda.modules.payloads.source.live_tar.live_tar import LiveTarSourceModule return LiveTarSourceModule() elif source_type == SourceType.CDROM: from pyanaconda.modules.payloads.source.cdrom.cdrom import CdromSourceModule return CdromSourceModule() elif source_type == SourceType.HMC: from pyanaconda.modules.payloads.source.hmc.hmc import HMCSourceModule return HMCSourceModule() elif source_type == SourceType.REPO_FILES: from pyanaconda.modules.payloads.source.repo_files.repo_files import \ RepoFilesSourceModule return RepoFilesSourceModule() elif source_type == SourceType.NFS: from pyanaconda.modules.payloads.source.nfs.nfs import NFSSourceModule return NFSSourceModule() elif source_type == SourceType.URL: from pyanaconda.modules.payloads.source.url.url import URLSourceModule return URLSourceModule() elif source_type == SourceType.HDD: from pyanaconda.modules.payloads.source.harddrive.harddrive import \ HardDriveSourceModule return HardDriveSourceModule() elif source_type == SourceType.CDN: from pyanaconda.modules.payloads.source.cdn.cdn import CDNSourceModule return CDNSourceModule() elif source_type == SourceType.CLOSEST_MIRROR: from pyanaconda.modules.payloads.source.closest_mirror.closest_mirror import \ ClosestMirrorSourceModule return ClosestMirrorSourceModule() elif source_type == SourceType.RPM_OSTREE: from pyanaconda.modules.payloads.source.rpm_ostree.rpm_ostree import \ RPMOSTreeSourceModule return RPMOSTreeSourceModule() elif source_type == SourceType.FLATPAK: from pyanaconda.modules.payloads.source.flatpak.flatpak import \ FlatpakSourceModule return FlatpakSourceModule() raise ValueError("Unknown source type: {}".format(source_type))
def setUp(self): self.module = CDNSourceModule() self.interface = CDNSourceInterface(self.module)