Files
2026-05-22 12:19:34 -07:00

84 lines
2.7 KiB
Python

import logging
from flexmock import flexmock
from borgmatic.borg import umount as module
from ..test_verbosity import insert_logging_mock
def insert_execute_command_mock(
command,
borg_local_path='borg',
working_directory=None,
borg_exit_codes=None,
):
flexmock(module.borgmatic.config.paths).should_receive('get_working_directory').and_return(
working_directory,
)
flexmock(module).should_receive('execute_command').with_args(
command,
working_directory=working_directory,
borg_local_path=borg_local_path,
borg_exit_codes=borg_exit_codes,
).once()
def test_unmount_archive_calls_borg_with_required_parameters():
insert_execute_command_mock(('borg', 'umount', '--log-json', '/mnt'))
insert_logging_mock(logging.WARNING)
module.unmount_archive(config={}, mount_point='/mnt')
def test_unmount_archive_with_log_info_calls_borg_with_info_parameter():
insert_execute_command_mock(('borg', 'umount', '--log-json', '--info', '/mnt'))
insert_logging_mock(logging.INFO)
module.unmount_archive(config={}, mount_point='/mnt')
def test_unmount_archive_with_log_debug_calls_borg_with_debug_parameters():
insert_execute_command_mock(('borg', 'umount', '--log-json', '--debug', '--show-rc', '/mnt'))
insert_logging_mock(logging.DEBUG)
module.unmount_archive(config={}, mount_point='/mnt')
def test_unmount_archive_calls_borg_with_extra_borg_options():
insert_execute_command_mock(
('borg', 'umount', '--log-json', '--extra', 'value with space', '/mnt'),
borg_local_path='borg',
)
insert_logging_mock(logging.WARNING)
module.unmount_archive(
config={'extra_borg_options': {'umount': '--extra "value with space"'}}, mount_point='/mnt'
)
def test_unmount_archive_calls_borg_with_local_path():
insert_execute_command_mock(('borg1', 'umount', '--log-json', '/mnt'), borg_local_path='borg1')
insert_logging_mock(logging.WARNING)
module.unmount_archive(config={}, mount_point='/mnt', local_path='borg1')
def test_unmount_archive_calls_borg_with_exit_codes():
borg_exit_codes = flexmock()
insert_execute_command_mock(
('borg', 'umount', '--log-json', '/mnt'), borg_exit_codes=borg_exit_codes
)
insert_logging_mock(logging.WARNING)
module.unmount_archive(config={'borg_exit_codes': borg_exit_codes}, mount_point='/mnt')
def test_unmount_archive_calls_borg_with_working_directory():
insert_execute_command_mock(
('borg', 'umount', '--log-json', '/mnt'), working_directory='/working/dir'
)
insert_logging_mock(logging.WARNING)
module.unmount_archive(config={'working_directory': '/working/dir'}, mount_point='/mnt')