d808b241b7
There are some directories needed during boot image creation time, e.g grub2 modules to create grub images. But at boot time this data is no longer required and would just waste space in the initrd
139 lines
4.7 KiB
Python
139 lines
4.7 KiB
Python
import sys
|
|
import mock
|
|
from nose.tools import *
|
|
from mock import patch
|
|
from mock import call
|
|
|
|
import kiwi
|
|
|
|
import nose_helper
|
|
|
|
from kiwi.internal_boot_image_task import BootImageTask
|
|
from kiwi.xml_description import XMLDescription
|
|
from kiwi.xml_state import XMLState
|
|
from kiwi.exceptions import *
|
|
|
|
|
|
class TestBootImageTask(object):
|
|
@patch('os.mkdir')
|
|
@patch('os.path.exists')
|
|
def setup(self, mock_os_path, mock_mkdir):
|
|
mock_os_path.return_value = True
|
|
description = XMLDescription('../data/example_config.xml')
|
|
self.xml_state = XMLState(
|
|
description.load()
|
|
)
|
|
|
|
self.manager = mock.Mock()
|
|
self.system = mock.Mock()
|
|
self.setup = mock.Mock()
|
|
self.profile = mock.Mock()
|
|
self.defaults = mock.Mock()
|
|
self.system.setup_repositories = mock.Mock(
|
|
return_value=self.manager
|
|
)
|
|
kiwi.internal_boot_image_task.System = mock.Mock(
|
|
return_value=self.system
|
|
)
|
|
kiwi.internal_boot_image_task.SystemSetup = mock.Mock(
|
|
return_value=self.setup
|
|
)
|
|
kiwi.internal_boot_image_task.Profile = mock.Mock(
|
|
return_value=self.profile
|
|
)
|
|
self.task = BootImageTask(
|
|
self.xml_state, 'some-target-dir'
|
|
)
|
|
self.task.boot_root_directory = 'boot-directory'
|
|
self.task.temp_boot_root_directory = 'temp-boot-directory'
|
|
|
|
@raises(KiwiTargetDirectoryNotFound)
|
|
def test_boot_image_task_raises(self):
|
|
BootImageTask(None, 'target-dir-does-not-exist')
|
|
|
|
@patch('kiwi.defaults.Defaults.get_boot_image_description_path')
|
|
def test_prepare(self, mock_boot_path):
|
|
mock_boot_path.return_value = '../data'
|
|
self.task.prepare()
|
|
self.system.setup_repositories.assert_called_once_with()
|
|
self.system.install_bootstrap.assert_called_once_with(
|
|
self.manager
|
|
)
|
|
self.system.install_system.assert_called_once_with(
|
|
self.manager
|
|
)
|
|
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.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.task.prepare()
|
|
|
|
def test_required_true(self):
|
|
assert self.task.required()
|
|
|
|
def test_required_false(self):
|
|
self.xml_state.build_type.set_boot(None)
|
|
assert self.task.required() == False
|
|
|
|
@patch('kiwi.internal_boot_image_task.Command.run')
|
|
@patch('os.path.exists')
|
|
def test_destructor(self, mock_path, mock_command):
|
|
mock_path.return_value = True
|
|
self.task.__del__()
|
|
assert mock_command.call_args_list == [
|
|
call(['rm', '-r', '-f', 'boot-directory']),
|
|
call(['rm', '-r', '-f', 'temp-boot-directory'])
|
|
]
|
|
|
|
@patch('kiwi.internal_boot_image_task.ArchiveCpio')
|
|
@patch('kiwi.internal_boot_image_task.Compress')
|
|
@patch('kiwi.internal_boot_image_task.Path.create')
|
|
@patch('kiwi.internal_boot_image_task.Path.wipe')
|
|
@patch('kiwi.internal_boot_image_task.Command.run')
|
|
def test_create_initrd(
|
|
self, mock_command, mock_wipe, mock_create, mock_compress, mock_cpio
|
|
):
|
|
mbrid = mock.Mock()
|
|
mbrid.write = mock.Mock()
|
|
cpio = mock.Mock()
|
|
compress = mock.Mock()
|
|
mock_cpio.return_value = cpio
|
|
mock_compress.return_value = compress
|
|
self.task.create_initrd(mbrid)
|
|
mock_command.assert_called_once_with(
|
|
['rsync', '-zav', 'boot-directory/', 'temp-boot-directory']
|
|
)
|
|
mock_cpio.assert_called_once_with(
|
|
self.task.target_dir + '/LimeJeOS-openSUSE-13.2.initrd'
|
|
)
|
|
mock_compress.assert_called_once_with(
|
|
self.task.target_dir + '/LimeJeOS-openSUSE-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=self.task.temp_boot_root_directory,
|
|
exclude=['/var/cache', '/image', '/usr/lib/grub2']
|
|
)
|
|
compress.xz.assert_called_once_with()
|