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