Add pagination for agency search

src/api/endpoints/search/agency/query.py

@@ -20,11 +20,13 @@ def __init__(
         location_id: int | None,
         query: str | None,
         jurisdiction_type: JurisdictionType | None,
+        page: int
     ):
         super().__init__()
         self.location_id = location_id
         self.query = query
         self.jurisdiction_type = jurisdiction_type
+        self.page = page
 
     async def run(self, session: AsyncSession) -> list[AgencySearchResponse]:
 
@@ -68,7 +70,7 @@ async def run(self, session: AsyncSession) -> list[AgencySearchResponse]:
                 ).desc()
             )
 
-        query = query.limit(50)
+        query = query.limit(10).offset((self.page - 1) * 10)
 
         mappings: Sequence[RowMapping] = await sh.mappings(session, query)
 

src/api/endpoints/search/routes.py

@@ -40,6 +40,10 @@ async def search_agency(
         description="The jurisdiction type to search for",
         default=None
     ),
+    page: int = Query(
+        description="The page to search for",
+        default=1
+    ),
     access_info: AccessInfo = Depends(get_access_info),
     async_core: AsyncCore = Depends(get_async_core),
 ) -> list[AgencySearchResponse]:
@@ -53,6 +57,7 @@ async def search_agency(
         SearchAgencyQueryBuilder(
             location_id=location_id,
             query=query,
-            jurisdiction_type=jurisdiction_type
+            jurisdiction_type=jurisdiction_type,
+            page=page
         )
     )

tests/automated/integration/api/search/agency/test_search.py

@@ -61,3 +61,13 @@ async def test_search_agency(
         }
     )
     assert len(responses) == 3
+
+    # Test pagination
+    responses = api_test_helper.request_validator.get_v2(
+        url="/search/agency",
+        params={
+            "query": "A Agency",
+            "location_id": allegheny_county.location_id,
+            "page": 2
+        }
+    )