|
| 1 | +import datetime |
| 2 | + |
| 3 | +from schedule.tests.factories import ( |
| 4 | + DayFactory, |
| 5 | + ScheduleItemAttendeeFactory, |
| 6 | + ScheduleItemFactory, |
| 7 | + SlotFactory, |
| 8 | +) |
| 9 | +from submissions.tests.factories import SubmissionFactory |
| 10 | +from users.tests.factories import UserFactory |
| 11 | +import pytest |
| 12 | + |
| 13 | +from schedule.models import ScheduleItem |
| 14 | + |
| 15 | +pytestmark = pytest.mark.django_db |
| 16 | + |
| 17 | + |
| 18 | +def _bookable_schedule_item(attendees_total_capacity: int = 30): |
| 19 | + submission = SubmissionFactory() |
| 20 | + return ScheduleItemFactory( |
| 21 | + status=ScheduleItem.STATUS.confirmed, |
| 22 | + submission=submission, |
| 23 | + type=ScheduleItem.TYPES.training, |
| 24 | + conference=submission.conference, |
| 25 | + attendees_total_capacity=attendees_total_capacity, |
| 26 | + slot=SlotFactory( |
| 27 | + day=DayFactory( |
| 28 | + day=datetime.date(2020, 10, 10), |
| 29 | + conference=submission.conference, |
| 30 | + ), |
| 31 | + hour=datetime.time(10, 10, 0), |
| 32 | + duration=30, |
| 33 | + ), |
| 34 | + ) |
| 35 | + |
| 36 | + |
| 37 | +def test_get_booked_schedule_items(graphql_client, user): |
| 38 | + graphql_client.force_login(user) |
| 39 | + |
| 40 | + booked_item = _bookable_schedule_item() |
| 41 | + _bookable_schedule_item() |
| 42 | + ScheduleItemAttendeeFactory(schedule_item=booked_item, user_id=user.id) |
| 43 | + |
| 44 | + response = graphql_client.query( |
| 45 | + """query($conference: String!) { |
| 46 | + me { |
| 47 | + bookedScheduleItems(conference: $conference) { |
| 48 | + id |
| 49 | + title |
| 50 | + slug |
| 51 | + start |
| 52 | + end |
| 53 | + } |
| 54 | + } |
| 55 | + }""", |
| 56 | + variables={"conference": booked_item.conference.code}, |
| 57 | + ) |
| 58 | + |
| 59 | + booked_schedule_items = response["data"]["me"]["bookedScheduleItems"] |
| 60 | + assert len(booked_schedule_items) == 1 |
| 61 | + assert booked_schedule_items[0]["id"] == str(booked_item.id) |
| 62 | + assert booked_schedule_items[0]["slug"] == booked_item.slug |
| 63 | + assert booked_schedule_items[0]["start"] is not None |
| 64 | + assert booked_schedule_items[0]["end"] is not None |
| 65 | + |
| 66 | + |
| 67 | +def test_booked_schedule_items_excludes_items_without_slot(graphql_client, user): |
| 68 | + graphql_client.force_login(user) |
| 69 | + |
| 70 | + submission = SubmissionFactory() |
| 71 | + unscheduled_workshop = ScheduleItemFactory( |
| 72 | + status=ScheduleItem.STATUS.confirmed, |
| 73 | + submission=submission, |
| 74 | + type=ScheduleItem.TYPES.training, |
| 75 | + conference=submission.conference, |
| 76 | + attendees_total_capacity=30, |
| 77 | + slot=None, |
| 78 | + ) |
| 79 | + ScheduleItemAttendeeFactory(schedule_item=unscheduled_workshop, user_id=user.id) |
| 80 | + |
| 81 | + response = graphql_client.query( |
| 82 | + """query($conference: String!) { |
| 83 | + me { |
| 84 | + bookedScheduleItems(conference: $conference) { |
| 85 | + id |
| 86 | + start |
| 87 | + end |
| 88 | + } |
| 89 | + } |
| 90 | + }""", |
| 91 | + variables={"conference": unscheduled_workshop.conference.code}, |
| 92 | + ) |
| 93 | + |
| 94 | + assert response["data"]["me"]["bookedScheduleItems"] == [] |
| 95 | + |
| 96 | + |
| 97 | +def test_booked_schedule_items_excludes_other_users_bookings(graphql_client, user): |
| 98 | + graphql_client.force_login(user) |
| 99 | + |
| 100 | + booked_item = _bookable_schedule_item() |
| 101 | + other_user = UserFactory() |
| 102 | + ScheduleItemAttendeeFactory(schedule_item=booked_item, user_id=other_user.id) |
| 103 | + |
| 104 | + response = graphql_client.query( |
| 105 | + """query($conference: String!) { |
| 106 | + me { |
| 107 | + bookedScheduleItems(conference: $conference) { |
| 108 | + id |
| 109 | + } |
| 110 | + } |
| 111 | + }""", |
| 112 | + variables={"conference": booked_item.conference.code}, |
| 113 | + ) |
| 114 | + |
| 115 | + assert response["data"]["me"]["bookedScheduleItems"] == [] |
| 116 | + |
| 117 | + |
| 118 | +def test_booked_schedule_items_requires_authentication(graphql_client): |
| 119 | + schedule_item = _bookable_schedule_item() |
| 120 | + ScheduleItemAttendeeFactory(schedule_item=schedule_item) |
| 121 | + |
| 122 | + response = graphql_client.query( |
| 123 | + """query($conference: String!) { |
| 124 | + me { |
| 125 | + bookedScheduleItems(conference: $conference) { |
| 126 | + id |
| 127 | + } |
| 128 | + } |
| 129 | + }""", |
| 130 | + variables={"conference": schedule_item.conference.code}, |
| 131 | + ) |
| 132 | + |
| 133 | + assert response["errors"][0]["message"] == "User not logged in" |
0 commit comments