Add custom methods to all resources

stripe · Apr 3, 2019 · 91f7806 · 91f7806
1 parent 6aa81a1
commit 91f7806
Show file tree

Hide file tree

Showing 29 changed files with 207 additions and 19 deletions.
diff --git a/stripe/api_resources/account.py b/stripe/api_resources/account.py
@@ -7,10 +7,11 @@
 from stripe.api_resources.abstract import UpdateableAPIResource
 from stripe.api_resources.abstract import ListableAPIResource
 from stripe.api_resources.abstract import nested_resource_class_methods
-
+from stripe.api_resources.abstract import custom_method
 from stripe.six.moves.urllib.parse import quote_plus
 
 
+@custom_method("reject", http_verb="post")
 @nested_resource_class_methods(
     "external_account",
     operations=["create", "retrieve", "update", "delete", "list"],

diff --git a/stripe/api_resources/customer.py b/stripe/api_resources/customer.py
@@ -5,9 +5,11 @@
 from stripe.api_resources.abstract import DeletableAPIResource
 from stripe.api_resources.abstract import UpdateableAPIResource
 from stripe.api_resources.abstract import ListableAPIResource
+from stripe.api_resources.abstract import custom_method
 from stripe.api_resources.abstract import nested_resource_class_methods
 
 
+@custom_method("delete_discount", http_verb="delete", http_path="discount")
 @nested_resource_class_methods(
     "source", operations=["create", "retrieve", "update", "delete", "list"]
 )

diff --git a/stripe/api_resources/dispute.py b/stripe/api_resources/dispute.py
@@ -3,8 +3,10 @@
 from stripe import util
 from stripe.api_resources.abstract import ListableAPIResource
 from stripe.api_resources.abstract import UpdateableAPIResource
+from stripe.api_resources.abstract import custom_method
 
 
+@custom_method("close", http_verb="post")
 class Dispute(ListableAPIResource, UpdateableAPIResource):
     OBJECT_NAME = "dispute"
 

diff --git a/stripe/api_resources/issuing/authorization.py b/stripe/api_resources/issuing/authorization.py
@@ -3,8 +3,11 @@
 from stripe import util
 from stripe.api_resources.abstract import UpdateableAPIResource
 from stripe.api_resources.abstract import ListableAPIResource
+from stripe.api_resources.abstract import custom_method
 
 
+@custom_method("approve", http_verb="post")
+@custom_method("decline", http_verb="post")
 class Authorization(ListableAPIResource, UpdateableAPIResource):
     OBJECT_NAME = "issuing.authorization"
 

diff --git a/stripe/api_resources/issuing/card.py b/stripe/api_resources/issuing/card.py
@@ -3,8 +3,10 @@
 from stripe.api_resources.abstract import CreateableAPIResource
 from stripe.api_resources.abstract import UpdateableAPIResource
 from stripe.api_resources.abstract import ListableAPIResource
+from stripe.api_resources.abstract import custom_method
 
 
+@custom_method("details", http_verb="get")
 class Card(CreateableAPIResource, ListableAPIResource, UpdateableAPIResource):
     OBJECT_NAME = "issuing.card"
 

diff --git a/stripe/api_resources/order.py b/stripe/api_resources/order.py
@@ -4,8 +4,11 @@
 from stripe.api_resources.abstract import CreateableAPIResource
 from stripe.api_resources.abstract import UpdateableAPIResource
 from stripe.api_resources.abstract import ListableAPIResource
+from stripe.api_resources.abstract import custom_method
 
 
+@custom_method("pay", http_verb="post")
+@custom_method("return_order", http_verb="post", http_path="returns")
 class Order(CreateableAPIResource, UpdateableAPIResource, ListableAPIResource):
     OBJECT_NAME = "order"
 

diff --git a/stripe/api_resources/payment_intent.py b/stripe/api_resources/payment_intent.py
@@ -4,8 +4,12 @@
 from stripe.api_resources.abstract import CreateableAPIResource
 from stripe.api_resources.abstract import UpdateableAPIResource
 from stripe.api_resources.abstract import ListableAPIResource
+from stripe.api_resources.abstract import custom_method
 
 
+@custom_method("cancel", http_verb="post")
+@custom_method("capture", http_verb="post")
+@custom_method("confirm", http_verb="post")
 class PaymentIntent(
     CreateableAPIResource, UpdateableAPIResource, ListableAPIResource
 ):

diff --git a/stripe/api_resources/payment_method.py b/stripe/api_resources/payment_method.py
@@ -4,8 +4,11 @@
 from stripe.api_resources.abstract import CreateableAPIResource
 from stripe.api_resources.abstract import ListableAPIResource
 from stripe.api_resources.abstract import UpdateableAPIResource
+from stripe.api_resources.abstract import custom_method
 
 
+@custom_method("attach", http_verb="post")
+@custom_method("detach", http_verb="post")
 class PaymentMethod(
     CreateableAPIResource, ListableAPIResource, UpdateableAPIResource
 ):

diff --git a/stripe/api_resources/payout.py b/stripe/api_resources/payout.py
@@ -4,8 +4,10 @@
 from stripe.api_resources.abstract import CreateableAPIResource
 from stripe.api_resources.abstract import UpdateableAPIResource
 from stripe.api_resources.abstract import ListableAPIResource
+from stripe.api_resources.abstract import custom_method
 
 
+@custom_method("cancel", http_verb="post")
 class Payout(
     CreateableAPIResource, UpdateableAPIResource, ListableAPIResource
 ):

diff --git a/stripe/api_resources/review.py b/stripe/api_resources/review.py
@@ -2,8 +2,10 @@
 
 from stripe import util
 from stripe.api_resources.abstract import ListableAPIResource
+from stripe.api_resources.abstract import custom_method
 
 
+@custom_method("approve", http_verb="post")
 class Review(ListableAPIResource):
     OBJECT_NAME = "review"
 

diff --git a/stripe/api_resources/subscription.py b/stripe/api_resources/subscription.py
@@ -5,8 +5,10 @@
 from stripe.api_resources.abstract import DeletableAPIResource
 from stripe.api_resources.abstract import UpdateableAPIResource
 from stripe.api_resources.abstract import ListableAPIResource
+from stripe.api_resources.abstract import custom_method
 
 
+@custom_method("delete_discount", http_verb="delete", http_path="discount")
 class Subscription(
     CreateableAPIResource,
     DeletableAPIResource,

diff --git a/stripe/api_resources/subscription_schedule.py b/stripe/api_resources/subscription_schedule.py
@@ -5,8 +5,11 @@
 from stripe.api_resources.abstract import UpdateableAPIResource
 from stripe.api_resources.abstract import ListableAPIResource
 from stripe.api_resources.abstract import nested_resource_class_methods
+from stripe.api_resources.abstract import custom_method
 
 
+@custom_method("cancel", http_verb="post")
+@custom_method("release", http_verb="post")
 @nested_resource_class_methods("revision", operations=["retrieve", "list"])
 class SubscriptionSchedule(
     CreateableAPIResource, UpdateableAPIResource, ListableAPIResource

diff --git a/stripe/api_resources/topup.py b/stripe/api_resources/topup.py
@@ -4,8 +4,10 @@
 from stripe.api_resources.abstract import CreateableAPIResource
 from stripe.api_resources.abstract import UpdateableAPIResource
 from stripe.api_resources.abstract import ListableAPIResource
+from stripe.api_resources.abstract import custom_method
 
 
+@custom_method("cancel", http_verb="post")
 class Topup(CreateableAPIResource, ListableAPIResource, UpdateableAPIResource):
     OBJECT_NAME = "topup"
 

diff --git a/stripe/api_resources/transfer.py b/stripe/api_resources/transfer.py
@@ -5,8 +5,10 @@
 from stripe.api_resources.abstract import UpdateableAPIResource
 from stripe.api_resources.abstract import ListableAPIResource
 from stripe.api_resources.abstract import nested_resource_class_methods
+from stripe.api_resources.abstract import custom_method
 
 
+@custom_method("cancel", http_verb="post")
 @nested_resource_class_methods(
     "reversal", operations=["create", "retrieve", "update", "list"]
 )

diff --git a/tests/api_resources/issuing/test_authorization.py b/tests/api_resources/issuing/test_authorization.py
@@ -39,7 +39,7 @@ def test_is_saveable(self, request_mock):
         assert isinstance(resource, stripe.issuing.Authorization)
         assert resource is authorization
 
-    def test_is_approveable(self, request_mock):
+    def test_can_approve(self, request_mock):
         resource = stripe.issuing.Authorization.retrieve(TEST_RESOURCE_ID)
         authorization = resource.approve()
         request_mock.assert_requested(
@@ -48,11 +48,25 @@ def test_is_approveable(self, request_mock):
         assert isinstance(resource, stripe.issuing.Authorization)
         assert resource is authorization
 
-    def test_is_declineable(self, request_mock):
+    def test_can_approve_classmethod(self, request_mock):
+        resource = stripe.issuing.Authorization.approve(TEST_RESOURCE_ID)
+        request_mock.assert_requested(
+            "post", "/v1/issuing/authorizations/%s/approve" % TEST_RESOURCE_ID
+        )
+        assert isinstance(resource, stripe.issuing.Authorization)
+
+    def test_can_decline(self, request_mock):
         resource = stripe.issuing.Authorization.retrieve(TEST_RESOURCE_ID)
         authorization = resource.decline()
         request_mock.assert_requested(
             "post", "/v1/issuing/authorizations/%s/decline" % TEST_RESOURCE_ID
         )
         assert isinstance(resource, stripe.issuing.Authorization)
         assert resource is authorization
+
+    def test_can_decline_classmethod(self, request_mock):
+        resource = stripe.issuing.Authorization.decline(TEST_RESOURCE_ID)
+        request_mock.assert_requested(
+            "post", "/v1/issuing/authorizations/%s/decline" % TEST_RESOURCE_ID
+        )
+        assert isinstance(resource, stripe.issuing.Authorization)
diff --git a/tests/api_resources/issuing/test_card.py b/tests/api_resources/issuing/test_card.py
@@ -51,3 +51,10 @@ def test_can_retrieve_details(self, request_mock):
             "get", "/v1/issuing/cards/%s/details" % TEST_RESOURCE_ID
         )
         assert isinstance(card_details, stripe.issuing.CardDetails)
+
+    def test_can_retrieve_details_classmethod(self, request_mock):
+        card_details = stripe.issuing.Card.details(TEST_RESOURCE_ID)
+        request_mock.assert_requested(
+            "get", "/v1/issuing/cards/%s/details" % TEST_RESOURCE_ID
+        )
+        assert isinstance(card_details, stripe.issuing.CardDetails)
diff --git a/tests/api_resources/test_account.py b/tests/api_resources/test_account.py
@@ -96,11 +96,22 @@ def test_can_reject(self, request_mock):
         account = stripe.Account.retrieve(TEST_RESOURCE_ID)
         resource = account.reject(reason="fraud")
         request_mock.assert_requested(
-            "post", "/v1/accounts/%s/reject" % TEST_RESOURCE_ID
+            "post",
+            "/v1/accounts/%s/reject" % TEST_RESOURCE_ID,
+            {"reason": "fraud"},
         )
         assert isinstance(resource, stripe.Account)
         assert resource is account
 
+    def test_can_reject_classmethod(self, request_mock):
+        resource = stripe.Account.reject(TEST_RESOURCE_ID, reason="fraud")
+        request_mock.assert_requested(
+            "post",
+            "/v1/accounts/%s/reject" % TEST_RESOURCE_ID,
+            {"reason": "fraud"},
+        )
+        assert isinstance(resource, stripe.Account)
+
     def test_is_deauthorizable(self, request_mock):
         account = stripe.Account.retrieve(TEST_RESOURCE_ID)
         request_mock.stub_request(

diff --git a/tests/api_resources/test_charge.py b/tests/api_resources/test_charge.py
@@ -52,15 +52,15 @@ def test_is_refundable(self, request_mock):
         )
         assert isinstance(resource, stripe.Charge)
 
-    def test_is_capturable(self, request_mock):
+    def test_can_capture(self, request_mock):
         charge = stripe.Charge.retrieve(TEST_RESOURCE_ID)
         resource = charge.capture()
         request_mock.assert_requested(
             "post", "/v1/charges/%s/capture" % TEST_RESOURCE_ID
         )
         assert isinstance(resource, stripe.Charge)
 
-    def test_can_capture(self, request_mock):
+    def test_can_capture_classmethod(self, request_mock):
         resource = stripe.Charge.capture(TEST_RESOURCE_ID)
         request_mock.assert_requested(
             "post", "/v1/charges/%s/capture" % TEST_RESOURCE_ID

diff --git a/tests/api_resources/test_customer.py b/tests/api_resources/test_customer.py
@@ -66,6 +66,12 @@ def test_can_delete_discount(self, request_mock):
             "delete", "/v1/customers/%s/discount" % TEST_RESOURCE_ID
         )
 
+    def test_can_delete_discount_class_method(self, request_mock):
+        stripe.Customer.delete_discount(TEST_RESOURCE_ID)
+        request_mock.assert_requested(
+            "delete", "/v1/customers/%s/discount" % TEST_RESOURCE_ID
+        )
+
 
 class TestCustomerSources(object):
     def test_is_creatable(self, request_mock):

diff --git a/tests/api_resources/test_dispute.py b/tests/api_resources/test_dispute.py
@@ -37,9 +37,17 @@ def test_is_modifiable(self, request_mock):
         )
         assert isinstance(resource, stripe.Dispute)
 
-    def test_is_closeable(self, request_mock):
+    def test_can_close(self, request_mock):
         resource = stripe.Dispute.retrieve(TEST_RESOURCE_ID)
         resource.close()
         request_mock.assert_requested(
             "post", "/v1/disputes/%s/close" % TEST_RESOURCE_ID
         )
+        assert isinstance(resource, stripe.Dispute)
+
+    def test_can_close_classmethod(self, request_mock):
+        resource = stripe.Dispute.close(TEST_RESOURCE_ID)
+        request_mock.assert_requested(
+            "post", "/v1/disputes/%s/close" % TEST_RESOURCE_ID
+        )
+        assert isinstance(resource, stripe.Dispute)
diff --git a/tests/api_resources/test_order.py b/tests/api_resources/test_order.py
@@ -42,19 +42,37 @@ def test_is_modifiable(self, request_mock):
         )
         assert isinstance(resource, stripe.Order)
 
-    def test_is_payable(self, request_mock):
+    def test_can_pay(self, request_mock):
         order = stripe.Order.retrieve(TEST_RESOURCE_ID)
         resource = order.pay(source="src_123")
         request_mock.assert_requested(
-            "post", "/v1/orders/%s/pay" % order.id, {"source": "src_123"}
+            "post",
+            "/v1/orders/%s/pay" % TEST_RESOURCE_ID,
+            {"source": "src_123"},
         )
         assert isinstance(resource, stripe.Order)
         assert resource is order
 
-    def test_is_returnable(self, request_mock):
+    def test_can_pay_classmethod(self, request_mock):
+        resource = stripe.Order.pay(TEST_RESOURCE_ID, source="src_123")
+        request_mock.assert_requested(
+            "post",
+            "/v1/orders/%s/pay" % TEST_RESOURCE_ID,
+            {"source": "src_123"},
+        )
+        assert isinstance(resource, stripe.Order)
+
+    def test_can_return(self, request_mock):
         order = stripe.Order.retrieve(TEST_RESOURCE_ID)
         resource = order.return_order()
         request_mock.assert_requested(
-            "post", "/v1/orders/%s/returns" % order.id
+            "post", "/v1/orders/%s/returns" % TEST_RESOURCE_ID
+        )
+        assert isinstance(resource, stripe.OrderReturn)
+
+    def test_can_return_classmethod(self, request_mock):
+        resource = stripe.Order.return_order(TEST_RESOURCE_ID)
+        request_mock.assert_requested(
+            "post", "/v1/orders/%s/returns" % TEST_RESOURCE_ID
         )
         assert isinstance(resource, stripe.OrderReturn)
diff --git a/tests/api_resources/test_payment_intent.py b/tests/api_resources/test_payment_intent.py
@@ -52,27 +52,45 @@ def test_is_saveable(self, request_mock):
 
     def test_can_cancel(self, request_mock):
         resource = stripe.PaymentIntent.retrieve(TEST_RESOURCE_ID)
-
         resource.cancel()
         request_mock.assert_requested(
             "post", "/v1/payment_intents/%s/cancel" % TEST_RESOURCE_ID
         )
         assert isinstance(resource, stripe.PaymentIntent)
 
+    def test_can_cancel_classmethod(self, request_mock):
+        resource = stripe.PaymentIntent.cancel(TEST_RESOURCE_ID)
+        request_mock.assert_requested(
+            "post", "/v1/payment_intents/%s/cancel" % TEST_RESOURCE_ID
+        )
+        assert isinstance(resource, stripe.PaymentIntent)
+
     def test_can_capture(self, request_mock):
         resource = stripe.PaymentIntent.retrieve(TEST_RESOURCE_ID)
-
         resource.capture()
         request_mock.assert_requested(
             "post", "/v1/payment_intents/%s/capture" % TEST_RESOURCE_ID
         )
         assert isinstance(resource, stripe.PaymentIntent)
 
+    def test_can_capture_classmethod(self, request_mock):
+        resource = stripe.PaymentIntent.capture(TEST_RESOURCE_ID)
+        request_mock.assert_requested(
+            "post", "/v1/payment_intents/%s/capture" % TEST_RESOURCE_ID
+        )
+        assert isinstance(resource, stripe.PaymentIntent)
+
     def test_can_confirm(self, request_mock):
         resource = stripe.PaymentIntent.retrieve(TEST_RESOURCE_ID)
-
         resource.confirm()
         request_mock.assert_requested(
             "post", "/v1/payment_intents/%s/confirm" % TEST_RESOURCE_ID
         )
         assert isinstance(resource, stripe.PaymentIntent)
+
+    def test_can_confirm_classmethod(self, request_mock):
+        resource = stripe.PaymentIntent.confirm(TEST_RESOURCE_ID)
+        request_mock.assert_requested(
+            "post", "/v1/payment_intents/%s/confirm" % TEST_RESOURCE_ID
+        )
+        assert isinstance(resource, stripe.PaymentIntent)