def setup(self, mock_boot, mock_filesystem): self.setup = Mock() kiwi.builder.pxe.SystemSetup = Mock(return_value=self.setup) self.boot_image_task = MagicMock() self.boot_image_task.boot_root_directory = 'initrd_dir' self.boot_image_task.initrd_filename = 'initrd_file_name' mock_boot.return_value = self.boot_image_task self.filesystem = MagicMock() self.filesystem.filename = 'myimage.fs' self.filesystem.root_uuid = 'some_uuid' mock_filesystem.return_value = self.filesystem self.xml_state = Mock() self.xml_state.get_image_version = Mock(return_value='1.2.3') self.xml_state.get_initrd_system = Mock(return_value='dracut') self.xml_state.xml_data.get_name = Mock(return_value='some-image') self.xml_state.build_type = Mock() self.xml_state.build_type.get_kernelcmdline = Mock( return_value='console=hvc0') kernel_type = namedtuple('kernel', ['filename', 'version']) xen_type = namedtuple('xen', ['filename', 'name']) self.kernel = Mock() self.kernel.get_kernel = Mock( return_value=kernel_type(filename='some-kernel', version='42')) self.kernel.get_xen_hypervisor = Mock( return_value=xen_type(filename='hypervisor', name='xen.gz')) kiwi.builder.pxe.Kernel = Mock(return_value=self.kernel) self.pxe = PxeBuilder( self.xml_state, 'target_dir', 'root_dir', custom_args={'signing_keys': ['key_file_a', 'key_file_b']}) self.pxe.image_name = 'myimage' self.pxe.compressed = True
def setup(self, mock_boot, mock_filesystem): self.setup = mock.Mock() kiwi.builder.pxe.SystemSetup = mock.Mock(return_value=self.setup) self.boot_image_task = mock.MagicMock() self.boot_image_task.boot_root_directory = 'initrd_dir' mock_boot.return_value = self.boot_image_task self.filesystem = mock.MagicMock() self.filesystem.filename = 'myimage.fs' mock_filesystem.return_value = self.filesystem self.xml_state = mock.Mock() self.xml_state.get_image_version = mock.Mock(return_value='1.2.3') self.xml_state.xml_data.get_name = mock.Mock(return_value='some-image') kernel_type = namedtuple('kernel', ['filename', 'version']) xen_type = namedtuple('xen', ['filename', 'name']) self.kernel = mock.Mock() self.kernel.get_kernel = mock.Mock( return_value=kernel_type(filename='some-kernel', version='42')) self.kernel.get_xen_hypervisor = mock.Mock( return_value=xen_type(filename='hypervisor', name='xen.gz')) kiwi.builder.pxe.Kernel = mock.Mock(return_value=self.kernel) self.pxe = PxeBuilder( self.xml_state, 'target_dir', 'root_dir', custom_args={'signing_keys': ['key_file_a', 'key_file_b']}) self.machine = mock.Mock() self.machine.get_domain = mock.Mock(return_value='dom0') self.pxe.machine = self.machine self.pxe.image_name = 'myimage'
def __new__(self, xml_state, target_dir, root_dir, custom_args=None): requested_image_type = xml_state.get_build_type_name() if requested_image_type in Defaults.get_filesystem_image_types(): return FileSystemBuilder( xml_state, target_dir, root_dir ) elif requested_image_type in Defaults.get_disk_image_types(): return DiskBuilder( xml_state, target_dir, root_dir, custom_args ) elif requested_image_type in Defaults.get_live_image_types(): return LiveImageBuilder( xml_state, target_dir, root_dir, custom_args ) elif requested_image_type in Defaults.get_network_image_types(): return PxeBuilder( xml_state, target_dir, root_dir, custom_args ) elif requested_image_type in Defaults.get_archive_image_types(): return ArchiveBuilder( xml_state, target_dir, root_dir, custom_args ) elif requested_image_type in Defaults.get_container_image_types(): return ContainerBuilder( xml_state, target_dir, root_dir, custom_args ) else: raise KiwiRequestedTypeError( 'requested image type %s not supported' % requested_image_type )