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
)
