Create getMealRequests query and tests

Author: ColinToft

backend/app/graphql/meal_request.py

@@ -306,6 +306,21 @@ class MealRequestMutations(MutationList):
 
 
 class MealRequestQueries(QueryList):
+    # Get all meal requests, with option to filter by certain statuses
+    getMealRequests = graphene.List(
+        MealRequestResponse,
+        admin_id=graphene.ID(required=True),  # Admin ID
+        min_drop_off_date=graphene.Date(default_value=None),
+        max_drop_off_date=graphene.Date(default_value=None),
+        status=graphene.List(
+            graphene.Enum.from_enum(MealStatus),
+            default_value=MEAL_STATUSES_ENUMS,
+        ),
+        offset=graphene.Int(default_value=0),
+        limit=graphene.Int(default_value=None),
+        sort_by_date_direction=SortDirection(default_value=SortDirection.ASCENDING),
+    )
+
     getMealRequestsByRequestorId = graphene.List(
         MealRequestResponse,
         requestor_id=graphene.ID(required=True),
@@ -334,6 +349,50 @@ class MealRequestQueries(QueryList):
         ids=graphene.List(graphene.ID),
     )
 
+    @secure_requestor_id
+    def resolve_getMealRequests(
+        self,
+        info,
+        admin_id: str,
+        min_drop_off_date: str,
+        max_drop_off_date: str,
+        status: List[MealStatus],
+        offset: int,
+        limit: int,
+        sort_by_date_direction: SortDirection,
+    ):
+        # The user must actually be an admin to view all meal requests
+        user_service = services["user_service"]
+        admin_auth_id = user_service.get_auth_id_by_user_id(admin_id)
+        admin_role = user_service.get_user_role_by_auth_id(admin_auth_id)
+        if admin_role != "Admin":
+            raise Exception("Only admins can view all meal requests")
+
+        meal_request_dtos = services["meal_request_service"].get_meal_requests(
+            min_drop_off_date,
+            max_drop_off_date,
+            status,
+            offset,
+            limit,
+            sort_by_date_direction,
+        )
+
+        return [
+            MealRequestResponse(
+                id=meal_request_dto.id,
+                requestor=meal_request_dto.requestor,
+                status=meal_request_dto.status,
+                drop_off_datetime=meal_request_dto.drop_off_datetime,
+                meal_info=meal_request_dto.meal_info,
+                onsite_contacts=meal_request_dto.onsite_contacts,
+                date_created=meal_request_dto.date_created,
+                date_updated=meal_request_dto.date_updated,
+                delivery_instructions=meal_request_dto.delivery_instructions,
+                donation_info=meal_request_dto.donation_info,
+            )
+            for meal_request_dto in meal_request_dtos
+        ]
+
     @secure_requestor_id
     def resolve_getMealRequestById(
         self,

backend/app/services/implementations/meal_request_service.py

@@ -260,6 +260,50 @@ def delete_meal_request(self, meal_request_id: str) -> MealRequestDTO:
             self.logger.error(str(error))
             raise error
 
+    def get_meal_requests(
+        self,
+        min_drop_off_date,
+        max_drop_off_date,
+        status: List[MealStatus],
+        offset,
+        limit,
+        sort_by_date_direction,
+    ) -> List[MealRequestDTO]:
+        status_value_list = list(map(lambda stat: stat.value, status))
+        try:
+            sort_prefix = "+"
+            if sort_by_date_direction == SortDirection.DESCENDING:
+                sort_prefix = "-"
+            requests = MealRequest.objects(
+                status__in=status_value_list,
+            ).order_by(f"{sort_prefix}drop_off_datetime")
+
+            # Filter results by optional parameters.
+            # Since we want to filter these optionally (i.e. filter only if specified),
+            # we cannot include them in the query above.
+            if min_drop_off_date is not None:
+                requests = requests.filter(
+                    drop_off_datetime__gte=min_drop_off_date,
+                )
+            if max_drop_off_date is not None:
+                requests = requests.filter(
+                    drop_off_datetime__lte=max_drop_off_date,
+                )
+            if limit is not None:
+                requests = requests[offset : offset + limit]
+            else:
+                requests = requests[offset:]
+
+            meal_request_dtos = []
+            for request in requests:
+                meal_request_dtos.append(request.to_dto())
+
+            return meal_request_dtos
+
+        except Exception as error:
+            self.logger.error(str(error))
+            raise error
+
     def get_meal_requests_by_requestor_id(
         self,
         requestor_id,

backend/app/services/interfaces/meal_request_service.py

@@ -63,6 +63,18 @@ def commit_to_meal_request(
     ) -> List[MealRequestDTO]:
         pass
 
+    @abstractmethod
+    def get_meal_requests(
+        self,
+        min_drop_off_date,
+        max_drop_off_date,
+        status,
+        offset,
+        limit,
+        sort_by_date_direction,
+    ) -> List[MealRequestDTO]:
+        pass
+
     @abstractmethod
     def get_meal_request_by_id(
         self,

backend/tests/graphql/test_meal_request.py

@@ -261,7 +261,7 @@ def test_create_meal_request_fails_repeat_date(
     existing_date = datetime.strptime(
         meal_request.drop_off_datetime, "%Y-%m-%dT%H:%M:%S"
     )
-    invalid_new_time = str((existing_date + timedelta(hours=3)).time())+ "Z"
+    invalid_new_time = str((existing_date + timedelta(hours=3)).time()) + "Z"
 
     counter_before = MealRequest.objects().count()
     mutation = f"""
@@ -937,6 +937,50 @@ def test_delete_meal_request_as_non_admin_fails_if_donor(meal_request_setup):
     assert MealRequest.objects(id=meal_request.id).first() is not None
 
 
+def test_get_meal_requests(meal_request_setup, onsite_contact_setup, user_setup):
+    asp, donor, meal_request = meal_request_setup
+    asp_onsite_contact, _, donor_onsite_contact, _ = onsite_contact_setup
+    _, _, admin = user_setup
+
+    executed = graphql_schema.execute(
+        f"""{{
+          getMealRequests(adminId: "{str(admin.id)}") {{
+            id
+            requestor {{
+              id
+            }},
+            status,
+            dropOffDatetime,
+            mealInfo {{
+              portions
+              dietaryRestrictions
+            }},
+            onsiteContacts {{
+              name
+              email
+              phone
+            }},
+            dateCreated,
+            dateUpdated,
+            deliveryInstructions,
+            donationInfo {{
+              donor {{
+                id
+              }},
+              commitmentDate
+              mealDescription
+              additionalInfo
+            }}
+          }}
+      }}"""
+    )
+
+    assert len(executed.data["getMealRequests"]) == 1
+    result = executed.data["getMealRequests"][0]
+    assert result["requestor"]["id"] == str(asp.id)
+    assert result["id"] == str(meal_request.id)
+
+
 def test_get_meal_request_by_donor_id(meal_request_setup, onsite_contact_setup):
     _, donor, meal_request = meal_request_setup