borgmatic/tests/unit/test_hook.py

53 lines
1.8 KiB
Python
Raw Normal View History

import logging
from flexmock import flexmock
2019-06-13 19:09:16 +02:00
from borgmatic import hook as module
def test_execute_hook_invokes_each_command():
flexmock(module.execute).should_receive('execute_command').with_args(
[':'], output_log_level=logging.WARNING, shell=True
).once()
module.execute_hook([':'], None, 'config.yaml', 'pre-backup', dry_run=False)
2017-10-26 07:32:06 +02:00
def test_execute_hook_with_multiple_commands_invokes_each_command():
flexmock(module.execute).should_receive('execute_command').with_args(
[':'], output_log_level=logging.WARNING, shell=True
).once()
flexmock(module.execute).should_receive('execute_command').with_args(
['true'], output_log_level=logging.WARNING, shell=True
).once()
2017-10-26 07:32:06 +02:00
module.execute_hook([':', 'true'], None, 'config.yaml', 'pre-backup', dry_run=False)
def test_execute_hook_with_umask_sets_that_umask():
flexmock(module.os).should_receive('umask').with_args(0o77).and_return(0o22).once()
flexmock(module.os).should_receive('umask').with_args(0o22).once()
flexmock(module.execute).should_receive('execute_command').with_args(
[':'], output_log_level=logging.WARNING, shell=True
)
module.execute_hook([':'], 77, 'config.yaml', 'pre-backup', dry_run=False)
def test_execute_hook_with_dry_run_skips_commands():
flexmock(module.execute).should_receive('execute_command').never()
module.execute_hook([':', 'true'], None, 'config.yaml', 'pre-backup', dry_run=True)
2017-10-26 07:32:06 +02:00
def test_execute_hook_with_empty_commands_does_not_raise():
module.execute_hook([], None, 'config.yaml', 'post-backup', dry_run=False)
2019-06-13 19:14:16 +02:00
def test_execute_hook_on_error_logs_as_error():
flexmock(module.execute).should_receive('execute_command').with_args(
[':'], output_log_level=logging.ERROR, shell=True
).once()
module.execute_hook([':'], None, 'config.yaml', 'on-error', dry_run=False)