def test_check_sensu_check_fail(mock_stdout, mock_is_file_in_dir): # monitoring.yaml doest exist mock_is_file_in_dir.return_value = False expected_output = "%s\n" % PaastaCheckMessages.SENSU_MONITORING_MISSING sensu_check('fake_service', 'path') output = mock_stdout.getvalue() assert output == expected_output
def test_check_sensu_team_missing(mock_stdout, mock_get_team, mock_is_file_in_dir): # monitoring.yaml exists but team is not found mock_is_file_in_dir.return_value = "/fake/path" mock_get_team.return_value = None expected_output = "%s\n%s\n" % (PaastaCheckMessages.SENSU_MONITORING_FOUND, PaastaCheckMessages.SENSU_TEAM_MISSING) sensu_check('fake_service', 'path') output = mock_stdout.getvalue() assert output == expected_output
def test_check_sensu_check_pass(mock_stdout, mock_get_team, mock_is_file_in_dir): # monitoring.yaml exists and team is found mock_is_file_in_dir.return_value = "/fake/path" team = 'team-service-infra' mock_get_team.return_value = team expected_output = "%s\n%s\n" % (PaastaCheckMessages.SENSU_MONITORING_FOUND, PaastaCheckMessages.sensu_team_found(team)) sensu_check('fake_service', 'path') output = mock_stdout.getvalue() assert output == expected_output mock_get_team.assert_called_once_with(service='fake_service', overrides={})