kiwi-el8/test/unit/system_kernel_test.py
Marcus Schäfer c3bad3f8f2
Fixup kernel name setup in disk builder
The Kernel instance is the only correct place to ask
for the kernel name. This class has the responsibility
to know information about the selected kernel. The
additional arch based assumption on the kernel name
in the disk builder were wrong. This fixes bnc#1011936
2016-11-24 10:39:10 +01:00

103 lines
3.3 KiB
Python

from mock import patch
import mock
from .test_helper import raises
from collections import namedtuple
from kiwi.exceptions import KiwiKernelLookupError
from kiwi.system.kernel import Kernel
class TestKernel(object):
@patch('os.listdir')
def setup(self, mock_listdir):
mock_listdir.return_value = ['1.2.3-default']
self.kernel = Kernel('root-dir')
assert self.kernel.kernel_names == [
'vmlinux',
'vmlinuz',
'zImage',
'uImage-1.2.3-default',
'Image-1.2.3-default',
'zImage-1.2.3-default',
'vmlinuz-1.2.3-default',
'vmlinux-1.2.3-default',
'image-1.2.3-default'
]
@raises(KiwiKernelLookupError)
def test_get_kernel_raises_if_no_kernel_found(self):
self.kernel.kernel_names = []
self.kernel.get_kernel(raise_on_not_found=True)
@patch('os.path.exists')
@patch('os.path.realpath')
@patch('kiwi.command.Command.run')
def test_get_kernel(self, mock_run, mock_realpath, mock_os):
run = namedtuple(
'run', ['output']
)
result = run(output='42')
mock_os.return_value = True
mock_run.return_value = result
mock_realpath.return_value = 'vmlinux-realpath'
data = self.kernel.get_kernel()
mock_run.assert_called_once_with(
command=['kversion', 'root-dir/boot/vmlinux'],
raise_on_error=False
)
assert data.filename == 'root-dir/boot/vmlinux'
assert data.version == '42'
assert data.name == 'vmlinux-realpath'
@patch('os.path.exists')
@patch('kiwi.command.Command.run')
def test_get_kernel_no_version(self, mock_run, mock_os):
run = namedtuple(
'run', ['output']
)
result = run(output=None)
mock_os.return_value = True
mock_run.return_value = result
data = self.kernel.get_kernel()
mock_run.assert_called_once_with(
command=['kversion', 'root-dir/boot/vmlinux'],
raise_on_error=False
)
assert data.filename == 'root-dir/boot/vmlinux'
assert data.version == 'no-version-found'
@patch('os.path.exists')
def test_get_xen_hypervisor(self, mock_os):
mock_os.return_value = True
data = self.kernel.get_xen_hypervisor()
assert data.filename == 'root-dir/boot/xen.gz'
assert data.name == 'xen.gz'
@patch('kiwi.system.kernel.Kernel.get_kernel')
@patch('kiwi.command.Command.run')
def test_copy_kernel(self, mock_run, mock_get_kernel):
result = mock.MagicMock()
result.version = '42'
result.filename = 'kernel'
mock_get_kernel.return_value = result
self.kernel.copy_kernel('target-dir')
mock_run.assert_called_once_with(
['cp', 'kernel', 'target-dir/kernel-42.kernel']
)
@patch('kiwi.system.kernel.Kernel.get_xen_hypervisor')
@patch('kiwi.command.Command.run')
def test_copy_xen_hypervisor(self, mock_run, mock_get_xen):
result = mock.MagicMock()
result.name = 'xen.gz'
result.filename = 'some/xen.gz'
mock_get_xen.return_value = result
self.kernel.copy_xen_hypervisor('target-dir')
mock_run.assert_called_once_with(
['cp', 'some/xen.gz', 'target-dir/hypervisor-xen.gz']
)