Mirror
/
zulip
mirror of https://github.com/zulip/zulip.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

billing: Disable fixture generation in setUp function. 

This means you'll need access to our Stripe API key to add new fixtures.

Will be undone eventually, but having this in place will make it easier to
finish the mock.patch to mock_stripe migration.
This commit is contained in:
Vishnu Ks 2018-10-30 10:33:24 +00:00 committed by Rishi Gupta
parent d3b6b4e885
commit be162d5428
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
corporate/tests/test_stripe.py
View File

@ -121,10 +121,13 @@ def read_stripe_fixture(decorated_function_name: str,
return _read_stripe_fixture
def mock_stripe(mocked_function_name: str,
generate_this_fixture: bool=False) -> Callable[[CallableT], CallableT]:
generate_this_fixture: Optional[bool]=None) -> Callable[[CallableT], CallableT]:
def _mock_stripe(decorated_function: CallableT) -> CallableT:
mocked_function = operator.attrgetter(mocked_function_name)(sys.modules[__name__])
if GENERATE_STRIPE_FIXTURES or generate_this_fixture:
generate_fixture = generate_this_fixture
if generate_fixture is None:
generate_fixture = GENERATE_STRIPE_FIXTURES
if generate_fixture:
side_effect = generate_and_save_stripe_fixture(
decorated_function.__name__, mocked_function_name, mocked_function) # nocoverage
else:
@ -144,9 +147,9 @@ class Kandra(object):
return True
class StripeTest(ZulipTestCase):
@mock_stripe("stripe.Coupon.create")
@mock_stripe("stripe.Plan.create")
@mock_stripe("stripe.Product.create")
@mock_stripe("stripe.Coupon.create", False)
@mock_stripe("stripe.Plan.create", False)
@mock_stripe("stripe.Product.create", False)
def setUp(self, mock3: Mock, mock2: Mock, mock1: Mock) -> None:
call_command("setup_stripe")