refactor tests

This commit is contained in:
Divyansh Singh 2023-04-07 01:00:38 +05:30
parent 091d60c226
commit 16d7131fb7
2 changed files with 15 additions and 12 deletions

View File

@ -267,10 +267,9 @@ def run_actions(
'repository': repository_path,
# Deprecated: For backwards compatibility with borgmatic < 1.6.0.
'repositories': ','.join([repo['path'] for repo in location['repositories']]),
'log_file': global_arguments.log_file if global_arguments.log_file else '',
}
hook_context['log_file'] = global_arguments.log_file if global_arguments.log_file else ''
command.execute_hook(
hooks.get('before_actions'),
hooks.get('umask'),

View File

@ -422,25 +422,27 @@ def test_run_actions_runs_rcreate():
)
)
def test_run_actions_adds_log_file_to_hook_context():
flexmock(module).should_receive('add_custom_log_levels')
flexmock(module.command).should_receive('execute_hook')
expected = flexmock()
flexmock(borgmatic.actions.create).should_receive('run_create').with_args(
config_filename=flexmock(),
config_filename=object,
repository={'path': 'repo'},
location={'repositories': []},
storage=flexmock(),
storage=object,
hooks={},
hook_context={'log_file': 'foo'},
local_borg_version=flexmock(),
create_arguments=flexmock(),
global_arguments=flexmock(dry_run=False, log_file='foo'),
hook_context={'repository': 'repo', 'repositories': '', 'log_file': 'foo'},
local_borg_version=object,
create_arguments=object,
global_arguments=object,
dry_run_label='',
local_path=flexmock(),
remote_path=flexmock(),
).once()
local_path=object,
remote_path=object,
).once().and_return(expected)
tuple(
result = tuple(
module.run_actions(
arguments={'global': flexmock(dry_run=False, log_file='foo'), 'create': flexmock()},
config_filename=flexmock(),
@ -455,6 +457,8 @@ def test_run_actions_adds_log_file_to_hook_context():
repository={'path': 'repo'},
)
)
assert result == (expected,)
def test_run_actions_runs_transfer():
flexmock(module).should_receive('add_custom_log_levels')