7a5ab282b6
The former implementation read the runtime config file every time a new RuntimeConfig instance was created. The runtime config is however static during the runtime of an image build process and not supposed to change. Therefore the file should be read in once and any new instance of RuntimConfig should just use what has been read in at the first invocation. This commit refactors the RuntimeConfig class to hold an application global RUNTIME_CONFIG variable and allows to reread the config on explicit request.
126 lines
4.2 KiB
Python
126 lines
4.2 KiB
Python
from mock import (
|
|
patch, Mock
|
|
)
|
|
from pytest import raises
|
|
|
|
from kiwi.storage.subformat.base import DiskFormatBase
|
|
|
|
import kiwi
|
|
|
|
from kiwi.exceptions import (
|
|
KiwiFormatSetupError,
|
|
KiwiResizeRawDiskError
|
|
)
|
|
|
|
|
|
class TestDiskFormatBase:
|
|
@patch('platform.machine')
|
|
def setup(self, mock_machine):
|
|
mock_machine.return_value = 'x86_64'
|
|
xml_data = Mock()
|
|
xml_data.get_name = Mock(
|
|
return_value='some-disk-image'
|
|
)
|
|
self.xml_state = Mock()
|
|
self.xml_state.xml_data = xml_data
|
|
self.xml_state.get_image_version = Mock(
|
|
return_value='1.2.3'
|
|
)
|
|
self.runtime_config = Mock()
|
|
self.runtime_config.get_bundle_compression.return_value = True
|
|
kiwi.storage.subformat.base.RuntimeConfig = Mock(
|
|
return_value=self.runtime_config
|
|
)
|
|
self.disk_format = DiskFormatBase(
|
|
self.xml_state, 'root_dir', 'target_dir'
|
|
)
|
|
|
|
def test_create_image_format(self):
|
|
with raises(NotImplementedError):
|
|
self.disk_format.create_image_format()
|
|
|
|
def test_get_target_file_path_for_format_invalid_format(self):
|
|
with raises(KiwiFormatSetupError):
|
|
self.disk_format.get_target_file_path_for_format('foo')
|
|
|
|
def test_post_init(self):
|
|
self.disk_format.post_init({'option': 'unhandled'})
|
|
assert self.disk_format.custom_args == {}
|
|
|
|
def test_get_qemu_option_list(self):
|
|
custom_args = {
|
|
'subformat=format': None,
|
|
'adapter_type=type': None
|
|
}
|
|
assert self.disk_format.get_qemu_option_list(custom_args) == [
|
|
'-o', 'adapter_type=type', '-o', 'subformat=format'
|
|
]
|
|
|
|
def test_get_target_file_path_for_format(self):
|
|
assert self.disk_format.get_target_file_path_for_format('vhd') == \
|
|
'target_dir/some-disk-image.x86_64-1.2.3.vhd'
|
|
|
|
def test_store_to_result(self):
|
|
result = Mock()
|
|
self.disk_format.image_format = 'qcow2'
|
|
self.disk_format.store_to_result(result)
|
|
result.add.assert_called_once_with(
|
|
compress=True,
|
|
filename='target_dir/some-disk-image.x86_64-1.2.3.qcow2',
|
|
key='disk_format_image',
|
|
shasum=True,
|
|
use_for_bundle=True
|
|
)
|
|
|
|
@patch('os.path.getsize')
|
|
def test_resize_raw_disk_raises_on_shrink_disk(self, mock_getsize):
|
|
mock_getsize.return_value = 42
|
|
with raises(KiwiResizeRawDiskError):
|
|
self.disk_format.resize_raw_disk(10)
|
|
|
|
@patch('os.path.getsize')
|
|
@patch('kiwi.storage.subformat.base.Command.run')
|
|
def test_resize_raw_disk(self, mock_command, mock_getsize):
|
|
mock_getsize.return_value = 42
|
|
assert self.disk_format.resize_raw_disk(1024) is True
|
|
mock_command.assert_called_once_with(
|
|
[
|
|
'qemu-img', 'resize',
|
|
'target_dir/some-disk-image.x86_64-1.2.3.raw', '1024'
|
|
]
|
|
)
|
|
|
|
@patch('os.path.getsize')
|
|
@patch('kiwi.storage.subformat.base.Command.run')
|
|
def test_resize_raw_disk_append(self, mock_command, mock_getsize):
|
|
mock_getsize.return_value = 42
|
|
assert self.disk_format.resize_raw_disk(1024, append=True) is True
|
|
mock_command.assert_called_once_with(
|
|
[
|
|
'qemu-img', 'resize',
|
|
'target_dir/some-disk-image.x86_64-1.2.3.raw', '+1024'
|
|
]
|
|
)
|
|
|
|
@patch('os.path.getsize')
|
|
def test_resize_raw_disk_same_size(self, mock_getsize):
|
|
mock_getsize.return_value = 42
|
|
assert self.disk_format.resize_raw_disk(42) is False
|
|
|
|
@patch('os.path.exists')
|
|
def test_has_raw_disk(self, mock_exists):
|
|
mock_exists.return_value = True
|
|
assert self.disk_format.has_raw_disk() is True
|
|
mock_exists.assert_called_once_with(
|
|
'target_dir/some-disk-image.x86_64-1.2.3.raw'
|
|
)
|
|
|
|
@patch('kiwi.storage.subformat.base.Path.wipe')
|
|
@patch('os.path.exists')
|
|
def test_destructor(self, mock_exists, mock_wipe):
|
|
mock_exists.return_value = True
|
|
self.disk_format.temp_image_dir = 'tmpdir'
|
|
self.disk_format.__del__()
|
|
self.disk_format.temp_image_dir = None
|
|
mock_wipe.assert_called_once_with('tmpdir')
|