def setup(self, mock_machine, mock_exists, mock_mkdtemp):
mock_machine.return_value = 'x86_64'
mock_exists.return_value = True
description = XMLDescription('../data/example_config.xml')
self.xml_state = XMLState(
description.load()
)
self.manager = mock.Mock()
self.system_prepare = mock.Mock()
self.setup = mock.Mock()
self.profile = mock.Mock()
self.system_prepare.setup_repositories = mock.Mock(
return_value=self.manager
)
kiwi.boot.image.kiwi.SystemPrepare = mock.Mock(
return_value=self.system_prepare
)
kiwi.boot.image.kiwi.SystemSetup = mock.Mock(
return_value=self.setup
)
kiwi.boot.image.kiwi.Profile = mock.Mock(
return_value=self.profile
)
mock_mkdtemp.return_value = 'boot-directory'
self.boot_image = BootImageKiwi(
self.xml_state, 'some-target-dir'
)
class TestBootImageKiwi(object):
@patch('kiwi.boot.image.base.mkdtemp')
@patch('kiwi.boot.image.base.os.path.exists')
@patch('platform.machine')
def setup(self, mock_machine, mock_exists, mock_mkdtemp):
mock_machine.return_value = 'x86_64'
mock_exists.return_value = True
description = XMLDescription('../data/example_config.xml')
self.xml_state = XMLState(
description.load()
)
self.manager = mock.Mock()
self.system_prepare = mock.Mock()
self.setup = mock.Mock()
self.profile = mock.Mock()
self.system_prepare.setup_repositories = mock.Mock(
return_value=self.manager
)
kiwi.boot.image.kiwi.SystemPrepare = mock.Mock(
return_value=self.system_prepare
)
kiwi.boot.image.kiwi.SystemSetup = mock.Mock(
return_value=self.setup
)
kiwi.boot.image.kiwi.Profile = mock.Mock(
return_value=self.profile
)
mock_mkdtemp.return_value = 'boot-directory'
self.boot_image = BootImageKiwi(
self.xml_state, 'some-target-dir'
)
@patch('kiwi.defaults.Defaults.get_boot_image_description_path')
def test_prepare(self, mock_boot_path):
mock_boot_path.return_value = '../data'
self.boot_image.prepare()
self.system_prepare.setup_repositories.assert_called_once_with()
self.system_prepare.install_bootstrap.assert_called_once_with(
self.manager
)
self.system_prepare.install_system.assert_called_once_with(
self.manager
)
assert self.profile.add.call_args_list[0] == call(
'kiwi_initrdname', 'initrd-oemboot-suse-13.2'
)
self.setup.import_shell_environment.assert_called_once_with(
self.profile
)
self.setup.import_description.assert_called_once_with()
self.setup.import_overlay_files.assert_called_once_with(
follow_links=True
)
self.setup.call_config_script.assert_called_once_with()
self.system_prepare.pinch_system.assert_called_once_with(
manager=self.manager, force=True
)
self.setup.call_image_script.assert_called_once_with()
@raises(KiwiConfigFileNotFound)
@patch('os.path.exists')
def test_prepare_no_boot_description_found(self, mock_os_path):
mock_os_path.return_value = False
self.boot_image.prepare()
@patch('kiwi.boot.image.kiwi.ArchiveCpio')
@patch('kiwi.boot.image.kiwi.Compress')
@patch('kiwi.boot.image.kiwi.Path.create')
@patch('kiwi.boot.image.kiwi.Path.wipe')
@patch('kiwi.boot.image.kiwi.DataSync')
@patch('kiwi.boot.image.base.BootImageBase.is_prepared')
@patch('kiwi.boot.image.kiwi.mkdtemp')
def test_create_initrd(
self, mock_mkdtemp, mock_prepared, mock_sync,
mock_wipe, mock_create, mock_compress, mock_cpio
):
data = mock.Mock()
mock_sync.return_value = data
mock_mkdtemp.return_value = 'temp-boot-directory'
mock_prepared.return_value = True
mbrid = mock.Mock()
mbrid.write = mock.Mock()
cpio = mock.Mock()
compress = mock.Mock()
mock_cpio.return_value = cpio
mock_compress.return_value = compress
self.boot_image.create_initrd(mbrid)
mock_sync.assert_called_once_with(
'boot-directory/', 'temp-boot-directory'
)
data.sync_data.assert_called_once_with(options=['-z', '-a'])
mock_cpio.assert_called_once_with(
self.boot_image.target_dir +
'/LimeJeOS-openSUSE-13.2.x86_64-1.13.2.initrd'
)
mock_compress.assert_called_once_with(
self.boot_image.target_dir +
'/LimeJeOS-openSUSE-13.2.x86_64-1.13.2.initrd'
)
mock_wipe.assert_called_once_with(
'temp-boot-directory/boot'
)
mock_create.assert_called_once_with(
'temp-boot-directory/boot'
)
mbrid.write.assert_called_once_with(
'temp-boot-directory/boot/mbrid'
)
cpio.create.assert_called_once_with(
source_dir='temp-boot-directory',
exclude=[
'/var/cache', '/image', '/usr/lib/grub*',
'/usr/share/doc', '/usr/share/man', '/home', '/media', '/srv'
]
)
compress.xz.assert_called_once_with()