Skip to content

Commit e755056

Browse files
authored
updated referral test (#207)
1 parent 5ede5f7 commit e755056

File tree

2 files changed

+33
-28
lines changed

2 files changed

+33
-28
lines changed

tests/cassettes/test_referral_user_all.yaml

Lines changed: 28 additions & 26 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

tests/test_referral.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,8 +40,11 @@ def test_referral_user_all(prod_api_key, page_size):
4040
"""This test requires a partner user's production API key via EASYPOST_PROD_API_KEY."""
4141
referral_users = easypost.beta.Referral.all(page_size=page_size)
4242

43-
assert len(referral_users) <= page_size
44-
assert all(isinstance(referral_user, easypost.User) for referral_user in referral_users)
43+
referral_users_array = referral_users["referral_customers"]
44+
45+
assert len(referral_users_array) <= page_size
46+
assert referral_users["has_more"] is not None
47+
assert all(isinstance(referral_user, easypost.User) for referral_user in referral_users_array)
4548

4649

4750
# PyVCR is having troubles matching the body of the form-encoded data here, override the default

0 commit comments

Comments
 (0)