def test_retrieve_stripe_subscriptions_name_error():
customer = mock(
{
"id": "cus_tester3",
"object": "customer",
"sources": {"data": [{"id": "src_123"}]},
},
spec=stripe.Customer,
)
with pytest.raises(AttributeError):
payments.retrieve_stripe_subscriptions(customer)
def test_retrieve_stripe_subscriptions_fail():
cancel_response = mock(
{
"id": "cus_tester3",
"object": "customer",
"subscriptions": {
"data": []
},
"sources": {
"data": [{
"id": "src_123"
}]
},
},
spec=stripe.Customer,
)
delete_response = mock(
{
"id": "cus_tester3",
"object": "customer",
"sources": []
},
spec=stripe.Customer)
when(stripe.Customer).delete_source("cus_tester3",
"src_123").thenReturn(delete_response)
test_fail = payments.retrieve_stripe_subscriptions(cancel_response)
logger.info("test fail", test_fail=test_fail)
assert test_fail == []
unstub()
def test_retrieve_stripe_subscriptions():
response = mock(
{
"id": "cus_tester1",
"object": "customer",
"subscriptions": {"data": [{"status": "active", "id": "sub_123"}]},
"sources": {"data": [{"id": "src_123"}]},
},
spec=stripe.Customer,
)
test_subscriptions = payments.retrieve_stripe_subscriptions(response)
logger.info("test subscriptions", test_subscriptions=test_subscriptions)
assert test_subscriptions[0]["id"] == "sub_123"
assert test_subscriptions[0]["status"] == "active"
unstub()
def test_retrieve_stripe_subscriptions_cancelled():
cancelled_sub = {"status": "cancelled", "id": "sub_124", "cancel_at": 232322}
cancel_response = mock(
{
"id": "cus_tester2",
"object": "customer",
"subscriptions": {"data": [cancelled_sub]},
"sources": {"data": [{"id": "src_123"}]},
},
spec=stripe.Customer,
)
delete_response = mock(
{"id": "cus_tester2", "object": "customer", "sources": []}, spec=stripe.Customer
)
when(stripe.Customer).delete_source("cus_tester2", "src_123").thenReturn(
delete_response
)
test_cancel = payments.retrieve_stripe_subscriptions(cancel_response)
logger.info("test cancel", test_cancel=test_cancel)
assert test_cancel[0]["status"] == "cancelled"
unstub()