paulmataruso/zulip
1
0
Fork 0
mirror of https://github.com/zulip/zulip.git synced 2025-10-23 04:52:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

billing: Set idempotency key to None when running test suite.

Browse Source
This commit is contained in:
Rishi Gupta
2019-01-30 10:04:06 -08:00
parent 60f2213451
commit db6e6472b1
2 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
corporate/lib/stripe.py
View File

@@ -405,6 +405,9 @@ def invoice_plan(plan: CustomerPlan, event_time: datetime) -> None:
plan.invoiced_through = ledger_entry
plan.invoicing_status = CustomerPlan.STARTED
plan.save(update_fields=['invoicing_status', 'invoiced_through'])
idempotency_key = 'ledger_entry:{}'.format(ledger_entry.id) # type: Optional[str]
if settings.TEST_SUITE:
idempotency_key = None
stripe.InvoiceItem.create(
currency='usd',
customer=plan.customer.stripe_customer_id,
@@ -412,7 +415,7 @@ def invoice_plan(plan: CustomerPlan, event_time: datetime) -> None:
discountable=False,
period = {'start': datetime_to_timestamp(ledger_entry.event_time),
'end': datetime_to_timestamp(next_renewal_date(plan, ledger_entry.event_time))},
idempotency_key='ledger_entry:{}'.format(ledger_entry.id),
idempotency_key=idempotency_key,
**price_args)
invoice_item_created = True
plan.invoiced_through = ledger_entry

2
stubs/stripe/__init__.pyi
View File

@@ -173,7 +173,7 @@ class InvoiceItem:
@staticmethod
def create(amount: int=..., currency: str=..., customer: str=..., description: str=...,
discountable: bool=..., period: Dict[str, int]=..., quantity: int=...,
unit_amount: int=..., idempotency_key: str=...) -> InvoiceItem:
unit_amount: int=..., idempotency_key: Optional[str]=...) -> InvoiceItem:
...
@staticmethod