def setup(self, mock_machine):
mock_machine.return_value = 'x86_64'
self.firmware = mock.Mock()
self.firmware.efi_mode = mock.Mock(return_value=True)
kiwi.live_image_builder.FirmWare = mock.Mock(
return_value=self.firmware)
self.system_setup = mock.Mock()
kiwi.live_image_builder.SystemSetup = mock.Mock(
return_value=self.system_setup)
self.boot_image_task = mock.Mock()
self.boot_image_task.boot_root_directory = 'initrd_dir'
self.boot_image_task.initrd_filename = 'initrd'
kiwi.live_image_builder.BootImageTask = mock.Mock(
return_value=self.boot_image_task)
self.mbrid = mock.Mock()
self.mbrid.get_id = mock.Mock(return_value='0xffffffff')
kiwi.live_image_builder.ImageIdentifier = mock.Mock(
return_value=self.mbrid)
kiwi.live_image_builder.Path = mock.Mock()
self.kernel = mock.Mock()
self.kernel.get_kernel = mock.Mock()
self.kernel.get_xen_hypervisor = mock.Mock()
self.kernel.copy_kernel = mock.Mock()
self.kernel.copy_xen_hypervisor = mock.Mock()
kiwi.live_image_builder.Kernel = mock.Mock(return_value=self.kernel)
self.xml_state = mock.Mock()
self.xml_state.build_type.get_flags = mock.Mock(return_value=None)
self.xml_state.get_image_version = mock.Mock(return_value='1.2.3')
self.xml_state.xml_data.get_name = mock.Mock(
return_value='result-image')
self.xml_state.build_type.get_volid = mock.Mock(return_value='volid')
self.xml_state.build_type.get_kernelcmdline = mock.Mock(
return_value='custom_cmdline')
self.live_image = LiveImageBuilder(self.xml_state, 'target_dir',
'root_dir')
self.live_image.machine = mock.Mock()
self.live_image.machine.get_domain = mock.Mock(return_value='dom0')
self.result = mock.Mock()
self.live_image.result = self.result
self.live_image.hybrid = True
