426221ce34
Extend the ArchiveTar and Compress classes such that their archiving and compression methods returns the result file name after the operation
92 lines
3.1 KiB
Python
92 lines
3.1 KiB
Python
from mock import patch
|
|
|
|
import mock
|
|
|
|
from .test_helper import raises
|
|
|
|
from kiwi.utils.compress import Compress
|
|
from kiwi.exceptions import (
|
|
KiwiFileNotFound,
|
|
KiwiCompressionFormatUnknown
|
|
)
|
|
|
|
|
|
class TestCompress(object):
|
|
@patch('os.path.exists')
|
|
def setup(self, mock_exists):
|
|
mock_exists.return_value = True
|
|
self.compress = Compress('some-file', True)
|
|
|
|
@raises(KiwiFileNotFound)
|
|
def test_source_file_not_found(self):
|
|
Compress('some-file')
|
|
|
|
@patch('kiwi.command.Command.run')
|
|
def test_xz(self, mock_command):
|
|
assert self.compress.xz() == 'some-file.xz'
|
|
mock_command.assert_called_once_with(
|
|
[
|
|
'xz', '-f', '--threads=0', '--keep',
|
|
'some-file'
|
|
]
|
|
)
|
|
assert self.compress.compressed_filename == 'some-file.xz'
|
|
|
|
@patch('kiwi.command.Command.run')
|
|
def test_xz_with_custom_options(self, mock_command):
|
|
assert self.compress.xz(options=['foo', 'bar']) == 'some-file.xz'
|
|
mock_command.assert_called_once_with(
|
|
[
|
|
'xz', '-f', 'foo', 'bar', '--keep',
|
|
'some-file'
|
|
]
|
|
)
|
|
assert self.compress.compressed_filename == 'some-file.xz'
|
|
|
|
@patch('kiwi.command.Command.run')
|
|
def test_gzip(self, mock_command):
|
|
assert self.compress.gzip() == 'some-file.gz'
|
|
mock_command.assert_called_once_with(
|
|
['gzip', '-f', '-9', '--keep', 'some-file']
|
|
)
|
|
assert self.compress.compressed_filename == 'some-file.gz'
|
|
|
|
@patch('kiwi.command.Command.run')
|
|
@patch('kiwi.utils.compress.NamedTemporaryFile')
|
|
@patch('kiwi.utils.compress.Compress.get_format')
|
|
def test_uncompress(self, mock_format, mock_temp, mock_command):
|
|
mock_format.return_value = 'xz'
|
|
self.compress.uncompress()
|
|
mock_command.assert_called_once_with(
|
|
['xz', '-d', 'some-file']
|
|
)
|
|
assert self.compress.uncompressed_filename == 'some-file'
|
|
|
|
@patch('kiwi.command.Command.run')
|
|
@patch('kiwi.utils.compress.NamedTemporaryFile')
|
|
@patch('kiwi.utils.compress.Compress.get_format')
|
|
def test_uncompress_temporary(self, mock_format, mock_temp, mock_command):
|
|
tempfile = mock.Mock()
|
|
tempfile.name = 'tempfile'
|
|
mock_temp.return_value = tempfile
|
|
mock_format.return_value = 'xz'
|
|
self.compress.uncompress(temporary=True)
|
|
mock_command.assert_called_once_with(
|
|
['bash', '-c', 'xz -c -d some-file > tempfile']
|
|
)
|
|
assert self.compress.uncompressed_filename == 'tempfile'
|
|
|
|
@raises(KiwiCompressionFormatUnknown)
|
|
@patch('kiwi.utils.compress.Compress.get_format')
|
|
def test_uncompress_unknown_format(self, mock_format):
|
|
mock_format.return_value = None
|
|
self.compress.uncompress()
|
|
|
|
@patch('kiwi.path.Path.which')
|
|
def test_get_format(self, mock_which):
|
|
mock_which.return_value = 'ziptool'
|
|
xz = Compress('../data/xz_data.xz')
|
|
assert xz.get_format() == 'xz'
|
|
gzip = Compress('../data/gz_data.gz')
|
|
assert gzip.get_format() == 'gzip'
|