def get_portal_url(self, redirect_url):
if not self.subscription_id:
return None
if not self.customer_id:
self.customer_id = get_customer_id_from_subscription_id(
self.subscription_id)
self.save()
return get_portal_url(self.customer_id, redirect_url)
def test_get_customer_id_from_subscription(self, mock_cb):
# Given
expected_customer_id = 'customer-id'
mock_cb.Subscription.retrieve.return_value = MockChargeBeeSubscriptionResponse(
customer_id=expected_customer_id)
# When
customer_id = get_customer_id_from_subscription_id('subscription-id')
# Then
assert customer_id == expected_customer_id
