f71f7aea18
RootImport has been refactored so the image is kept with a known name that can be obtained with the Defaults class.
96 lines
3.1 KiB
Python
96 lines
3.1 KiB
Python
import mock
|
|
from mock import patch
|
|
from mock import call
|
|
|
|
from .test_helper import raises
|
|
|
|
from kiwi.system.uri import Uri
|
|
from kiwi.system.root_import.docker import RootImportDocker
|
|
from kiwi.exceptions import KiwiRootImportError
|
|
|
|
|
|
class TestRootImportDocker(object):
|
|
@patch('os.path.exists')
|
|
def setup(self, mock_path):
|
|
mock_path.return_value = True
|
|
self.docker_import = RootImportDocker(
|
|
'root_dir', Uri('file:///image.tar.xz')
|
|
)
|
|
assert self.docker_import.image_file == '/image.tar.xz'
|
|
|
|
@patch('os.path.exists')
|
|
@raises(KiwiRootImportError)
|
|
def test_failed_init(self, mock_path):
|
|
mock_path.return_value = False
|
|
RootImportDocker(
|
|
'root_dir', Uri('file:///image.tar.xz')
|
|
)
|
|
|
|
@patch('kiwi.system.root_import.base.shutil.copy')
|
|
@patch('kiwi.system.root_import.base.Checksum')
|
|
@patch('kiwi.system.root_import.base.Path.create')
|
|
@patch('kiwi.system.root_import.docker.Compress')
|
|
@patch('kiwi.system.root_import.docker.Command.run')
|
|
@patch('kiwi.system.root_import.docker.DataSync')
|
|
@patch('kiwi.system.root_import.docker.mkdtemp')
|
|
def test_sync_data(
|
|
self, mock_mkdtemp, mock_sync, mock_run,
|
|
mock_compress, mock_path, mock_md5, mock_copy
|
|
):
|
|
uncompress = mock.Mock()
|
|
uncompress.uncompressed_filename = 'tmp_uncompressed'
|
|
mock_compress.return_value = uncompress
|
|
|
|
tmpdirs = ['kiwi_unpack_dir', 'kiwi_layout_dir']
|
|
|
|
def call_mkdtemp(prefix):
|
|
return tmpdirs.pop()
|
|
|
|
mock_mkdtemp.side_effect = call_mkdtemp
|
|
|
|
sync = mock.Mock()
|
|
mock_sync.return_value = sync
|
|
|
|
md5 = mock.Mock()
|
|
mock_md5.return_value = mock.Mock()
|
|
|
|
self.docker_import.sync_data()
|
|
|
|
mock_compress.assert_called_once_with('/image.tar.xz')
|
|
uncompress.uncompress.assert_called_once_with(True)
|
|
|
|
assert mock_run.call_args_list == [
|
|
call([
|
|
'skopeo', 'copy', 'docker-archive:tmp_uncompressed',
|
|
'oci:kiwi_layout_dir'
|
|
]),
|
|
call([
|
|
'umoci', 'unpack', '--image',
|
|
'kiwi_layout_dir', 'kiwi_unpack_dir'
|
|
])
|
|
]
|
|
|
|
mock_sync.assert_called_once_with(
|
|
'kiwi_unpack_dir/rootfs/', 'root_dir/'
|
|
)
|
|
sync.sync_data.assert_called_once_with(
|
|
options=['-a', '-H', '-X', '-A']
|
|
)
|
|
mock_md5.assert_called_once_with('root_dir/image/imported_root')
|
|
md5.md5.called_once_with('root_dir/image/imported_root.md5')
|
|
mock_copy.assert_called_once_with(
|
|
'tmp_uncompressed', 'root_dir/image/imported_root'
|
|
)
|
|
mock_path.assert_called_once_with('root_dir/image')
|
|
|
|
@patch('kiwi.system.root_import.docker.Path.wipe')
|
|
def test_del(self, mock_path):
|
|
self.docker_import.oci_layout_dir = 'layout_dir'
|
|
self.docker_import.oci_unpack_dir = 'unpack_dir'
|
|
self.docker_import.uncompressed_image = 'uncompressed_file'
|
|
self.docker_import.__del__()
|
|
|
|
assert mock_path.call_args_list == [
|
|
call('layout_dir'), call('unpack_dir'), call('uncompressed_file')
|
|
]
|