def test_list_teams():
fake_team_data = {
'team_data': {
'red_jaguars': {
'pagerduty_api_key': 'aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa',
'pages_slack_channel': 'red_jaguars_pages',
'notifications_slack_channel': 'red_jaguars_notifications',
'notification_email': '*****@*****.**',
'project': 'REDJAGS',
},
'blue_barracudas': {
'pagerduty_api_key': 'bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbb',
'pages_slack_channel': 'blue_barracudas_pages',
},
},
}
expected = {
'red_jaguars',
'blue_barracudas',
}
with mock.patch(
'paasta_tools.monitoring_tools._load_sensu_team_data',
autospec=True,
return_value=fake_team_data,
):
actual = monitoring_tools.list_teams()
assert actual == expected
def check_monitoring(self) -> Tuple[bool, str]:
monitoring = self.get_monitoring()
valid_teams = list_teams()
if monitoring is not None:
team_name = monitoring.get('team', None)
if team_name is None:
return False, 'Team name is required for monitoring'
elif team_name not in valid_teams:
suggest_teams = difflib.get_close_matches(word=team_name, possibilities=valid_teams)
return False, f'Invalid team name: {team_name}. Do you mean one of these: {suggest_teams}'
return True, ''
def test_list_teams():
fake_team_data = {
"team_data": {
"red_jaguars": {
"pagerduty_api_key": "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa",
"pages_slack_channel": "red_jaguars_pages",
"notifications_slack_channel": "red_jaguars_notifications",
"notification_email": "*****@*****.**",
"project": "REDJAGS",
},
"blue_barracudas": {
"pagerduty_api_key": "bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbb",
"pages_slack_channel": "blue_barracudas_pages",
},
}
}
expected = {"red_jaguars", "blue_barracudas"}
with mock.patch(
"paasta_tools.monitoring_tools._load_sensu_team_data",
autospec=True,
return_value=fake_team_data,
):
actual = monitoring_tools.list_teams()
assert actual == expected
