74 lines
2.3 KiB
Python
74 lines
2.3 KiB
Python
import os
|
|
import errno
|
|
import logging
|
|
from pytest import fixture
|
|
from stat import ST_MODE
|
|
from unittest.mock import patch
|
|
|
|
from kiwi.utils.sync import DataSync
|
|
|
|
|
|
class TestDataSync:
|
|
@fixture(autouse=True)
|
|
def inject_fixtures(self, caplog):
|
|
self._caplog = caplog
|
|
|
|
def setup(self):
|
|
self.sync = DataSync('source_dir', 'target_dir')
|
|
|
|
def setup_method(self, cls):
|
|
self.setup()
|
|
|
|
@patch('kiwi.utils.sync.Command.run')
|
|
@patch('kiwi.utils.sync.DataSync.target_supports_extended_attributes')
|
|
@patch('os.chmod')
|
|
@patch('os.stat')
|
|
def test_sync_data(
|
|
self, mock_stat, mock_chmod, mock_xattr_support, mock_command
|
|
):
|
|
mock_stat.return_value = os.stat('.')
|
|
mock_xattr_support.return_value = False
|
|
with self._caplog.at_level(logging.WARNING):
|
|
self.sync.sync_data(
|
|
options=[
|
|
'--archive', '--hard-links', '--xattrs',
|
|
'--acls', '--one-file-system'
|
|
],
|
|
exclude=['exclude_me']
|
|
)
|
|
mock_command.assert_called_once_with(
|
|
[
|
|
'rsync', '--archive', '--hard-links', '--one-file-system',
|
|
'--exclude', '/exclude_me', 'source_dir', 'target_dir'
|
|
]
|
|
)
|
|
mock_chmod.assert_called_once_with(
|
|
'target_dir', mock_stat.return_value[ST_MODE]
|
|
)
|
|
|
|
@patch('kiwi.utils.sync.Command.run')
|
|
@patch('os.chmod')
|
|
@patch('os.stat')
|
|
def test_sync_data_force_trailing_slash(
|
|
self, mock_stat, mock_chmod, mock_command
|
|
):
|
|
mock_stat.return_value = os.stat('.')
|
|
self.sync.sync_data(force_trailing_slash=True)
|
|
mock_command.assert_called_once_with(
|
|
['rsync', 'source_dir/', 'target_dir']
|
|
)
|
|
|
|
@patch('os.getxattr')
|
|
def test_target_supports_extended_attributes(self, mock_getxattr):
|
|
assert self.sync.target_supports_extended_attributes() is True
|
|
mock_getxattr.assert_called_once_with(
|
|
'target_dir', 'user.mime_type'
|
|
)
|
|
|
|
@patch('os.getxattr')
|
|
def test_target_does_not_support_extended_attributes(self, mock_getxattr):
|
|
effect = OSError()
|
|
effect.errno = errno.ENOTSUP
|
|
mock_getxattr.side_effect = effect
|
|
assert self.sync.target_supports_extended_attributes() is False
|