Allow filtering for schedules by priority in application event endpoint

Author: ranta

api/graphql/types/application_event/types.py

@@ -32,7 +32,7 @@ class ApplicationEventNode(DjangoAuthNode):
     min_duration = Duration()
     max_duration = Duration()
 
-    application_event_schedules = ApplicationEventScheduleNode.ListField()
+    application_event_schedules = ApplicationEventScheduleNode.ListField(priority=graphene.List(graphene.Int))
     event_reservation_units = EventReservationUnitNode.ListField(preferred_order=graphene.Int())
 
     class Meta:
@@ -71,6 +71,12 @@ def filter_queryset(cls, queryset: models.QuerySet, user: AnyUser) -> models.Que
             | models.Q(application__user=user)
         ).distinct()
 
+    def resolve_application_event_schedules(root: ApplicationEvent, info: GQLInfo, **kwargs: Any) -> models.QuerySet:
+        priority: list[int] | None = kwargs.get("priority")
+        if priority is not None:
+            return root.application_event_schedules.filter(priority__in=priority)
+        return root.application_event_schedules.all()
+
     def resolve_event_reservation_units(root: ApplicationEvent, info: GQLInfo, **kwargs: Any) -> models.QuerySet:
         preferred_order: int | None = kwargs.get("preferred_order")
         if preferred_order is not None:

tests/test_graphql_api/test_application_event/test_filtering.py

@@ -5,6 +5,7 @@
 from tests.factories import (
     AgeGroupFactory,
     ApplicationEventFactory,
+    ApplicationEventScheduleFactory,
     ApplicationFactory,
     CityFactory,
     ReservationPurposeFactory,
@@ -491,6 +492,39 @@ def test_application_event__filter__by_priority__multiple(graphql):
     assert response.node(1) == {"pk": event_2.pk}
 
 
+def test_application_event__filter__schedule_priority(graphql):
+    # given:
+    # - There is an application event with schedules of different priorities
+    # - A superuser is using the system
+    schedule_high = ApplicationEventScheduleFactory.create(priority=PriorityChoice.HIGH)
+    ApplicationEventScheduleFactory.create(
+        application_event=schedule_high.application_event, priority=PriorityChoice.MEDIUM
+    )
+    schedule_low = ApplicationEventScheduleFactory.create(
+        application_event=schedule_high.application_event, priority=PriorityChoice.LOW
+    )
+    graphql.login_user_based_on_type(UserType.SUPERUSER)
+
+    # when:
+    # - User tries to filter application event schedules with a specific priority
+    query = events_query(
+        fields="pk applicationEventSchedules { pk priority }",
+        application_event_schedules__priority=[PriorityChoice.HIGH.value, PriorityChoice.LOW.value],
+    )
+    response = graphql(query)
+
+    # then:
+    # - The response contains no errors
+    # - The response contains the right application event with only the requested schedules
+    assert response.has_errors is False, response
+    assert len(response.edges) == 1, response
+    assert response.node(0)["pk"] == schedule_high.application_event.pk
+    assert response.node(0)["applicationEventSchedules"] == [
+        {"pk": schedule_high.pk, "priority": PriorityChoice.HIGH},
+        {"pk": schedule_low.pk, "priority": PriorityChoice.LOW},
+    ]
+
+
 def test_application_event__filter__by_preferred_order(graphql):
     # given:
     # - There is a draft application with application events with different preferred orders