paulmataruso/zulip
1
0
Fork 0
mirror of https://github.com/zulip/zulip.git synced 2025-11-03 05:23:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

billing: Update test_billing_page_permissions to use @mock_stripe.

Browse Source
This commit is contained in:
Rishi Gupta
2018-10-18 09:49:50 -07:00
parent c4bad258f7
commit 509707007c
8 changed files with 690 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
corporate/tests/test_stripe.py
View File

@@ -253,21 +253,20 @@ class StripeTest(ZulipTestCase):
self.assertEqual(response.status_code, 302)
self.assertEqual('/billing/', response.url)
@patch("stripe.Invoice.upcoming", side_effect=mock_upcoming_invoice)
@patch("stripe.Customer.retrieve", side_effect=mock_customer_with_subscription)
@patch("stripe.Customer.create", side_effect=mock_create_customer)
@patch("stripe.Subscription.create", side_effect=mock_create_subscription)
def test_billing_page_permissions(self, mock_create_subscription: Mock,
mock_create_customer: Mock,
mock_customer_with_subscription: Mock,
mock_upcoming_invoice: Mock) -> None:
@mock_stripe("stripe.Token.create")
@mock_stripe("stripe.Invoice.upcoming")
@mock_stripe("stripe.Customer.retrieve")
@mock_stripe("stripe.Customer.create")
@mock_stripe("stripe.Subscription.create")
def test_billing_page_permissions(self, mock5: Mock, mock4: Mock, mock3: Mock,
mock2: Mock, mock1: Mock) -> None:
# Check that non-admins can access /upgrade via /billing, when there is no Customer object
self.login(self.example_email('hamlet'))
response = self.client_get("/billing/")
self.assertEqual(response.status_code, 302)
self.assertEqual('/upgrade/', response.url)
# Check that non-admins can sign up and pay
self.client_post("/upgrade/", {'stripeToken': self.token,
self.client_post("/upgrade/", {'stripeToken': stripe_create_token().id,
'signed_seat_count': self.signed_seat_count,
'salt': self.salt,
'plan': Plan.CLOUD_ANNUAL})