|  | @@ -1,5 +1,3 @@
 | 
	
		
			
				|  |  | -from enum import Enum
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  |  from flexmock import flexmock
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  import borgmatic.hooks.monitor
 | 
	
	
		
			
				|  | @@ -15,58 +13,70 @@ KEY = 'borg.status'
 | 
	
		
			
				|  |  |  VALUE = 'fail'
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  DATA_HOST_KEY = {
 | 
	
		
			
				|  |  | -    "jsonrpc": "2.0",
 | 
	
		
			
				|  |  | -    "method": "history.push",
 | 
	
		
			
				|  |  | -    "params": {"host": HOST, "key": KEY, "value": VALUE},
 | 
	
		
			
				|  |  | -    "id": 1,
 | 
	
		
			
				|  |  | +    'jsonrpc': '2.0',
 | 
	
		
			
				|  |  | +    'method': 'history.push',
 | 
	
		
			
				|  |  | +    'params': {'host': HOST, 'key': KEY, 'value': VALUE},
 | 
	
		
			
				|  |  | +    'id': 1,
 | 
	
		
			
				|  |  |  }
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -DATA_HOST_KEY_WITH_TOKEN = {
 | 
	
		
			
				|  |  | -    "jsonrpc": "2.0",
 | 
	
		
			
				|  |  | -    "method": "history.push",
 | 
	
		
			
				|  |  | -    "params": {"host": HOST, "key": KEY, "value": VALUE},
 | 
	
		
			
				|  |  | -    "id": 1,
 | 
	
		
			
				|  |  | -    "auth": "3fe6ed01a69ebd79907a120bcd04e494"
 | 
	
		
			
				|  |  | +DATA_HOST_KEY_WITH_KEY_VALUE = {
 | 
	
		
			
				|  |  | +    'jsonrpc': '2.0',
 | 
	
		
			
				|  |  | +    'method': 'history.push',
 | 
	
		
			
				|  |  | +    'params': {'host': HOST, 'key': KEY, 'value': VALUE},
 | 
	
		
			
				|  |  | +    'id': 1,
 | 
	
		
			
				|  |  | +    'auth': '3fe6ed01a69ebd79907a120bcd04e494',
 | 
	
		
			
				|  |  |  }
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  DATA_ITEMID = {
 | 
	
		
			
				|  |  | -    "jsonrpc": "2.0",
 | 
	
		
			
				|  |  | -    "method": "history.push",
 | 
	
		
			
				|  |  | -    "params": {"itemid": ITEMID, "value": VALUE},
 | 
	
		
			
				|  |  | -    "id": 1,
 | 
	
		
			
				|  |  | +    'jsonrpc': '2.0',
 | 
	
		
			
				|  |  | +    'method': 'history.push',
 | 
	
		
			
				|  |  | +    'params': {'itemid': ITEMID, 'value': VALUE},
 | 
	
		
			
				|  |  | +    'id': 1,
 | 
	
		
			
				|  |  |  }
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -DATA_HOST_KEY_WITH_TOKEN = {
 | 
	
		
			
				|  |  | -    "jsonrpc": "2.0",
 | 
	
		
			
				|  |  | -    "method": "history.push",
 | 
	
		
			
				|  |  | -    "params": {"itemid": ITEMID, "value": VALUE},
 | 
	
		
			
				|  |  | -    "id": 1,
 | 
	
		
			
				|  |  | -    "auth": "3fe6ed01a69ebd79907a120bcd04e494"
 | 
	
		
			
				|  |  | +DATA_HOST_KEY_WITH_ITEMID = {
 | 
	
		
			
				|  |  | +    'jsonrpc': '2.0',
 | 
	
		
			
				|  |  | +    'method': 'history.push',
 | 
	
		
			
				|  |  | +    'params': {'itemid': ITEMID, 'value': VALUE},
 | 
	
		
			
				|  |  | +    'id': 1,
 | 
	
		
			
				|  |  | +    'auth': '3fe6ed01a69ebd79907a120bcd04e494',
 | 
	
		
			
				|  |  |  }
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  DATA_USER_LOGIN = {
 | 
	
		
			
				|  |  | -    "jsonrpc": "2.0",
 | 
	
		
			
				|  |  | -    "method": "user.login",
 | 
	
		
			
				|  |  | -    "params": {"username": USERNAME, "password": PASSWORD},
 | 
	
		
			
				|  |  | -    "id": 1,
 | 
	
		
			
				|  |  | +    'jsonrpc': '2.0',
 | 
	
		
			
				|  |  | +    'method': 'user.login',
 | 
	
		
			
				|  |  | +    'params': {'username': USERNAME, 'password': PASSWORD},
 | 
	
		
			
				|  |  | +    'id': 1,
 | 
	
		
			
				|  |  |  }
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  AUTH_HEADERS_API_KEY = {
 | 
	
		
			
				|  |  |      'Content-Type': 'application/json-rpc',
 | 
	
		
			
				|  |  | -    'Authorization': f'Bearer {API_KEY}'
 | 
	
		
			
				|  |  | +    'Authorization': f'Bearer {API_KEY}',
 | 
	
		
			
				|  |  |  }
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -AUTH_HEADERS_USERNAME_PASSWORD = {
 | 
	
		
			
				|  |  | -    'Content-Type': 'application/json-rpc'
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | +AUTH_HEADERS_USERNAME_PASSWORD = {'Content-Type': 'application/json-rpc'}
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +def test_ping_monitor_with_non_matching_state_exits_early():
 | 
	
		
			
				|  |  | +    hook_config = {'api_key': API_KEY}
 | 
	
		
			
				|  |  | +    flexmock(module.requests).should_receive('post').never()
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +    module.ping_monitor(
 | 
	
		
			
				|  |  | +        hook_config,
 | 
	
		
			
				|  |  | +        {},
 | 
	
		
			
				|  |  | +        'config.yaml',
 | 
	
		
			
				|  |  | +        borgmatic.hooks.monitor.State.START,
 | 
	
		
			
				|  |  | +        monitoring_log_level=1,
 | 
	
		
			
				|  |  | +        dry_run=False,
 | 
	
		
			
				|  |  | +    )
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  def test_ping_monitor_config_with_api_key_only_exit_early():
 | 
	
		
			
				|  |  | -    # This test should exit early since only providing an API KEY is not enough 
 | 
	
		
			
				|  |  | +    # This test should exit early since only providing an API KEY is not enough
 | 
	
		
			
				|  |  |      # for the hook to work
 | 
	
		
			
				|  |  | -    hook_config = {
 | 
	
		
			
				|  |  | -        'api_key': API_KEY
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | +    hook_config = {'api_key': API_KEY}
 | 
	
		
			
				|  |  |      flexmock(module.logger).should_receive('warning').once()
 | 
	
		
			
				|  |  | +    flexmock(module.requests).should_receive('post').never()
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |      module.ping_monitor(
 | 
	
		
			
				|  |  |          hook_config,
 | 
	
	
		
			
				|  | @@ -77,13 +87,13 @@ def test_ping_monitor_config_with_api_key_only_exit_early():
 | 
	
		
			
				|  |  |          dry_run=False,
 | 
	
		
			
				|  |  |      )
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  |  def test_ping_monitor_config_with_host_only_exit_early():
 | 
	
		
			
				|  |  | -    # This test should exit early since only providing a HOST is not enough 
 | 
	
		
			
				|  |  | +    # This test should exit early since only providing a HOST is not enough
 | 
	
		
			
				|  |  |      # for the hook to work
 | 
	
		
			
				|  |  | -    hook_config = {
 | 
	
		
			
				|  |  | -        'host': HOST
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | +    hook_config = {'host': HOST}
 | 
	
		
			
				|  |  |      flexmock(module.logger).should_receive('warning').once()
 | 
	
		
			
				|  |  | +    flexmock(module.requests).should_receive('post').never()
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |      module.ping_monitor(
 | 
	
		
			
				|  |  |          hook_config,
 | 
	
	
		
			
				|  | @@ -94,13 +104,13 @@ def test_ping_monitor_config_with_host_only_exit_early():
 | 
	
		
			
				|  |  |          dry_run=False,
 | 
	
		
			
				|  |  |      )
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  |  def test_ping_monitor_config_with_key_only_exit_early():
 | 
	
		
			
				|  |  | -    # This test should exit early since only providing a KEY is not enough 
 | 
	
		
			
				|  |  | +    # This test should exit early since only providing a KEY is not enough
 | 
	
		
			
				|  |  |      # for the hook to work
 | 
	
		
			
				|  |  | -    hook_config = {
 | 
	
		
			
				|  |  | -        'key': KEY
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | +    hook_config = {'key': KEY}
 | 
	
		
			
				|  |  |      flexmock(module.logger).should_receive('warning').once()
 | 
	
		
			
				|  |  | +    flexmock(module.requests).should_receive('post').never()
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |      module.ping_monitor(
 | 
	
		
			
				|  |  |          hook_config,
 | 
	
	
		
			
				|  | @@ -111,13 +121,13 @@ def test_ping_monitor_config_with_key_only_exit_early():
 | 
	
		
			
				|  |  |          dry_run=False,
 | 
	
		
			
				|  |  |      )
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  |  def test_ping_monitor_config_with_server_only_exit_early():
 | 
	
		
			
				|  |  | -    # This test should exit early since only providing a SERVER is not enough 
 | 
	
		
			
				|  |  | +    # This test should exit early since only providing a SERVER is not enough
 | 
	
		
			
				|  |  |      # for the hook to work
 | 
	
		
			
				|  |  | -    hook_config = {
 | 
	
		
			
				|  |  | -        'server': SERVER
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | +    hook_config = {'server': SERVER}
 | 
	
		
			
				|  |  |      flexmock(module.logger).should_receive('warning').once()
 | 
	
		
			
				|  |  | +    flexmock(module.requests).should_receive('post').never()
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |      module.ping_monitor(
 | 
	
		
			
				|  |  |          hook_config,
 | 
	
	
		
			
				|  | @@ -128,14 +138,12 @@ def test_ping_monitor_config_with_server_only_exit_early():
 | 
	
		
			
				|  |  |          dry_run=False,
 | 
	
		
			
				|  |  |      )
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  |  def test_ping_monitor_config_user_password_no_zabbix_data_exit_early():
 | 
	
		
			
				|  |  |      # This test should exit early since there are HOST/KEY or ITEMID provided to publish data to
 | 
	
		
			
				|  |  | -    hook_config = {
 | 
	
		
			
				|  |  | -        'server': SERVER,
 | 
	
		
			
				|  |  | -        'username': USERNAME,
 | 
	
		
			
				|  |  | -        'password': PASSWORD
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | +    hook_config = {'server': SERVER, 'username': USERNAME, 'password': PASSWORD}
 | 
	
		
			
				|  |  |      flexmock(module.logger).should_receive('warning').once()
 | 
	
		
			
				|  |  | +    flexmock(module.requests).should_receive('post').never()
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |      module.ping_monitor(
 | 
	
		
			
				|  |  |          hook_config,
 | 
	
	
		
			
				|  | @@ -146,13 +154,12 @@ def test_ping_monitor_config_user_password_no_zabbix_data_exit_early():
 | 
	
		
			
				|  |  |          dry_run=False,
 | 
	
		
			
				|  |  |      )
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  |  def test_ping_monitor_config_api_key_no_zabbix_data_exit_early():
 | 
	
		
			
				|  |  |      # This test should exit early since there are HOST/KEY or ITEMID provided to publish data to
 | 
	
		
			
				|  |  | -    hook_config = {
 | 
	
		
			
				|  |  | -        'server': SERVER,
 | 
	
		
			
				|  |  | -        'api_key': API_KEY
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | +    hook_config = {'server': SERVER, 'api_key': API_KEY}
 | 
	
		
			
				|  |  |      flexmock(module.logger).should_receive('warning').once()
 | 
	
		
			
				|  |  | +    flexmock(module.requests).should_receive('post').never()
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |      module.ping_monitor(
 | 
	
		
			
				|  |  |          hook_config,
 | 
	
	
		
			
				|  | @@ -163,14 +170,13 @@ def test_ping_monitor_config_api_key_no_zabbix_data_exit_early():
 | 
	
		
			
				|  |  |          dry_run=False,
 | 
	
		
			
				|  |  |      )
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  |  def test_ping_monitor_config_itemid_no_auth_data_exit_early():
 | 
	
		
			
				|  |  | -    # This test should exit early since there is no authentication provided 
 | 
	
		
			
				|  |  | +    # This test should exit early since there is no authentication provided
 | 
	
		
			
				|  |  |      # and Zabbix requires authentication to use it's API
 | 
	
		
			
				|  |  | -    hook_config = {
 | 
	
		
			
				|  |  | -        'server': SERVER,
 | 
	
		
			
				|  |  | -        'itemid': ITEMID
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | +    hook_config = {'server': SERVER, 'itemid': ITEMID}
 | 
	
		
			
				|  |  |      flexmock(module.logger).should_receive('warning').once()
 | 
	
		
			
				|  |  | +    flexmock(module.requests).should_receive('post').never()
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |      module.ping_monitor(
 | 
	
		
			
				|  |  |          hook_config,
 | 
	
	
		
			
				|  | @@ -181,15 +187,13 @@ def test_ping_monitor_config_itemid_no_auth_data_exit_early():
 | 
	
		
			
				|  |  |          dry_run=False,
 | 
	
		
			
				|  |  |      )
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  |  def test_ping_monitor_config_host_and_key_no_auth_data_exit_early():
 | 
	
		
			
				|  |  | -    # This test should exit early since there is no authentication provided 
 | 
	
		
			
				|  |  | +    # This test should exit early since there is no authentication provided
 | 
	
		
			
				|  |  |      # and Zabbix requires authentication to use it's API
 | 
	
		
			
				|  |  | -    hook_config = {
 | 
	
		
			
				|  |  | -        'server': SERVER,
 | 
	
		
			
				|  |  | -        'host': HOST,
 | 
	
		
			
				|  |  | -        'key': KEY
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | +    hook_config = {'server': SERVER, 'host': HOST, 'key': KEY}
 | 
	
		
			
				|  |  |      flexmock(module.logger).should_receive('warning').once()
 | 
	
		
			
				|  |  | +    flexmock(module.requests).should_receive('post').never()
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |      module.ping_monitor(
 | 
	
		
			
				|  |  |          hook_config,
 | 
	
	
		
			
				|  | @@ -200,15 +204,11 @@ def test_ping_monitor_config_host_and_key_no_auth_data_exit_early():
 | 
	
		
			
				|  |  |          dry_run=False,
 | 
	
		
			
				|  |  |      )
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  |  def test_ping_monitor_config_host_and_key_with_api_key_auth_data_successful():
 | 
	
		
			
				|  |  |      # This test should simulate a successful POST to a Zabbix server. This test uses API_KEY
 | 
	
		
			
				|  |  |      # to authenticate and HOST/KEY to know which item to populate in Zabbix.
 | 
	
		
			
				|  |  | -    hook_config = {
 | 
	
		
			
				|  |  | -        'server': SERVER,
 | 
	
		
			
				|  |  | -        'host': HOST,
 | 
	
		
			
				|  |  | -        'key': KEY,
 | 
	
		
			
				|  |  | -        'api_key': API_KEY
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | +    hook_config = {'server': SERVER, 'host': HOST, 'key': KEY, 'api_key': API_KEY}
 | 
	
		
			
				|  |  |      flexmock(module.requests).should_receive('post').with_args(
 | 
	
		
			
				|  |  |          f'{SERVER}',
 | 
	
		
			
				|  |  |          headers=AUTH_HEADERS_API_KEY,
 | 
	
	
		
			
				|  | @@ -225,6 +225,37 @@ def test_ping_monitor_config_host_and_key_with_api_key_auth_data_successful():
 | 
	
		
			
				|  |  |          dry_run=False,
 | 
	
		
			
				|  |  |      )
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +def test_ping_monitor_config_host_and_missing_key_exits_early():
 | 
	
		
			
				|  |  | +    hook_config = {'server': SERVER, 'host': HOST, 'api_key': API_KEY}
 | 
	
		
			
				|  |  | +    flexmock(module.logger).should_receive('warning').once()
 | 
	
		
			
				|  |  | +    flexmock(module.requests).should_receive('post').never()
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +    module.ping_monitor(
 | 
	
		
			
				|  |  | +        hook_config,
 | 
	
		
			
				|  |  | +        {},
 | 
	
		
			
				|  |  | +        'config.yaml',
 | 
	
		
			
				|  |  | +        borgmatic.hooks.monitor.State.FAIL,
 | 
	
		
			
				|  |  | +        monitoring_log_level=1,
 | 
	
		
			
				|  |  | +        dry_run=False,
 | 
	
		
			
				|  |  | +    )
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +def test_ping_monitor_config_key_and_missing_host_exits_early():
 | 
	
		
			
				|  |  | +    hook_config = {'server': SERVER, 'key': KEY, 'api_key': API_KEY}
 | 
	
		
			
				|  |  | +    flexmock(module.logger).should_receive('warning').once()
 | 
	
		
			
				|  |  | +    flexmock(module.requests).should_receive('post').never()
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +    module.ping_monitor(
 | 
	
		
			
				|  |  | +        hook_config,
 | 
	
		
			
				|  |  | +        {},
 | 
	
		
			
				|  |  | +        'config.yaml',
 | 
	
		
			
				|  |  | +        borgmatic.hooks.monitor.State.FAIL,
 | 
	
		
			
				|  |  | +        monitoring_log_level=1,
 | 
	
		
			
				|  |  | +        dry_run=False,
 | 
	
		
			
				|  |  | +    )
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  |  def test_ping_monitor_config_host_and_key_with_username_password_auth_data_successful():
 | 
	
		
			
				|  |  |      # This test should simulate a successful POST to a Zabbix server. This test uses USERNAME/PASSWORD
 | 
	
		
			
				|  |  |      # to authenticate and HOST/KEY to know which item to populate in Zabbix.
 | 
	
	
		
			
				|  | @@ -233,11 +264,13 @@ def test_ping_monitor_config_host_and_key_with_username_password_auth_data_succe
 | 
	
		
			
				|  |  |          'host': HOST,
 | 
	
		
			
				|  |  |          'key': KEY,
 | 
	
		
			
				|  |  |          'username': USERNAME,
 | 
	
		
			
				|  |  | -        'password': PASSWORD
 | 
	
		
			
				|  |  | +        'password': PASSWORD,
 | 
	
		
			
				|  |  |      }
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |      auth_response = flexmock(ok=True)
 | 
	
		
			
				|  |  | -    auth_response.should_receive('json').and_return({"jsonrpc":"2.0","result":"3fe6ed01a69ebd79907a120bcd04e494","id":1})
 | 
	
		
			
				|  |  | +    auth_response.should_receive('json').and_return(
 | 
	
		
			
				|  |  | +        {'jsonrpc': '2.0', 'result': '3fe6ed01a69ebd79907a120bcd04e494', 'id': 1}
 | 
	
		
			
				|  |  | +    )
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |      flexmock(module.requests).should_receive('post').with_args(
 | 
	
		
			
				|  |  |          f'{SERVER}',
 | 
	
	
		
			
				|  | @@ -250,7 +283,7 @@ def test_ping_monitor_config_host_and_key_with_username_password_auth_data_succe
 | 
	
		
			
				|  |  |      flexmock(module.requests).should_receive('post').with_args(
 | 
	
		
			
				|  |  |          f'{SERVER}',
 | 
	
		
			
				|  |  |          headers=AUTH_HEADERS_USERNAME_PASSWORD,
 | 
	
		
			
				|  |  | -        json=DATA_HOST_KEY_WITH_TOKEN,
 | 
	
		
			
				|  |  | +        json=DATA_HOST_KEY_WITH_KEY_VALUE,
 | 
	
		
			
				|  |  |      ).and_return(flexmock(ok=True)).once()
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |      module.ping_monitor(
 | 
	
	
		
			
				|  | @@ -262,14 +295,92 @@ def test_ping_monitor_config_host_and_key_with_username_password_auth_data_succe
 | 
	
		
			
				|  |  |          dry_run=False,
 | 
	
		
			
				|  |  |      )
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -def test_ping_monitor_config_itemid_with_api_key_auth_data_successful():
 | 
	
		
			
				|  |  | -    # This test should simulate a successful POST to a Zabbix server. This test uses API_KEY
 | 
	
		
			
				|  |  | -    # to authenticate and HOST/KEY to know which item to populate in Zabbix.
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +def test_ping_monitor_config_host_and_key_with_username_password_auth_data_and_auth_post_error_exits_early():
 | 
	
		
			
				|  |  | +    hook_config = {
 | 
	
		
			
				|  |  | +        'server': SERVER,
 | 
	
		
			
				|  |  | +        'host': HOST,
 | 
	
		
			
				|  |  | +        'key': KEY,
 | 
	
		
			
				|  |  | +        'username': USERNAME,
 | 
	
		
			
				|  |  | +        'password': PASSWORD,
 | 
	
		
			
				|  |  | +    }
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +    auth_response = flexmock(ok=False)
 | 
	
		
			
				|  |  | +    auth_response.should_receive('json').and_return(
 | 
	
		
			
				|  |  | +        {'jsonrpc': '2.0', 'result': '3fe6ed01a69ebd79907a120bcd04e494', 'id': 1}
 | 
	
		
			
				|  |  | +    )
 | 
	
		
			
				|  |  | +    auth_response.should_receive('raise_for_status').and_raise(
 | 
	
		
			
				|  |  | +        module.requests.ConnectionError
 | 
	
		
			
				|  |  | +    ).once()
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +    flexmock(module.requests).should_receive('post').with_args(
 | 
	
		
			
				|  |  | +        f'{SERVER}',
 | 
	
		
			
				|  |  | +        headers=AUTH_HEADERS_USERNAME_PASSWORD,
 | 
	
		
			
				|  |  | +        json=DATA_USER_LOGIN,
 | 
	
		
			
				|  |  | +    ).and_return(auth_response).once()
 | 
	
		
			
				|  |  | +    flexmock(module.logger).should_receive('warning').once()
 | 
	
		
			
				|  |  | +    flexmock(module.requests).should_receive('post').with_args(
 | 
	
		
			
				|  |  | +        f'{SERVER}',
 | 
	
		
			
				|  |  | +        headers=AUTH_HEADERS_USERNAME_PASSWORD,
 | 
	
		
			
				|  |  | +        json=DATA_HOST_KEY_WITH_KEY_VALUE,
 | 
	
		
			
				|  |  | +    ).never()
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +    module.ping_monitor(
 | 
	
		
			
				|  |  | +        hook_config,
 | 
	
		
			
				|  |  | +        {},
 | 
	
		
			
				|  |  | +        'config.yaml',
 | 
	
		
			
				|  |  | +        borgmatic.hooks.monitor.State.FAIL,
 | 
	
		
			
				|  |  | +        monitoring_log_level=1,
 | 
	
		
			
				|  |  | +        dry_run=False,
 | 
	
		
			
				|  |  | +    )
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +def test_ping_monitor_config_host_and_key_with_username_and_missing_password_exits_early():
 | 
	
		
			
				|  |  | +    hook_config = {
 | 
	
		
			
				|  |  | +        'server': SERVER,
 | 
	
		
			
				|  |  | +        'host': HOST,
 | 
	
		
			
				|  |  | +        'key': KEY,
 | 
	
		
			
				|  |  | +        'username': USERNAME,
 | 
	
		
			
				|  |  | +    }
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +    flexmock(module.logger).should_receive('warning').once()
 | 
	
		
			
				|  |  | +    flexmock(module.requests).should_receive('post').never()
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +    module.ping_monitor(
 | 
	
		
			
				|  |  | +        hook_config,
 | 
	
		
			
				|  |  | +        {},
 | 
	
		
			
				|  |  | +        'config.yaml',
 | 
	
		
			
				|  |  | +        borgmatic.hooks.monitor.State.FAIL,
 | 
	
		
			
				|  |  | +        monitoring_log_level=1,
 | 
	
		
			
				|  |  | +        dry_run=False,
 | 
	
		
			
				|  |  | +    )
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +def test_ping_monitor_config_host_and_key_with_passing_and_missing_username_exits_early():
 | 
	
		
			
				|  |  |      hook_config = {
 | 
	
		
			
				|  |  |          'server': SERVER,
 | 
	
		
			
				|  |  | -        'itemid': ITEMID,
 | 
	
		
			
				|  |  | -        'api_key': API_KEY
 | 
	
		
			
				|  |  | +        'host': HOST,
 | 
	
		
			
				|  |  | +        'key': KEY,
 | 
	
		
			
				|  |  | +        'password': PASSWORD,
 | 
	
		
			
				|  |  |      }
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +    flexmock(module.logger).should_receive('warning').once()
 | 
	
		
			
				|  |  | +    flexmock(module.requests).should_receive('post').never()
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +    module.ping_monitor(
 | 
	
		
			
				|  |  | +        hook_config,
 | 
	
		
			
				|  |  | +        {},
 | 
	
		
			
				|  |  | +        'config.yaml',
 | 
	
		
			
				|  |  | +        borgmatic.hooks.monitor.State.FAIL,
 | 
	
		
			
				|  |  | +        monitoring_log_level=1,
 | 
	
		
			
				|  |  | +        dry_run=False,
 | 
	
		
			
				|  |  | +    )
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +def test_ping_monitor_config_itemid_with_api_key_auth_data_successful():
 | 
	
		
			
				|  |  | +    # This test should simulate a successful POST to a Zabbix server. This test uses API_KEY
 | 
	
		
			
				|  |  | +    # to authenticate and HOST/KEY to know which item to populate in Zabbix.
 | 
	
		
			
				|  |  | +    hook_config = {'server': SERVER, 'itemid': ITEMID, 'api_key': API_KEY}
 | 
	
		
			
				|  |  |      flexmock(module.requests).should_receive('post').with_args(
 | 
	
		
			
				|  |  |          f'{SERVER}',
 | 
	
		
			
				|  |  |          headers=AUTH_HEADERS_API_KEY,
 | 
	
	
		
			
				|  | @@ -286,18 +397,16 @@ def test_ping_monitor_config_itemid_with_api_key_auth_data_successful():
 | 
	
		
			
				|  |  |          dry_run=False,
 | 
	
		
			
				|  |  |      )
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  |  def test_ping_monitor_config_itemid_with_username_password_auth_data_successful():
 | 
	
		
			
				|  |  |      # This test should simulate a successful POST to a Zabbix server. This test uses USERNAME/PASSWORD
 | 
	
		
			
				|  |  |      # to authenticate and HOST/KEY to know which item to populate in Zabbix.
 | 
	
		
			
				|  |  | -    hook_config = {
 | 
	
		
			
				|  |  | -        'server': SERVER,
 | 
	
		
			
				|  |  | -        'itemid': ITEMID,
 | 
	
		
			
				|  |  | -        'username': USERNAME,
 | 
	
		
			
				|  |  | -        'password': PASSWORD
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | +    hook_config = {'server': SERVER, 'itemid': ITEMID, 'username': USERNAME, 'password': PASSWORD}
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |      auth_response = flexmock(ok=True)
 | 
	
		
			
				|  |  | -    auth_response.should_receive('json').and_return({"jsonrpc":"2.0","result":"3fe6ed01a69ebd79907a120bcd04e494","id":1})
 | 
	
		
			
				|  |  | +    auth_response.should_receive('json').and_return(
 | 
	
		
			
				|  |  | +        {'jsonrpc': '2.0', 'result': '3fe6ed01a69ebd79907a120bcd04e494', 'id': 1}
 | 
	
		
			
				|  |  | +    )
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |      flexmock(module.requests).should_receive('post').with_args(
 | 
	
		
			
				|  |  |          f'{SERVER}',
 | 
	
	
		
			
				|  | @@ -310,7 +419,7 @@ def test_ping_monitor_config_itemid_with_username_password_auth_data_successful(
 | 
	
		
			
				|  |  |      flexmock(module.requests).should_receive('post').with_args(
 | 
	
		
			
				|  |  |          f'{SERVER}',
 | 
	
		
			
				|  |  |          headers=AUTH_HEADERS_USERNAME_PASSWORD,
 | 
	
		
			
				|  |  | -        json=DATA_HOST_KEY_WITH_TOKEN,
 | 
	
		
			
				|  |  | +        json=DATA_HOST_KEY_WITH_ITEMID,
 | 
	
		
			
				|  |  |      ).and_return(flexmock(ok=True)).once()
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |      module.ping_monitor(
 | 
	
	
		
			
				|  | @@ -321,4 +430,39 @@ def test_ping_monitor_config_itemid_with_username_password_auth_data_successful(
 | 
	
		
			
				|  |  |          monitoring_log_level=1,
 | 
	
		
			
				|  |  |          dry_run=False,
 | 
	
		
			
				|  |  |      )
 | 
	
		
			
				|  |  | -test_ping_monitor_config_itemid_with_username_password_auth_data_successful()
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +def test_ping_monitor_config_itemid_with_username_password_auth_data_and_push_post_error_exits_early():
 | 
	
		
			
				|  |  | +    hook_config = {'server': SERVER, 'itemid': ITEMID, 'username': USERNAME, 'password': PASSWORD}
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +    auth_response = flexmock(ok=True)
 | 
	
		
			
				|  |  | +    auth_response.should_receive('json').and_return(
 | 
	
		
			
				|  |  | +        {'jsonrpc': '2.0', 'result': '3fe6ed01a69ebd79907a120bcd04e494', 'id': 1}
 | 
	
		
			
				|  |  | +    )
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +    flexmock(module.requests).should_receive('post').with_args(
 | 
	
		
			
				|  |  | +        f'{SERVER}',
 | 
	
		
			
				|  |  | +        headers=AUTH_HEADERS_USERNAME_PASSWORD,
 | 
	
		
			
				|  |  | +        json=DATA_USER_LOGIN,
 | 
	
		
			
				|  |  | +    ).and_return(auth_response).once()
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +    push_response = flexmock(ok=False)
 | 
	
		
			
				|  |  | +    push_response.should_receive('raise_for_status').and_raise(
 | 
	
		
			
				|  |  | +        module.requests.ConnectionError
 | 
	
		
			
				|  |  | +    ).once()
 | 
	
		
			
				|  |  | +    flexmock(module.requests).should_receive('post').with_args(
 | 
	
		
			
				|  |  | +        f'{SERVER}',
 | 
	
		
			
				|  |  | +        headers=AUTH_HEADERS_USERNAME_PASSWORD,
 | 
	
		
			
				|  |  | +        json=DATA_HOST_KEY_WITH_ITEMID,
 | 
	
		
			
				|  |  | +    ).and_return(push_response).once()
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +    flexmock(module.logger).should_receive('warning').once()
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +    module.ping_monitor(
 | 
	
		
			
				|  |  | +        hook_config,
 | 
	
		
			
				|  |  | +        {},
 | 
	
		
			
				|  |  | +        'config.yaml',
 | 
	
		
			
				|  |  | +        borgmatic.hooks.monitor.State.FAIL,
 | 
	
		
			
				|  |  | +        monitoring_log_level=1,
 | 
	
		
			
				|  |  | +        dry_run=False,
 | 
	
		
			
				|  |  | +    )
 |