Merge pull request #252 from EasyPost/all_pickups

feat: add retrieve all pickups

Author: Justintime50

CHANGELOG.md

@@ -1,5 +1,9 @@
 # CHANGELOG
 
+## Next Release
+
+- Adds `all` function to `Pickup` to retrieve all pickups
+
 ## v7.8.0 (2023-01-11)
 
 - Adds new beta billing functionality for ReferralCustomer users

easypost/pickup.py

@@ -4,11 +4,14 @@
     RequestMethod,
     Requestor,
 )
-from easypost.resource import CreateResource
+from easypost.resource import (
+    AllResource,
+    CreateResource,
+)
 from easypost.util import get_lowest_object_rate
 
 
-class Pickup(CreateResource):
+class Pickup(CreateResource, AllResource):
     def buy(self, **params) -> "Pickup":
         """Buy a pickup."""
         requestor = Requestor(local_api_key=self._api_key)

tests/cassettes/test_pickup_all.yaml

@@ -17,6 +17,17 @@ def test_pickup_create(one_call_buy_shipment, basic_pickup):
     assert pickup.pickup_rates is not None
 
 
+@pytest.mark.vcr()
+def test_pickup_all(page_size):
+    pickups = easypost.Pickup.all(page_size=page_size)
+
+    pickup_array = pickups["pickups"]
+
+    assert len(pickup_array) <= page_size
+    assert pickups["has_more"] is not None
+    assert all(isinstance(pickup, easypost.Pickup) for pickup in pickup_array)
+
+
 @pytest.mark.vcr()
 def test_pickup_retrieve(one_call_buy_shipment, basic_pickup):
     shipment = easypost.Shipment.create(**one_call_buy_shipment)