6b40f95033
Co-authored-by: Eve832 <eve.meelan@grafana.com>
Co-authored-by: Francisco Montes de Oca <nevermind89x@gmail.com>
Co-authored-by: Ildar Iskhakov <ildar.iskhakov@grafana.com>
Co-authored-by: Innokentii Konstantinov <innokenty.konstantinov@grafana.com>
Co-authored-by: Julia <ferril.darkdiver@gmail.com>
Co-authored-by: maskin25 <kengurek@gmail.com>
Co-authored-by: Matias Bordese <mbordese@gmail.com>
Co-authored-by: Matvey Kukuy <motakuk@gmail.com>
Co-authored-by: Michael Derynck <michael.derynck@grafana.com>
Co-authored-by: Richard Hartmann <richih@richih.org>
Co-authored-by: Robby Milo <robbymilo@fastmail.com>
Co-authored-by: Timur Olzhabayev <timur.olzhabayev@grafana.com>
Co-authored-by: Vadim Stepanov <vadimkerr@gmail.com>
Co-authored-by: Yulia Shanyrova <yulia.shanyrova@grafana.com>
75 lines
2.6 KiB
Python
75 lines
2.6 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.models import OnCallScheduleICal
|
|
|
|
ICAL_URL = "https://calendar.google.com/calendar/ical/c_6i1aprpgaqu89hqeelv7mrj264%40group.calendar.google.com/private-6a995cea6e74dd2cdc5d8c75bee06a2f/basic.ics" # noqa
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_export_calendar(make_organization_and_user_with_token, make_schedule):
|
|
|
|
organization, user, _ = make_organization_and_user_with_token()
|
|
|
|
schedule = make_schedule(
|
|
organization,
|
|
schedule_class=OnCallScheduleICal,
|
|
name="test_ical_schedule",
|
|
ical_url_primary=ICAL_URL,
|
|
)
|
|
_, 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) == Calendar
|
|
assert len(cal.subcomponents) == 2
|
|
|
|
|
|
@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",
|
|
ical_url_primary=ICAL_URL,
|
|
)
|
|
|
|
_, 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) == 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//"
|