|  | @@ -5,14 +5,22 @@ from flexmock import flexmock
 | 
	
		
			
				|  |  |  import borgmatic.hooks.monitor
 | 
	
		
			
				|  |  |  from borgmatic.hooks import apprise as module
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -topic = 'borgmatic-unit-testing'
 | 
	
		
			
				|  |  | +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
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  def test_ping_monitor_adheres_dry_run():
 | 
	
		
			
				|  |  | -    flexmock(apprise.Apprise).should_receive('notify').never()
 | 
	
		
			
				|  |  | +    mock_apprise().should_receive('notify').never()
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |      module.ping_monitor(
 | 
	
		
			
				|  |  | -        {'services': [{'url': f'ntfys://{topic}', 'label': 'ntfys'}]},
 | 
	
		
			
				|  |  | +        {'services': [{'url': f'ntfys://{TOPIC}', 'label': 'ntfys'}]},
 | 
	
		
			
				|  |  |          {},
 | 
	
		
			
				|  |  |          'config.yaml',
 | 
	
		
			
				|  |  |          borgmatic.hooks.monitor.State.FAIL,
 | 
	
	
		
			
				|  | @@ -22,10 +30,10 @@ def test_ping_monitor_adheres_dry_run():
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  def test_ping_monitor_does_not_hit_with_no_states():
 | 
	
		
			
				|  |  | -    flexmock(apprise.Apprise).should_receive('notify').never()
 | 
	
		
			
				|  |  | +    mock_apprise().should_receive('notify').never()
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |      module.ping_monitor(
 | 
	
		
			
				|  |  | -        {'services': [{'url': f'ntfys://{topic}', 'label': 'ntfys'}], 'states': []},
 | 
	
		
			
				|  |  | +        {'services': [{'url': f'ntfys://{TOPIC}', 'label': 'ntfys'}], 'states': []},
 | 
	
		
			
				|  |  |          {},
 | 
	
		
			
				|  |  |          'config.yaml',
 | 
	
		
			
				|  |  |          borgmatic.hooks.monitor.State.FAIL,
 | 
	
	
		
			
				|  | @@ -35,11 +43,16 @@ def test_ping_monitor_does_not_hit_with_no_states():
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  def test_ping_monitor_hits_fail_by_default():
 | 
	
		
			
				|  |  | -    flexmock(apprise.Apprise).should_receive('notify').once()
 | 
	
		
			
				|  |  | +    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()
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |      for state in borgmatic.hooks.monitor.State:
 | 
	
		
			
				|  |  |          module.ping_monitor(
 | 
	
		
			
				|  |  | -            {'services': [{'url': f'ntfys://{topic}', 'label': 'ntfys'}]},
 | 
	
		
			
				|  |  | +            {'services': [{'url': f'ntfys://{TOPIC}', 'label': 'ntfys'}]},
 | 
	
		
			
				|  |  |              {},
 | 
	
		
			
				|  |  |              'config.yaml',
 | 
	
		
			
				|  |  |              state,
 | 
	
	
		
			
				|  | @@ -49,7 +62,7 @@ def test_ping_monitor_hits_fail_by_default():
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  def test_ping_monitor_hits_with_finish_default_config():
 | 
	
		
			
				|  |  | -    flexmock(apprise.Apprise).should_receive('notify').with_args(
 | 
	
		
			
				|  |  | +    mock_apprise().should_receive('notify').with_args(
 | 
	
		
			
				|  |  |          title='A borgmatic FINISH event happened',
 | 
	
		
			
				|  |  |          body='A borgmatic FINISH event happened',
 | 
	
		
			
				|  |  |          body_format=NotifyFormat.TEXT,
 | 
	
	
		
			
				|  | @@ -57,7 +70,7 @@ def test_ping_monitor_hits_with_finish_default_config():
 | 
	
		
			
				|  |  |      ).once()
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |      module.ping_monitor(
 | 
	
		
			
				|  |  | -        {'services': [{'url': f'ntfys://{topic}', 'label': 'ntfys'}], 'states': ['finish']},
 | 
	
		
			
				|  |  | +        {'services': [{'url': f'ntfys://{TOPIC}', 'label': 'ntfys'}], 'states': ['finish']},
 | 
	
		
			
				|  |  |          {},
 | 
	
		
			
				|  |  |          'config.yaml',
 | 
	
		
			
				|  |  |          borgmatic.hooks.monitor.State.FINISH,
 | 
	
	
		
			
				|  | @@ -67,7 +80,7 @@ def test_ping_monitor_hits_with_finish_default_config():
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  def test_ping_monitor_hits_with_start_default_config():
 | 
	
		
			
				|  |  | -    flexmock(apprise.Apprise).should_receive('notify').with_args(
 | 
	
		
			
				|  |  | +    mock_apprise().should_receive('notify').with_args(
 | 
	
		
			
				|  |  |          title='A borgmatic START event happened',
 | 
	
		
			
				|  |  |          body='A borgmatic START event happened',
 | 
	
		
			
				|  |  |          body_format=NotifyFormat.TEXT,
 | 
	
	
		
			
				|  | @@ -75,7 +88,7 @@ def test_ping_monitor_hits_with_start_default_config():
 | 
	
		
			
				|  |  |      ).once()
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |      module.ping_monitor(
 | 
	
		
			
				|  |  | -        {'services': [{'url': f'ntfys://{topic}', 'label': 'ntfys'}], 'states': ['start']},
 | 
	
		
			
				|  |  | +        {'services': [{'url': f'ntfys://{TOPIC}', 'label': 'ntfys'}], 'states': ['start']},
 | 
	
		
			
				|  |  |          {},
 | 
	
		
			
				|  |  |          'config.yaml',
 | 
	
		
			
				|  |  |          borgmatic.hooks.monitor.State.START,
 | 
	
	
		
			
				|  | @@ -85,7 +98,7 @@ def test_ping_monitor_hits_with_start_default_config():
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  def test_ping_monitor_hits_with_fail_default_config():
 | 
	
		
			
				|  |  | -    flexmock(apprise.Apprise).should_receive('notify').with_args(
 | 
	
		
			
				|  |  | +    mock_apprise().should_receive('notify').with_args(
 | 
	
		
			
				|  |  |          title='A borgmatic FAIL event happened',
 | 
	
		
			
				|  |  |          body='A borgmatic FAIL event happened',
 | 
	
		
			
				|  |  |          body_format=NotifyFormat.TEXT,
 | 
	
	
		
			
				|  | @@ -93,7 +106,7 @@ def test_ping_monitor_hits_with_fail_default_config():
 | 
	
		
			
				|  |  |      ).once()
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |      module.ping_monitor(
 | 
	
		
			
				|  |  | -        {'services': [{'url': f'ntfys://{topic}', 'label': 'ntfys'}], 'states': ['fail']},
 | 
	
		
			
				|  |  | +        {'services': [{'url': f'ntfys://{TOPIC}', 'label': 'ntfys'}], 'states': ['fail']},
 | 
	
		
			
				|  |  |          {},
 | 
	
		
			
				|  |  |          'config.yaml',
 | 
	
		
			
				|  |  |          borgmatic.hooks.monitor.State.FAIL,
 | 
	
	
		
			
				|  | @@ -103,7 +116,7 @@ def test_ping_monitor_hits_with_fail_default_config():
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  def test_ping_monitor_hits_with_log_default_config():
 | 
	
		
			
				|  |  | -    flexmock(apprise.Apprise).should_receive('notify').with_args(
 | 
	
		
			
				|  |  | +    mock_apprise().should_receive('notify').with_args(
 | 
	
		
			
				|  |  |          title='A borgmatic LOG event happened',
 | 
	
		
			
				|  |  |          body='A borgmatic LOG event happened',
 | 
	
		
			
				|  |  |          body_format=NotifyFormat.TEXT,
 | 
	
	
		
			
				|  | @@ -111,7 +124,7 @@ def test_ping_monitor_hits_with_log_default_config():
 | 
	
		
			
				|  |  |      ).once()
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |      module.ping_monitor(
 | 
	
		
			
				|  |  | -        {'services': [{'url': f'ntfys://{topic}', 'label': 'ntfys'}], 'states': ['log']},
 | 
	
		
			
				|  |  | +        {'services': [{'url': f'ntfys://{TOPIC}', 'label': 'ntfys'}], 'states': ['log']},
 | 
	
		
			
				|  |  |          {},
 | 
	
		
			
				|  |  |          'config.yaml',
 | 
	
		
			
				|  |  |          borgmatic.hooks.monitor.State.LOG,
 | 
	
	
		
			
				|  | @@ -120,8 +133,8 @@ def test_ping_monitor_hits_with_log_default_config():
 | 
	
		
			
				|  |  |      )
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -def test_ping_monitor_with_custom_message_title():
 | 
	
		
			
				|  |  | -    flexmock(apprise.Apprise).should_receive('notify').with_args(
 | 
	
		
			
				|  |  | +def test_ping_monitor_passes_through_custom_message_title():
 | 
	
		
			
				|  |  | +    mock_apprise().should_receive('notify').with_args(
 | 
	
		
			
				|  |  |          title='foo',
 | 
	
		
			
				|  |  |          body='bar',
 | 
	
		
			
				|  |  |          body_format=NotifyFormat.TEXT,
 | 
	
	
		
			
				|  | @@ -130,7 +143,7 @@ def test_ping_monitor_with_custom_message_title():
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |      module.ping_monitor(
 | 
	
		
			
				|  |  |          {
 | 
	
		
			
				|  |  | -            'services': [{'url': f'ntfys://{topic}', 'label': 'ntfys'}],
 | 
	
		
			
				|  |  | +            'services': [{'url': f'ntfys://{TOPIC}', 'label': 'ntfys'}],
 | 
	
		
			
				|  |  |              'states': ['fail'],
 | 
	
		
			
				|  |  |              'fail': {'title': 'foo', 'body': 'bar'},
 | 
	
		
			
				|  |  |          },
 | 
	
	
		
			
				|  | @@ -142,8 +155,8 @@ def test_ping_monitor_with_custom_message_title():
 | 
	
		
			
				|  |  |      )
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -def test_ping_monitor_with_custom_message_body():
 | 
	
		
			
				|  |  | -    flexmock(apprise.Apprise).should_receive('notify').with_args(
 | 
	
		
			
				|  |  | +def test_ping_monitor_passes_through_custom_message_body():
 | 
	
		
			
				|  |  | +    mock_apprise().should_receive('notify').with_args(
 | 
	
		
			
				|  |  |          title='',
 | 
	
		
			
				|  |  |          body='baz',
 | 
	
		
			
				|  |  |          body_format=NotifyFormat.TEXT,
 | 
	
	
		
			
				|  | @@ -152,7 +165,7 @@ def test_ping_monitor_with_custom_message_body():
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |      module.ping_monitor(
 | 
	
		
			
				|  |  |          {
 | 
	
		
			
				|  |  | -            'services': [{'url': f'ntfys://{topic}', 'label': 'ntfys'}],
 | 
	
		
			
				|  |  | +            'services': [{'url': f'ntfys://{TOPIC}', 'label': 'ntfys'}],
 | 
	
		
			
				|  |  |              'states': ['fail'],
 | 
	
		
			
				|  |  |              'fail': {'body': 'baz'},
 | 
	
		
			
				|  |  |          },
 | 
	
	
		
			
				|  | @@ -164,16 +177,14 @@ def test_ping_monitor_with_custom_message_body():
 | 
	
		
			
				|  |  |      )
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -def test_ping_monitor_multiple_services():
 | 
	
		
			
				|  |  | -    flexmock(apprise.Apprise).should_receive('add').with_args(
 | 
	
		
			
				|  |  | -        [f'ntfys://{topic}', f'ntfy://{topic}']
 | 
	
		
			
				|  |  | -    ).once()
 | 
	
		
			
				|  |  | +def test_ping_monitor_pings_multiple_services():
 | 
	
		
			
				|  |  | +    mock_apprise().should_receive('add').with_args([f'ntfys://{TOPIC}', f'ntfy://{TOPIC}']).once()
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |      module.ping_monitor(
 | 
	
		
			
				|  |  |          {
 | 
	
		
			
				|  |  |              'services': [
 | 
	
		
			
				|  |  | -                {'url': f'ntfys://{topic}', 'label': 'ntfys'},
 | 
	
		
			
				|  |  | -                {'url': f'ntfy://{topic}', 'label': 'ntfy'},
 | 
	
		
			
				|  |  | +                {'url': f'ntfys://{TOPIC}', 'label': 'ntfys'},
 | 
	
		
			
				|  |  | +                {'url': f'ntfy://{TOPIC}', 'label': 'ntfy'},
 | 
	
		
			
				|  |  |              ]
 | 
	
		
			
				|  |  |          },
 | 
	
		
			
				|  |  |          {},
 |