cc152d0a57
Due to the workaround to lookup the gzip compressed variant of a zImage to find its version, it's required to find the full name of the zImage in order to rematch the gzip variant. Looking at the zImage link will make this to fail because there is no vmlinux link
123 lines
4.1 KiB
Python
123 lines
4.1 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',
|
|
'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('os.path.realpath')
|
|
@patch('kiwi.command.Command.run')
|
|
def test_get_kernel_from_zImage(self, mock_run, mock_realpath, mock_os):
|
|
self.kernel.kernel_names = ['zImage-1.2.3-default']
|
|
run = namedtuple(
|
|
'run', ['output']
|
|
)
|
|
result = run(output='42')
|
|
mock_os.return_value = True
|
|
mock_run.return_value = result
|
|
mock_realpath.return_value = 'zImage-realpath'
|
|
data = self.kernel.get_kernel()
|
|
mock_run.assert_called_once_with(
|
|
command=['kversion', 'root-dir/boot/vmlinux-1.2.3-default.gz'],
|
|
raise_on_error=False
|
|
)
|
|
assert data.filename == 'root-dir/boot/zImage-1.2.3-default'
|
|
assert data.version == '42'
|
|
assert data.name == 'zImage-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']
|
|
)
|