add page_size kwarg to list_subscriptions()

Author: asonnenschein

planet/clients/subscriptions.py

@@ -69,6 +69,7 @@ async def list_subscriptions(
             self,
             status: Optional[Sequence[str]] = None,
             limit: int = 100,
+            page_size: Optional[int] = None,
             created: Optional[str] = None,
             end_time: Optional[str] = None,
             hosting: Optional[bool] = None,
@@ -112,6 +113,7 @@ async def list_subscriptions(
             updated (str): filter by updated time or interval.
             limit (int): limit the number of subscriptions in the
                 results. When set to 0, no maximum is applied.
+            page_size (int): number of subscriptions to return per page, default 20.
             TODO: user_id
 
         Datetime args (created, end_time, start_time, updated) can either be a
@@ -156,6 +158,8 @@ class _SubscriptionsPager(Paged):
             params['sort_by'] = sort_by
         if updated is not None:
             params['updated'] = updated
+        if page_size is not None:
+            params['page_size'] = page_size
 
         try:
             response = await self._session.request(method='GET',

tests/integration/test_subscriptions_api.py

@@ -267,6 +267,17 @@ async def test_list_subscriptions_filtering_and_sorting():
         ]) == 2
 
 
+@pytest.mark.parametrize("page_size, count", [(50, 100), (100, 100)])
+@pytest.mark.anyio
+@api_mock
+async def test_list_subscriptions_page_size_success(page_size, count):
+    async with Session() as session:
+        client = SubscriptionsClient(session, base_url=TEST_URL)
+        assert len([
+            sub async for sub in client.list_subscriptions(page_size=page_size)
+        ]) == count
+
+
 @pytest.mark.anyio
 @failing_api_mock
 async def test_create_subscription_failure():