borgmatic/tests/unit/hooks/test_apprise.py

224 lines
6.2 KiB
Python
Raw Normal View History

2023-10-01 18:52:20 +02:00
import apprise
from apprise import NotifyFormat, NotifyType
from flexmock import flexmock
import borgmatic.hooks.monitor
from borgmatic.hooks import apprise as module
2023-10-04 12:36:54 +02:00
TOPIC = 'borgmatic-unit-testing'
def mock_apprise():
apprise_mock = flexmock(
add=lambda servers: None, notify=lambda title, body, body_format, notify_type: None
)
flexmock(apprise.Apprise).new_instances(apprise_mock)
return apprise_mock
2023-10-01 18:52:20 +02:00
def test_ping_monitor_adheres_dry_run():
2023-10-04 12:36:54 +02:00
mock_apprise().should_receive('notify').never()
2023-10-01 18:52:20 +02:00
module.ping_monitor(
2023-10-04 12:36:54 +02:00
{'services': [{'url': f'ntfys://{TOPIC}', 'label': 'ntfys'}]},
2023-10-01 18:52:20 +02:00
{},
'config.yaml',
borgmatic.hooks.monitor.State.FAIL,
monitoring_log_level=1,
dry_run=True,
)
def test_ping_monitor_does_not_hit_with_no_states():
2023-10-04 12:36:54 +02:00
mock_apprise().should_receive('notify').never()
2023-10-01 18:52:20 +02:00
module.ping_monitor(
2023-10-04 12:36:54 +02:00
{'services': [{'url': f'ntfys://{TOPIC}', 'label': 'ntfys'}], 'states': []},
2023-10-01 18:52:20 +02:00
{},
'config.yaml',
borgmatic.hooks.monitor.State.FAIL,
monitoring_log_level=1,
dry_run=True,
)
def test_ping_monitor_hits_fail_by_default():
2023-10-04 12:36:54 +02:00
mock_apprise().should_receive('notify').with_args(
title='A borgmatic FAIL event happened',
body='A borgmatic FAIL event happened',
body_format=NotifyFormat.TEXT,
notify_type=NotifyType.FAILURE,
).once()
2023-10-01 18:52:20 +02:00
for state in borgmatic.hooks.monitor.State:
module.ping_monitor(
2023-10-04 12:36:54 +02:00
{'services': [{'url': f'ntfys://{TOPIC}', 'label': 'ntfys'}]},
2023-10-01 18:52:20 +02:00
{},
'config.yaml',
state,
monitoring_log_level=1,
dry_run=False,
)
def test_ping_monitor_hits_with_finish_default_config():
2023-10-04 12:36:54 +02:00
mock_apprise().should_receive('notify').with_args(
2023-10-01 18:52:20 +02:00
title='A borgmatic FINISH event happened',
body='A borgmatic FINISH event happened',
body_format=NotifyFormat.TEXT,
notify_type=NotifyType.SUCCESS,
).once()
module.ping_monitor(
2023-10-04 12:36:54 +02:00
{'services': [{'url': f'ntfys://{TOPIC}', 'label': 'ntfys'}], 'states': ['finish']},
2023-10-01 18:52:20 +02:00
{},
'config.yaml',
borgmatic.hooks.monitor.State.FINISH,
monitoring_log_level=1,
dry_run=False,
)
def test_ping_monitor_hits_with_start_default_config():
2023-10-04 12:36:54 +02:00
mock_apprise().should_receive('notify').with_args(
2023-10-01 18:52:20 +02:00
title='A borgmatic START event happened',
body='A borgmatic START event happened',
body_format=NotifyFormat.TEXT,
notify_type=NotifyType.INFO,
).once()
module.ping_monitor(
2023-10-04 12:36:54 +02:00
{'services': [{'url': f'ntfys://{TOPIC}', 'label': 'ntfys'}], 'states': ['start']},
2023-10-01 18:52:20 +02:00
{},
'config.yaml',
borgmatic.hooks.monitor.State.START,
monitoring_log_level=1,
dry_run=False,
)
def test_ping_monitor_hits_with_fail_default_config():
2023-10-04 12:36:54 +02:00
mock_apprise().should_receive('notify').with_args(
2023-10-01 18:52:20 +02:00
title='A borgmatic FAIL event happened',
body='A borgmatic FAIL event happened',
body_format=NotifyFormat.TEXT,
notify_type=NotifyType.FAILURE,
).once()
module.ping_monitor(
2023-10-04 12:36:54 +02:00
{'services': [{'url': f'ntfys://{TOPIC}', 'label': 'ntfys'}], 'states': ['fail']},
2023-10-01 18:52:20 +02:00
{},
'config.yaml',
borgmatic.hooks.monitor.State.FAIL,
monitoring_log_level=1,
dry_run=False,
)
def test_ping_monitor_hits_with_log_default_config():
2023-10-04 12:36:54 +02:00
mock_apprise().should_receive('notify').with_args(
2023-10-01 18:52:20 +02:00
title='A borgmatic LOG event happened',
body='A borgmatic LOG event happened',
body_format=NotifyFormat.TEXT,
notify_type=NotifyType.INFO,
).once()
module.ping_monitor(
2023-10-04 12:36:54 +02:00
{'services': [{'url': f'ntfys://{TOPIC}', 'label': 'ntfys'}], 'states': ['log']},
2023-10-01 18:52:20 +02:00
{},
'config.yaml',
borgmatic.hooks.monitor.State.LOG,
monitoring_log_level=1,
dry_run=False,
)
2023-10-04 12:36:54 +02:00
def test_ping_monitor_passes_through_custom_message_title():
mock_apprise().should_receive('notify').with_args(
2023-10-01 18:52:20 +02:00
title='foo',
body='bar',
body_format=NotifyFormat.TEXT,
notify_type=NotifyType.FAILURE,
).once()
module.ping_monitor(
{
2023-10-04 12:36:54 +02:00
'services': [{'url': f'ntfys://{TOPIC}', 'label': 'ntfys'}],
2023-10-01 18:52:20 +02:00
'states': ['fail'],
'fail': {'title': 'foo', 'body': 'bar'},
},
{},
'config.yaml',
borgmatic.hooks.monitor.State.FAIL,
monitoring_log_level=1,
dry_run=False,
)
2023-10-04 12:36:54 +02:00
def test_ping_monitor_passes_through_custom_message_body():
mock_apprise().should_receive('notify').with_args(
2023-10-01 18:52:20 +02:00
title='',
body='baz',
body_format=NotifyFormat.TEXT,
notify_type=NotifyType.FAILURE,
).once()
module.ping_monitor(
{
2023-10-04 12:36:54 +02:00
'services': [{'url': f'ntfys://{TOPIC}', 'label': 'ntfys'}],
2023-10-01 18:52:20 +02:00
'states': ['fail'],
'fail': {'body': 'baz'},
},
{},
'config.yaml',
borgmatic.hooks.monitor.State.FAIL,
monitoring_log_level=1,
dry_run=False,
)
2023-10-04 12:36:54 +02:00
def test_ping_monitor_pings_multiple_services():
mock_apprise().should_receive('add').with_args([f'ntfys://{TOPIC}', f'ntfy://{TOPIC}']).once()
2023-10-01 18:52:20 +02:00
module.ping_monitor(
{
'services': [
2023-10-04 12:36:54 +02:00
{'url': f'ntfys://{TOPIC}', 'label': 'ntfys'},
{'url': f'ntfy://{TOPIC}', 'label': 'ntfy'},
2023-10-01 18:52:20 +02:00
]
},
{},
'config.yaml',
borgmatic.hooks.monitor.State.FAIL,
monitoring_log_level=1,
dry_run=False,
)
2023-10-05 04:23:53 +02:00
def test_ping_monitor_logs_info_for_no_services():
2023-10-01 18:52:20 +02:00
flexmock(module.logger).should_receive('info').once()
module.ping_monitor(
{'services': []},
{},
'config.yaml',
borgmatic.hooks.monitor.State.FAIL,
monitoring_log_level=1,
dry_run=False,
)
2023-10-05 04:23:53 +02:00
def test_ping_monitor_logs_warning_when_notify_fails():
mock_apprise().should_receive('notify').and_return(False)
flexmock(module.logger).should_receive('warning').once()
for state in borgmatic.hooks.monitor.State:
module.ping_monitor(
{'services': [{'url': f'ntfys://{TOPIC}', 'label': 'ntfys'}]},
{},
'config.yaml',
state,
monitoring_log_level=1,
dry_run=False,
)