def test_do_replication_check():
base_module = 'paasta_tools.monitoring'
check_classic_module = base_module + '.check_classic_service_replication'
check_method = check_classic_module + '.check_replication'
read_key_method = check_classic_module + '.read_key'
mock_keys = [
'team', 'notification_email', 'runbook', 'tip', 'page', 'alert_after',
'realert_every'
]
mock_default_data = dict([(key, None) for key in mock_keys])
mock_default_data['team'] = 'test_team'
mock_specific_data = dict([(key, "test_{0}".format(key))
for key in mock_keys])
mock_specific_data['extra'] = {
'replication': {
'key': 'test_key',
'default': 'test_default',
'map': 'test_map'
}
}
with nested(
mock.patch(check_method, return_value=(-1, 'bar'), autospec=True),
mock.patch(read_key_method, return_value=-2, autospec=True),
):
expected = {
'name': 'replication_test_service',
'status': -1,
'output': 'bar',
'team': 'test_team',
'notification_email': None,
'runbook': 'no runbook',
'tip': 'no tip',
'page': False,
'check_every': '1m',
'alert_after': '0s',
'realert_every': -1
}
results = do_replication_check('test_service', mock_default_data, 3)
assert expected == results
expected = {
'name': 'replication_test_service',
'status': -1,
'output': 'bar',
'team': 'test_team',
'notification_email': 'test_notification_email',
'runbook': 'test_runbook',
'tip': 'test_tip',
'page': 'test_page',
'check_every': '1m',
'alert_after': 'test_alert_after',
'realert_every': 'test_realert_every'
}
results = do_replication_check('test_service', mock_specific_data, 3)
assert expected == results
def test_do_replication_check():
base_module = 'paasta_tools.monitoring'
check_classic_module = base_module + '.check_classic_service_replication'
check_method = check_classic_module + '.check_replication'
read_key_method = check_classic_module + '.read_key'
mock_keys = ['team', 'notification_email', 'runbook', 'tip', 'page',
'alert_after', 'realert_every']
mock_default_data = dict([(key, None) for key in mock_keys])
mock_default_data['team'] = 'test_team'
mock_specific_data = dict(
[(key, "test_{0}".format(key)) for key in mock_keys]
)
mock_specific_data['extra'] = {
'replication': {
'key': 'test_key',
'default': 'test_default',
'map': 'test_map'
}
}
with nested(
mock.patch(check_method, return_value=(-1, 'bar')),
mock.patch(read_key_method, return_value=-2)):
expected = {
'name': 'replication_test_service',
'status': -1,
'output': 'bar',
'team': 'test_team',
'notification_email': None,
'runbook': 'no runbook',
'tip': 'no tip',
'page': False,
'check_every': '1m',
'alert_after': '0s',
'realert_every': -1
}
results = do_replication_check('test_service', mock_default_data,
3)
assert expected == results
expected = {
'name': 'replication_test_service',
'status': -1,
'output': 'bar',
'team': 'test_team',
'notification_email': 'test_notification_email',
'runbook': 'test_runbook',
'tip': 'test_tip',
'page': 'test_page',
'check_every': '1m',
'alert_after': 'test_alert_after',
'realert_every': 'test_realert_every'
}
results = do_replication_check('test_service', mock_specific_data,
3)
assert expected == results
