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

billing: Mention the exact return type for mock functions.

This commit is contained in:
Vishnu Ks 2018-07-27 09:54:36 +00:00 committed by Rishi Gupta
parent 8bcd4131fe
commit 036ad03b3a
2 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
stubs/stripe/util/__init__.pyi
View File

@ -1,5 +1,4 @@
from typing import Any, Dict
from stripe.api_resources.list_object import ListObject
def convert_to_stripe_object(fixture_data: Dict[Any, Any]) -> ListObject:
def convert_to_stripe_object(fixture_data: Dict[Any, Any]) -> Any:
...

8
zilencer/tests/test_stripe.py
View File

@ -21,16 +21,16 @@ from zilencer.models import Customer, Plan
fixture_data_file = open(os.path.join(os.path.dirname(__file__), 'stripe_fixtures.json'), 'r')
fixture_data = ujson.load(fixture_data_file)
def mock_create_customer(*args: Any, **kwargs: Any) -> ListObject:
def mock_create_customer(*args: Any, **kwargs: Any) -> stripe.Customer:
return stripe.util.convert_to_stripe_object(fixture_data["create_customer"])
def mock_create_subscription(*args: Any, **kwargs: Any) -> ListObject:
def mock_create_subscription(*args: Any, **kwargs: Any) -> stripe.Subscription:
return stripe.util.convert_to_stripe_object(fixture_data["create_subscription"])
def mock_customer_with_active_subscription(*args: Any, **kwargs: Any) -> ListObject:
def mock_customer_with_active_subscription(*args: Any, **kwargs: Any) -> stripe.Customer:
return stripe.util.convert_to_stripe_object(fixture_data["customer_with_active_subscription"])
def mock_upcoming_invoice(*args: Any, **kwargs: Any) -> ListObject:
def mock_upcoming_invoice(*args: Any, **kwargs: Any) -> stripe.Invoice:
return stripe.util.convert_to_stripe_object(fixture_data["upcoming_invoice"])
class StripeTest(ZulipTestCase):