4fc0d2147d
Change the DiskFormat Factory to be a context manager.
All code using DiskFormat was updated to the following
with statement:
with DiskFormat(...).new as disk_format:
disk_format.some_member()
This is related to Issue #2412
204 lines
7.6 KiB
Python
204 lines
7.6 KiB
Python
import logging
|
|
import os
|
|
import sys
|
|
from unittest.mock import (
|
|
Mock, patch
|
|
)
|
|
from pytest import (
|
|
raises, fixture
|
|
)
|
|
|
|
from ..test_helper import argv_kiwi_tests
|
|
|
|
import kiwi
|
|
from kiwi.tasks.image_resize import ImageResizeTask
|
|
|
|
from kiwi.exceptions import (
|
|
KiwiImageResizeError,
|
|
KiwiSizeError,
|
|
KiwiConfigFileNotFound
|
|
)
|
|
|
|
|
|
class TestImageResizeTask:
|
|
@fixture(autouse=True)
|
|
def inject_fixtures(self, caplog):
|
|
self._caplog = caplog
|
|
|
|
def setup(self):
|
|
sys.argv = [
|
|
sys.argv[0], '--profile', 'vmxSimpleFlavour', '--type', 'oem',
|
|
'image', 'resize', '--target-dir', 'target_dir', '--size', '20g',
|
|
'--root', '../data/root-dir'
|
|
]
|
|
self.abs_root_dir = os.path.abspath('../data/root-dir')
|
|
|
|
kiwi.tasks.image_resize.Help = Mock(
|
|
return_value=Mock()
|
|
)
|
|
|
|
self.firmware = Mock()
|
|
self.firmware.get_partition_table_type = Mock(
|
|
return_value='gpt'
|
|
)
|
|
kiwi.tasks.image_resize.FirmWare = Mock(
|
|
return_value=self.firmware
|
|
)
|
|
|
|
self.task = ImageResizeTask()
|
|
|
|
def setup_method(self, cls):
|
|
self.setup()
|
|
|
|
def teardown(self):
|
|
sys.argv = argv_kiwi_tests
|
|
|
|
def teardown_method(self, cls):
|
|
self.teardown()
|
|
|
|
def _init_command_args(self):
|
|
self.task.command_args = {}
|
|
self.task.command_args['help'] = False
|
|
self.task.command_args['resize'] = False
|
|
self.task.command_args['--target-dir'] = 'target_dir'
|
|
self.task.command_args['--size'] = '42g'
|
|
self.task.command_args['--root'] = '../data/root-dir'
|
|
|
|
def test_process_no_root_directory_specified(self):
|
|
self.task.command_args['--root'] = None
|
|
with raises(KiwiConfigFileNotFound):
|
|
self.task.process()
|
|
|
|
@patch('kiwi.tasks.image_resize.DiskFormat.new')
|
|
def test_process_no_raw_disk_found(self, mock_DiskFormat):
|
|
image_format = Mock()
|
|
image_format.has_raw_disk.return_value = False
|
|
mock_DiskFormat.return_value.__enter__.return_value = image_format
|
|
self._init_command_args()
|
|
self.task.command_args['resize'] = True
|
|
with raises(KiwiImageResizeError):
|
|
self.task.process()
|
|
|
|
@patch('kiwi.tasks.image_resize.DiskFormat.new')
|
|
def test_process_unsupported_size_format(self, mock_DiskFormat):
|
|
image_format = Mock()
|
|
image_format.has_raw_disk.return_value = True
|
|
mock_DiskFormat.return_value.__enter__.return_value = image_format
|
|
self._init_command_args()
|
|
self.task.command_args['--size'] = '20x'
|
|
self.task.command_args['resize'] = True
|
|
with raises(KiwiSizeError):
|
|
self.task.process()
|
|
|
|
@patch('kiwi.tasks.image_resize.LoopDevice')
|
|
@patch('kiwi.tasks.image_resize.DiskFormat.new')
|
|
@patch('kiwi.tasks.image_resize.Partitioner.new')
|
|
def test_process_image_resize_gb(
|
|
self, mock_Partitioner, mock_DiskFormat, mock_LoopDevice
|
|
):
|
|
loop_provider = Mock()
|
|
mock_LoopDevice.return_value.__enter__.return_value = loop_provider
|
|
partitioner = Mock()
|
|
mock_Partitioner.return_value = partitioner
|
|
image_format = Mock()
|
|
image_format.resize_raw_disk.return_value = True
|
|
mock_DiskFormat.return_value.__enter__.return_value = image_format
|
|
self._init_command_args()
|
|
self.task.command_args['resize'] = True
|
|
self.task.process()
|
|
loop_provider.create.assert_called_once_with(overwrite=False)
|
|
partitioner.resize_table.assert_called_once_with()
|
|
image_format.resize_raw_disk.assert_called_once_with(
|
|
42 * 1024 * 1024 * 1024
|
|
)
|
|
image_format.create_image_format.assert_called_once_with()
|
|
|
|
@patch('kiwi.tasks.image_resize.LoopDevice')
|
|
@patch('kiwi.tasks.image_resize.DiskFormat.new')
|
|
@patch('kiwi.tasks.image_resize.Partitioner.new')
|
|
def test_process_image_resize_mb(
|
|
self, mock_Partitioner, mock_DiskFormat, mock_LoopDevice
|
|
):
|
|
loop_provider = Mock()
|
|
mock_LoopDevice.return_value.__enter__.return_value = loop_provider
|
|
partitioner = Mock()
|
|
mock_Partitioner.return_value = partitioner
|
|
image_format = Mock()
|
|
image_format.resize_raw_disk.return_value = True
|
|
mock_DiskFormat.return_value.__enter__.return_value = image_format
|
|
self._init_command_args()
|
|
self.task.command_args['resize'] = True
|
|
self.task.command_args['--size'] = '42m'
|
|
self.task.process()
|
|
loop_provider.create.assert_called_once_with(overwrite=False)
|
|
partitioner.resize_table.assert_called_once_with()
|
|
image_format.resize_raw_disk.assert_called_once_with(
|
|
42 * 1024 * 1024
|
|
)
|
|
image_format.create_image_format.assert_called_once_with()
|
|
|
|
@patch('kiwi.tasks.image_resize.LoopDevice')
|
|
@patch('kiwi.tasks.image_resize.DiskFormat.new')
|
|
@patch('kiwi.tasks.image_resize.Partitioner.new')
|
|
def test_process_image_resize_bytes(
|
|
self, mock_Partitioner, mock_DiskFormat, mock_LoopDevice
|
|
):
|
|
loop_provider = Mock()
|
|
mock_LoopDevice.return_value.__enter__.return_value = loop_provider
|
|
partitioner = Mock()
|
|
mock_Partitioner.return_value = partitioner
|
|
image_format = Mock()
|
|
image_format.resize_raw_disk.return_value = True
|
|
mock_DiskFormat.return_value.__enter__.return_value = image_format
|
|
self._init_command_args()
|
|
self.task.command_args['resize'] = True
|
|
self.task.command_args['--size'] = '42'
|
|
self.task.process()
|
|
loop_provider.create.assert_called_once_with(overwrite=False)
|
|
partitioner.resize_table.assert_called_once_with()
|
|
image_format.resize_raw_disk.assert_called_once_with(
|
|
42
|
|
)
|
|
image_format.create_image_format.assert_called_once_with()
|
|
|
|
@patch('kiwi.tasks.image_resize.LoopDevice')
|
|
@patch('kiwi.tasks.image_resize.DiskFormat.new')
|
|
@patch('kiwi.tasks.image_resize.Partitioner.new')
|
|
def test_process_image_resize_not_needed(
|
|
self, mock_Partitioner, mock_DiskFormat, mock_LoopDevice
|
|
):
|
|
loop_provider = Mock()
|
|
mock_LoopDevice.return_value.__enter__.return_value = loop_provider
|
|
partitioner = Mock()
|
|
mock_Partitioner.return_value = partitioner
|
|
image_format = Mock()
|
|
image_format.resize_raw_disk.return_value = False
|
|
mock_DiskFormat.return_value.__enter__.return_value = image_format
|
|
self._init_command_args()
|
|
self.task.command_args['resize'] = True
|
|
self.task.command_args['--size'] = '42'
|
|
with self._caplog.at_level(logging.INFO):
|
|
self.task.process()
|
|
loop_provider.create.assert_called_once_with(overwrite=False)
|
|
partitioner.resize_table.assert_called_once_with()
|
|
image_format.resize_raw_disk.assert_called_once_with(
|
|
42
|
|
)
|
|
assert 'Loading XML description' in self._caplog.text
|
|
assert '--> loaded {0}'.format(
|
|
os.sep.join([self.abs_root_dir, 'image', 'config.xml'])
|
|
) in self._caplog.text
|
|
assert '--> Selected build type: oem' in self._caplog.text
|
|
assert '--> Selected profiles: vmxSimpleFlavour' in self._caplog.text
|
|
assert 'Resizing raw disk to 42 bytes' in self._caplog.text
|
|
assert 'Raw disk is already at 42 bytes' in self._caplog.text
|
|
|
|
def test_process_image_resize_help(self):
|
|
self._init_command_args()
|
|
self.task.command_args['help'] = True
|
|
self.task.command_args['resize'] = True
|
|
self.task.process()
|
|
self.task.manual.show.assert_called_once_with(
|
|
'kiwi::image::resize'
|
|
)
|