afc688feda
# Which issue(s) this PR closes Fixes [this issue](https://github.com/grafana/oncall-private/pull/2620/files#diff-0144920543fd191db13f76c9fb797116e26eda2bdd2b79332b61bfbf5846208eR193-R197) (https://github.com/PyCQA/pycodestyle/issues/334#issuecomment-2027394413) in `grafana/oncall-private` ## Checklist - [ ] Unit, integration, and e2e (if applicable) tests updated (N/A) - [x] Documentation added (or `pr:no public docs` PR label added if not required) - [x] Added the relevant release notes label (see labels prefixed w/ `release:`). These labels dictate how your PR will show up in the autogenerated release notes.
119 lines
3.7 KiB
Python
119 lines
3.7 KiB
Python
import pytest
|
|
from django.urls import reverse
|
|
from icalendar import Calendar
|
|
from rest_framework import status
|
|
from rest_framework.test import APIClient
|
|
|
|
from apps.auth_token.models import ScheduleExportAuthToken, UserScheduleExportAuthToken
|
|
from apps.schedules.constants import ICAL_COMPONENT_VEVENT, ICAL_SUMMARY
|
|
from apps.schedules.models import OnCallScheduleICal
|
|
|
|
ICAL_DATA = """
|
|
BEGIN:VCALENDAR
|
|
PRODID://Grafana Labs//Grafana On-Call//
|
|
CALSCALE:GREGORIAN
|
|
X-WR-CALNAME:test_ical_schedule
|
|
X-WR-TIMEZONE:UTC
|
|
BEGIN:VEVENT
|
|
SUMMARY:justin.hunthrop@grafana.com
|
|
DTSTART;TZID=America/Chicago:20211015T000000
|
|
DTEND;TZID=America/Chicago:20211015T120000
|
|
DTSTAMP:20230223T144743Z
|
|
UID:03vjiku070po61a9t8t7ln9q4o@google.com
|
|
SEQUENCE:1
|
|
RRULE:FREQ=DAILY
|
|
CREATED:20211015T013834Z
|
|
DESCRIPTION:
|
|
LAST-MODIFIED:20211015T142118Z
|
|
LOCATION:
|
|
STATUS:CONFIRMED
|
|
TRANSP:TRANSPARENT
|
|
END:VEVENT
|
|
BEGIN:VEVENT
|
|
SUMMARY:amixr
|
|
DTSTART;TZID=America/Chicago:20211015T120000
|
|
DTEND;TZID=America/Chicago:20211016T000000
|
|
DTSTAMP:20230223T144743Z
|
|
UID:0g1cuqi56qtaqvgb38crsh0mpa@google.com
|
|
SEQUENCE:1
|
|
RRULE:FREQ=DAILY
|
|
CREATED:20211015T020105Z
|
|
DESCRIPTION:
|
|
LAST-MODIFIED:20211015T140758Z
|
|
LOCATION:
|
|
STATUS:CONFIRMED
|
|
TRANSP:OPAQUE
|
|
END:VEVENT
|
|
END:VCALENDAR
|
|
"""
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_export_calendar(make_organization_and_user_with_token, make_user_for_organization, make_schedule):
|
|
organization, user, _ = make_organization_and_user_with_token()
|
|
usernames = {"amixr", "justin.hunthrop@grafana.com"}
|
|
# setup users for shifts
|
|
for u in usernames:
|
|
make_user_for_organization(organization, username=u)
|
|
|
|
schedule = make_schedule(
|
|
organization,
|
|
schedule_class=OnCallScheduleICal,
|
|
name="test_ical_schedule",
|
|
cached_ical_file_primary=ICAL_DATA,
|
|
)
|
|
_, schedule_token = ScheduleExportAuthToken.create_auth_token(
|
|
user=user, organization=organization, schedule=schedule
|
|
)
|
|
|
|
client = APIClient()
|
|
|
|
url = reverse("api-public:schedules-export", kwargs={"pk": schedule.public_primary_key})
|
|
url = url + "?token={0}".format(schedule_token)
|
|
|
|
response = client.get(url, format="text/calendar")
|
|
|
|
assert response.status_code == status.HTTP_200_OK
|
|
assert response.headers["content-type"] == "text/calendar; charset=utf-8"
|
|
|
|
cal = Calendar.from_ical(response.data)
|
|
|
|
assert type(cal) is Calendar
|
|
# check there are events
|
|
assert len(cal.subcomponents) > 0
|
|
for component in cal.walk():
|
|
if component.name == ICAL_COMPONENT_VEVENT:
|
|
assert component[ICAL_SUMMARY] in usernames
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_export_user_calendar(make_organization_and_user_with_token, make_schedule):
|
|
organization, user, _ = make_organization_and_user_with_token()
|
|
|
|
# make a schedule so that one is available
|
|
make_schedule(
|
|
organization,
|
|
schedule_class=OnCallScheduleICal,
|
|
name="test_ical_schedule",
|
|
cached_ical_file_primary=ICAL_DATA,
|
|
)
|
|
|
|
_, schedule_token = UserScheduleExportAuthToken.create_auth_token(user=user, organization=organization)
|
|
|
|
url = reverse("api-public:users-schedule-export", kwargs={"pk": user.public_primary_key})
|
|
url = url + "?token={0}".format(schedule_token)
|
|
|
|
client = APIClient()
|
|
|
|
response = client.get(url, format="text/calendar")
|
|
|
|
assert response.status_code == status.HTTP_200_OK
|
|
assert response.headers["content-type"] == "text/calendar; charset=utf-8"
|
|
|
|
cal = Calendar.from_ical(response.data)
|
|
|
|
assert type(cal) is Calendar
|
|
assert cal.get("x-wr-calname") == "On-Call Schedule for {0}".format(user.username)
|
|
assert cal.get("x-wr-timezone") == "UTC"
|
|
assert cal.get("calscale") == "GREGORIAN"
|
|
assert cal.get("prodid") == "//Grafana Labs//Grafana On-Call//"
|