kiwi-el8/test/unit/filesystem/squashfs_test.py
Alexandre Detiste fb69627ad3
Use unittest.mock from core python everywhere
mock was an independent module that has been merged into the Python standard library.
2024-02-18 22:15:30 +01:00

77 lines
2.6 KiB
Python

from unittest.mock import patch
import unittest.mock as mock
from kiwi.defaults import Defaults
from kiwi.filesystem.squashfs import FileSystemSquashFs
class TestFileSystemSquashfs:
@patch('os.path.exists')
def setup(self, mock_exists):
mock_exists.return_value = True
self.squashfs = FileSystemSquashFs(mock.Mock(), 'root_dir')
@patch('os.path.exists')
def setup_method(self, cls, mock_exists):
self.setup()
@patch('kiwi.filesystem.squashfs.Command.run')
def test_create_on_file(self, mock_command):
Defaults.set_platform_name('x86_64')
self.squashfs.create_on_file('myimage', 'label')
mock_command.assert_called_once_with(
[
'mksquashfs', 'root_dir', 'myimage', '-noappend',
'-b', '1M', '-comp', 'xz', '-Xbcj', 'x86'
]
)
@patch('kiwi.filesystem.squashfs.Command.run')
def test_create_on_file_exclude_data(self, mock_command):
Defaults.set_platform_name('ppc64le')
self.squashfs.create_on_file('myimage', 'label', ['foo'])
mock_command.assert_called_once_with(
[
'mksquashfs', 'root_dir', 'myimage', '-noappend', '-b', '1M',
'-comp', 'xz', '-Xbcj', 'powerpc', '-wildcards', '-e', 'foo'
]
)
@patch('kiwi.filesystem.squashfs.Command.run')
def test_create_on_file_unkown_arch(self, mock_command):
Defaults.set_platform_name('aarch64')
self.squashfs.create_on_file('myimage', 'label')
mock_command.assert_called_once_with(
[
'mksquashfs', 'root_dir', 'myimage',
'-noappend', '-b', '1M', '-comp', 'xz'
]
)
@patch('kiwi.filesystem.squashfs.Command.run')
def test_create_on_file_gzip(self, mock_command):
self.squashfs.custom_args = {
'compression': 'gzip', 'create_options': []
}
self.squashfs.create_on_file('myimage', 'label')
mock_command.assert_called_once_with(
[
'mksquashfs', 'root_dir', 'myimage',
'-noappend', '-b', '1M', '-comp', 'gzip'
]
)
@patch('kiwi.filesystem.squashfs.Command.run')
def test_create_on_file_no_compression(self, mock_command):
self.squashfs.custom_args = {
'compression': 'uncompressed', 'create_options': []
}
self.squashfs.create_on_file('myimage', 'label')
mock_command.assert_called_once_with(
[
'mksquashfs', 'root_dir', 'myimage', '-noappend',
'-b', '1M', '-noI', '-noD', '-noF', '-noX'
]
)