4b3a105026
This commit relocates unit tests to a folder structure that matches the source code structure. Fixes #1128
173 lines
6.6 KiB
Python
173 lines
6.6 KiB
Python
import io
|
|
from collections import namedtuple
|
|
from mock import (
|
|
patch, Mock, MagicMock
|
|
)
|
|
from pytest import raises
|
|
|
|
from kiwi.exceptions import (
|
|
KiwiTargetDirectoryNotFound,
|
|
KiwiBootImageDumpError,
|
|
KiwiConfigFileNotFound,
|
|
KiwiDiskBootImageError
|
|
)
|
|
|
|
from kiwi.boot.image.base import BootImageBase
|
|
|
|
|
|
class TestBootImageBase:
|
|
@patch('kiwi.boot.image.base.os.path.exists')
|
|
@patch('platform.machine')
|
|
def setup(self, mock_machine, mock_exists):
|
|
mock_machine.return_value = 'x86_64'
|
|
self.boot_xml_state = Mock()
|
|
self.xml_state = Mock()
|
|
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.get_image_version = Mock(
|
|
return_value='1.2.3'
|
|
)
|
|
self.xml_state.build_type.get_boot = Mock(
|
|
return_value='oemboot/suse-13.2'
|
|
)
|
|
mock_exists.return_value = True
|
|
self.boot_image = BootImageBase(
|
|
self.xml_state, 'some-target-dir', 'system-directory'
|
|
)
|
|
|
|
def test_boot_image_raises(self):
|
|
with raises(KiwiTargetDirectoryNotFound):
|
|
BootImageBase(
|
|
self.xml_state, 'target-dir-does-not-exist', 'some-root-dir'
|
|
)
|
|
|
|
def test_prepare(self):
|
|
with raises(NotImplementedError):
|
|
self.boot_image.prepare()
|
|
|
|
def test_create_initrd(self):
|
|
with raises(NotImplementedError):
|
|
self.boot_image.create_initrd()
|
|
|
|
@patch('pickle.dump')
|
|
def test_dump_error(self, mock_dump):
|
|
mock_dump.side_effect = Exception
|
|
with patch('builtins.open'):
|
|
with raises(KiwiBootImageDumpError):
|
|
self.boot_image.dump('filename')
|
|
|
|
@patch('pickle.dump')
|
|
@patch('kiwi.boot.image.base.BootImageBase.disable_cleanup')
|
|
def test_dump(self, mock_disable_cleanup, mock_dump):
|
|
with patch('builtins.open', create=True) as mock_open:
|
|
mock_open.return_value = MagicMock(spec=io.IOBase)
|
|
file_handle = mock_open.return_value.__enter__.return_value
|
|
self.boot_image.dump('filename')
|
|
mock_open.assert_called_once_with('filename', 'wb')
|
|
mock_dump.assert_called_once_with(
|
|
self.boot_image, file_handle
|
|
)
|
|
mock_disable_cleanup.assert_called_once_with()
|
|
|
|
def test_disable_cleanup(self):
|
|
self.boot_image.disable_cleanup()
|
|
assert self.boot_image.call_destructor is False
|
|
|
|
def test_enable_cleanup(self):
|
|
self.boot_image.enable_cleanup()
|
|
assert self.boot_image.call_destructor is True
|
|
|
|
@patch('os.listdir')
|
|
def test_is_prepared(self, mock_listdir):
|
|
mock_listdir.return_value = []
|
|
assert self.boot_image.is_prepared() is False
|
|
mock_listdir.return_value = ['a', 'b', 'c']
|
|
assert self.boot_image.is_prepared() is True
|
|
|
|
@patch('kiwi.boot.image.base.XMLState.copy_strip_sections')
|
|
def test_import_system_description_elements(self, mock_strip):
|
|
self.boot_image.import_system_description_elements()
|
|
assert self.xml_state.copy_displayname.called
|
|
assert self.xml_state.copy_name.called
|
|
assert self.xml_state.copy_repository_sections.called
|
|
assert self.xml_state.copy_drivers_sections.called
|
|
assert mock_strip.called
|
|
assert self.xml_state.copy_strip_sections.called
|
|
assert self.xml_state.copy_preferences_subsections.called
|
|
assert self.xml_state.copy_bootincluded_packages.called
|
|
assert self.xml_state.copy_bootincluded_archives.called
|
|
assert self.xml_state.copy_bootdelete_packages.called
|
|
assert self.xml_state.copy_build_type_attributes.called
|
|
assert self.xml_state.copy_systemdisk_section.called
|
|
assert self.xml_state.copy_machine_section.called
|
|
assert self.xml_state.copy_oemconfig_section.called
|
|
|
|
def test_get_boot_description_directory(self):
|
|
assert self.boot_image.get_boot_description_directory() == \
|
|
'/usr/share/kiwi/custom_boot/oemboot/suse-13.2'
|
|
|
|
@patch('kiwi.boot.image.base.BootImageBase.get_boot_description_directory')
|
|
def test_load_boot_xml_description(self, mock_boot_dir):
|
|
mock_boot_dir.return_value = None
|
|
with raises(KiwiConfigFileNotFound):
|
|
self.boot_image.load_boot_xml_description()
|
|
|
|
@patch('kiwi.boot.image.base.Kernel')
|
|
def test_get_boot_names_raises(self, mock_Kernel):
|
|
kernel = Mock()
|
|
mock_Kernel.return_value = kernel
|
|
kernel.get_kernel.return_value = None
|
|
with raises(KiwiDiskBootImageError):
|
|
self.boot_image.get_boot_names()
|
|
|
|
@patch('kiwi.boot.image.base.Kernel')
|
|
@patch('kiwi.boot.image.base.Path.which')
|
|
@patch('kiwi.boot.image.base.log.warning')
|
|
def test_get_boot_names(
|
|
self, mock_warning, mock_Path_which, mock_Kernel
|
|
):
|
|
boot_names_type = namedtuple(
|
|
'boot_names_type', ['kernel_name', 'initrd_name']
|
|
)
|
|
mock_Path_which.return_value = 'dracut'
|
|
kernel = Mock()
|
|
kernel_info = Mock()
|
|
kernel_info.name = 'kernel_name'
|
|
kernel_info.version = 'kernel_version'
|
|
kernel.get_kernel.return_value = kernel_info
|
|
mock_Kernel.return_value = kernel
|
|
|
|
with patch('builtins.open', create=True) as mock_open:
|
|
mock_open.return_value = MagicMock(spec=io.IOBase)
|
|
file_handle = mock_open.return_value.__enter__.return_value
|
|
file_handle.read.return_value = 'outfile="/boot/initrd-$kernel"'
|
|
assert self.boot_image.get_boot_names() == boot_names_type(
|
|
kernel_name='kernel_name',
|
|
initrd_name='initrd-kernel_version'
|
|
)
|
|
|
|
with patch('builtins.open', create=True) as mock_open:
|
|
mock_open.return_value = MagicMock(spec=io.IOBase)
|
|
file_handle = mock_open.return_value.__enter__.return_value
|
|
file_handle.read.return_value = 'outfile="foo"'
|
|
assert self.boot_image.get_boot_names() == boot_names_type(
|
|
kernel_name='kernel_name',
|
|
initrd_name='initramfs-kernel_version.img'
|
|
)
|
|
|
|
self.xml_state.get_initrd_system.return_value = 'kiwi'
|
|
mock_Path_which.return_value = None
|
|
assert self.boot_image.get_boot_names() == boot_names_type(
|
|
kernel_name='kernel_name',
|
|
initrd_name='initrd-kernel_version'
|
|
)
|
|
|
|
def test_noop_methods(self):
|
|
self.boot_image.include_module('module')
|
|
self.boot_image.omit_module('module')
|
|
self.boot_image.write_system_config_file({'config_key': 'value'})
|