294 lines
10 KiB
Python
294 lines
10 KiB
Python
from mock import patch
|
|
from mock import call
|
|
import mock
|
|
import kiwi
|
|
|
|
from .test_helper import raises, patch_open
|
|
|
|
from kiwi.exceptions import KiwiLiveBootImageError
|
|
|
|
from kiwi.builder.live import LiveImageBuilder
|
|
|
|
|
|
class TestLiveImageBuilder(object):
|
|
@patch('platform.machine')
|
|
def setup(self, mock_machine):
|
|
mock_machine.return_value = 'x86_64'
|
|
self.firmware = mock.Mock()
|
|
self.firmware.efi_mode = mock.Mock(
|
|
return_value='uefi'
|
|
)
|
|
kiwi.builder.live.FirmWare = mock.Mock(
|
|
return_value=self.firmware
|
|
)
|
|
self.setup = mock.Mock()
|
|
kiwi.builder.live.SystemSetup = mock.Mock(
|
|
return_value=self.setup
|
|
)
|
|
self.boot_image_task = mock.Mock()
|
|
self.boot_image_task.boot_root_directory = 'initrd_dir'
|
|
self.boot_image_task.initrd_filename = 'initrd'
|
|
kiwi.builder.live.BootImage = mock.Mock(
|
|
return_value=self.boot_image_task
|
|
)
|
|
self.mbrid = mock.Mock()
|
|
self.mbrid.get_id = mock.Mock(
|
|
return_value='0xffffffff'
|
|
)
|
|
kiwi.builder.live.SystemIdentifier = mock.Mock(
|
|
return_value=self.mbrid
|
|
)
|
|
kiwi.builder.live.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.builder.live.Kernel = mock.Mock(
|
|
return_value=self.kernel
|
|
)
|
|
self.xml_state = mock.Mock()
|
|
self.xml_state.get_fs_mount_option_list = mock.Mock(
|
|
return_value='async'
|
|
)
|
|
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
|
|
|
|
@patch('platform.machine')
|
|
def test_init_for_ix86_platform(self, mock_machine):
|
|
xml_state = mock.Mock()
|
|
xml_state.xml_data.get_name = mock.Mock(
|
|
return_value='some-image'
|
|
)
|
|
xml_state.get_image_version = mock.Mock(
|
|
return_value='1.2.3'
|
|
)
|
|
mock_machine.return_value = 'i686'
|
|
live_image = LiveImageBuilder(
|
|
xml_state, 'target_dir', 'root_dir'
|
|
)
|
|
assert live_image.arch == 'ix86'
|
|
|
|
@patch('shutil.copy')
|
|
@patch('kiwi.builder.live.mkdtemp')
|
|
@patch('kiwi.builder.live.Command.run')
|
|
@patch('kiwi.builder.live.Iso.create_hybrid')
|
|
@patch('kiwi.builder.live.FileSystem')
|
|
@patch('kiwi.builder.live.FileSystemIsoFs')
|
|
@patch('kiwi.builder.live.BootLoaderConfig')
|
|
@patch('kiwi.builder.live.SystemSize')
|
|
@patch_open
|
|
def test_create_overlay_structure(
|
|
self, mock_open, mock_size, mock_bootloader, mock_isofs, mock_fs,
|
|
mock_hybrid, mock_command, mock_dtemp, mock_copy
|
|
):
|
|
tmpdir_name = ['temp-squashfs', 'temp_media_dir']
|
|
|
|
def side_effect(prefix, dir):
|
|
return tmpdir_name.pop()
|
|
|
|
mock_dtemp.side_effect = side_effect
|
|
context_manager_mock = mock.Mock()
|
|
mock_open.return_value = context_manager_mock
|
|
file_mock = mock.Mock()
|
|
enter_mock = mock.Mock()
|
|
exit_mock = mock.Mock()
|
|
enter_mock.return_value = file_mock
|
|
setattr(context_manager_mock, '__enter__', enter_mock)
|
|
setattr(context_manager_mock, '__exit__', exit_mock)
|
|
self.live_image.live_type = 'overlay'
|
|
live_type_image = mock.Mock()
|
|
mock_fs.return_value = live_type_image
|
|
bootloader = mock.Mock()
|
|
mock_bootloader.return_value = bootloader
|
|
iso_image = mock.Mock()
|
|
iso_image.create_on_file.return_value = 'offset'
|
|
mock_isofs.return_value = iso_image
|
|
rootsize = mock.Mock()
|
|
rootsize.accumulate_mbyte_file_sizes = mock.Mock(
|
|
return_value=8192
|
|
)
|
|
mock_size.return_value = rootsize
|
|
self.setup.export_rpm_package_verification.return_value = '.verified'
|
|
self.setup.export_rpm_package_list.return_value = '.packages'
|
|
|
|
self.live_image.create()
|
|
|
|
self.live_image.boot_image_task.prepare.assert_called_once_with()
|
|
self.setup.export_modprobe_setup.assert_called_once_with(
|
|
'initrd_dir'
|
|
)
|
|
mock_fs.assert_called_once_with(
|
|
custom_args={'mount_options': 'async'},
|
|
device_provider=None, name='squashfs', root_dir='root_dir'
|
|
)
|
|
live_type_image.create_on_file.assert_called_once_with(
|
|
'target_dir/result-image-read-only.x86_64-1.2.3'
|
|
)
|
|
assert mock_command.call_args_list[0] == call(
|
|
[
|
|
'mv', 'target_dir/result-image-read-only.x86_64-1.2.3',
|
|
'temp_media_dir'
|
|
]
|
|
)
|
|
mock_open.assert_called_once_with(
|
|
'temp_media_dir/config.isoclient', 'w'
|
|
)
|
|
assert file_mock.write.call_args_list == [
|
|
call('IMAGE="loop;result-image.x86_64;1.2.3"\n'),
|
|
call('UNIONFS_CONFIG="tmpfs,loop,overlay"\n')
|
|
]
|
|
|
|
assert mock_bootloader.call_args_list[0] == call(
|
|
'isolinux', self.xml_state, 'temp_media_dir'
|
|
)
|
|
assert bootloader.setup_live_boot_images.call_args_list[0] == call(
|
|
lookup_path=self.live_image.boot_image_task.boot_root_directory,
|
|
mbrid=None
|
|
)
|
|
assert bootloader.setup_live_image_config.call_args_list[0] == call(
|
|
mbrid=None
|
|
)
|
|
assert bootloader.write.call_args_list[0] == call()
|
|
mock_copy.assert_called_once_with(
|
|
'temp_media_dir/boot/grub2/grub.cfg', 'temp_media_dir/EFI/BOOT'
|
|
)
|
|
|
|
assert mock_bootloader.call_args_list[1] == call(
|
|
'grub2', self.xml_state, 'temp_media_dir'
|
|
)
|
|
assert bootloader.setup_live_boot_images.call_args_list[1] == call(
|
|
lookup_path=self.live_image.boot_image_task.boot_root_directory,
|
|
mbrid=self.mbrid
|
|
)
|
|
assert bootloader.setup_live_image_config.call_args_list[1] == call(
|
|
mbrid=self.mbrid
|
|
)
|
|
assert bootloader.write.call_args_list[1] == call()
|
|
|
|
self.boot_image_task.create_initrd.assert_called_once_with(
|
|
self.mbrid
|
|
)
|
|
self.kernel.copy_kernel.assert_called_once_with(
|
|
'temp_media_dir/boot/x86_64/loader', '/linux'
|
|
)
|
|
self.kernel.copy_xen_hypervisor.assert_called_once_with(
|
|
'temp_media_dir/boot/x86_64/loader', '/xen.gz'
|
|
)
|
|
assert mock_command.call_args_list[1] == call(
|
|
['mv', 'initrd', 'temp_media_dir/boot/x86_64/loader/initrd']
|
|
)
|
|
mock_size.assert_called_once_with(
|
|
'temp_media_dir'
|
|
)
|
|
rootsize.accumulate_mbyte_file_sizes.assert_called_once_with()
|
|
mock_isofs.assert_called_once_with(
|
|
custom_args={
|
|
'create_options': [
|
|
'-A', '0xffffffff',
|
|
'-p', '"KIWI - http://suse.github.com/kiwi"',
|
|
'-publisher', '"SUSE LINUX GmbH"',
|
|
'-V', '"volid"',
|
|
'-iso-level', '3', '-udf'
|
|
]
|
|
}, device_provider=None, root_dir='temp_media_dir'
|
|
)
|
|
iso_image.create_on_file.assert_called_once_with(
|
|
'target_dir/result-image.x86_64-1.2.3.iso'
|
|
)
|
|
mock_hybrid.assert_called_once_with(
|
|
'offset', self.mbrid, 'target_dir/result-image.x86_64-1.2.3.iso'
|
|
)
|
|
assert self.result.add.call_args_list == [
|
|
call(
|
|
key='live_image',
|
|
filename='target_dir/result-image.x86_64-1.2.3.iso',
|
|
use_for_bundle=True,
|
|
compress=False,
|
|
shasum=True
|
|
),
|
|
call(
|
|
key='image_packages',
|
|
filename='.packages',
|
|
use_for_bundle=True,
|
|
compress=False,
|
|
shasum=False
|
|
),
|
|
call(
|
|
key='image_verified',
|
|
filename='.verified',
|
|
use_for_bundle=True,
|
|
compress=False,
|
|
shasum=False
|
|
)
|
|
]
|
|
self.setup.export_rpm_package_verification.assert_called_once_with(
|
|
'target_dir'
|
|
)
|
|
self.setup.export_rpm_package_list.assert_called_once_with(
|
|
'target_dir'
|
|
)
|
|
|
|
@patch('kiwi.builder.live.mkdtemp')
|
|
@patch('kiwi.builder.live.Command.run')
|
|
@raises(KiwiLiveBootImageError)
|
|
def test_create_invalid_iso_structure(self, mock_command, mock_dtemp):
|
|
self.live_image.live_type = 'bogus'
|
|
self.live_image.create()
|
|
|
|
@patch('shutil.copy')
|
|
@patch('kiwi.builder.live.mkdtemp')
|
|
@patch('kiwi.builder.live.Command.run')
|
|
@patch('kiwi.builder.live.BootLoaderConfig')
|
|
@patch_open
|
|
@raises(KiwiLiveBootImageError)
|
|
def test_create_no_kernel_found(
|
|
self, mock_open, mock_boot, mock_command, mock_dtemp, mock_copy
|
|
):
|
|
self.kernel.get_kernel.return_value = False
|
|
self.live_image.create()
|
|
|
|
@patch('shutil.copy')
|
|
@patch('kiwi.builder.live.mkdtemp')
|
|
@patch('kiwi.builder.live.Command.run')
|
|
@patch('kiwi.builder.live.BootLoaderConfig')
|
|
@patch_open
|
|
@raises(KiwiLiveBootImageError)
|
|
def test_create_no_hypervisor_found(
|
|
self, mock_open, mock_boot, mock_command, mock_dtemp, mock_copy
|
|
):
|
|
self.kernel.get_xen_hypervisor.return_value = False
|
|
self.live_image.create()
|
|
|
|
@patch('kiwi.builder.live.Path.wipe')
|
|
def test_destructor(self, mock_wipe):
|
|
self.live_image.media_dir = 'media-dir'
|
|
self.live_image.__del__()
|
|
assert mock_wipe.call_args_list == [
|
|
call('media-dir')
|
|
]
|
|
self.live_image.media_dir = None
|