Added some tests

stripe · Nov 8, 2017 · 05645d9 · 05645d9
1 parent 24c8719
commit 05645d9
Show file tree

Hide file tree

Showing 5 changed files with 78 additions and 10 deletions.
diff --git a/tests/api_resources/test_invoice.py b/tests/api_resources/test_invoice.py
@@ -69,19 +69,35 @@ def test_retrieve_customer_invoices(self):
         )
 
     def test_pay_invoice(self):
+        self.mock_response({
+            'id': 'ii_pay',
+            'paid': True,
+        })
+
         invoice = stripe.Invoice(id="ii_pay")
-        invoice.pay()
+
+        self.assertTrue(invoice is invoice.pay(idempotency_key='idem-foo'))
+        self.assertEquals(True, invoice.paid)
+        self.assertEquals('ii_pay', invoice.id)
 
         self.requestor_mock.request.assert_called_with(
             'post',
             '/v1/invoices/ii_pay/pay',
             {},
-            None
+            {'Idempotency-Key': 'idem-foo'}
         )
 
     def test_pay_invoice_params(self):
+        self.mock_response({
+            'id': 'ii_pay',
+            'paid': True,
+        })
+
         invoice = stripe.Invoice(id="ii_pay")
-        invoice.pay(source="src_foo")
+
+        self.assertTrue(invoice is invoice.pay(source="src_foo"))
+        self.assertEquals(True, invoice.paid)
+        self.assertEquals('ii_pay', invoice.id)
 
         self.requestor_mock.request.assert_called_with(
             'post',

diff --git a/tests/api_resources/test_order.py b/tests/api_resources/test_order.py
@@ -15,8 +15,16 @@ def test_list_orders(self):
         )
 
     def test_pay_order(self):
+        self.mock_response({
+            'id': 'or_pay',
+            'status': 'paid',
+        })
+
         order = stripe.Order(id="or_pay")
-        order.pay()
+
+        self.assertTrue(order is order.pay())
+        self.assertEquals('paid', order.status)
+        self.assertEquals('or_pay', order.id)
 
         self.requestor_mock.request.assert_called_with(
             'post',
@@ -25,6 +33,27 @@ def test_pay_order(self):
             None
         )
 
+    def test_pay_order_with_params(self):
+        self.mock_response({
+            'id': 'or_pay',
+            'status': 'paid',
+        })
+
+        order = stripe.Order(id="or_pay")
+
+        self.assertTrue(order is order.pay(source="src_foo"))
+        self.assertEquals('paid', order.status)
+        self.assertEquals('or_pay', order.id)
+
+        self.requestor_mock.request.assert_called_with(
+            'post',
+            '/v1/orders/or_pay/pay',
+            {
+                'source': 'src_foo',
+            },
+            None
+        )
+
     def test_return_order(self):
         order = stripe.Order(id="or_return")
         order.return_order()

diff --git a/tests/api_resources/test_payout.py b/tests/api_resources/test_payout.py
@@ -15,12 +15,20 @@ def test_list_payouts(self):
         )
 
     def test_cancel_payout(self):
+        self.mock_response({
+            'id': 'po_cancel',
+            'status': 'canceled',
+        })
+
         payout = stripe.Payout(id='po_cancel')
-        payout.cancel()
+
+        self.assertTrue(payout is payout.cancel(idempotency_key='idem-foo'))
+        self.assertEquals('canceled', payout.status)
+        self.assertEquals('po_cancel', payout.id)
 
         self.requestor_mock.request.assert_called_with(
             'post',
             '/v1/payouts/po_cancel/cancel',
             {},
-            None
+            {'Idempotency-Key': 'idem-foo'}
         )
diff --git a/tests/api_resources/test_source.py b/tests/api_resources/test_source.py
@@ -57,17 +57,24 @@ def test_detach_source_unattached(self):
         self.assertRaises(NotImplementedError, source.detach)
 
     def test_detach_source_attached(self):
+        self.mock_response({
+            'id': 'src_foo'
+        })
+
         source = stripe.Source.construct_from({
             'id': 'src_foo',
             'customer': 'cus_bar',
         }, 'api_key')
-        source.detach()
+
+        self.assertTrue(source is source.detach(idempotency_key='idem-foo'))
+        self.assertFalse('customer' in source)
+        self.assertEquals('src_foo', source.id)
 
         self.requestor_mock.request.assert_called_with(
             'delete',
             '/v1/customers/cus_bar/sources/src_foo',
             {},
-            None
+            {'Idempotency-Key': 'idem-foo'}
         )
 
     def test_delete_source(self):

diff --git a/tests/api_resources/test_transfer.py b/tests/api_resources/test_transfer.py
@@ -15,14 +15,22 @@ def test_list_transfers(self):
         )
 
     def test_cancel_transfer(self):
+        self.mock_response({
+            'id': 'tr_cancel',
+            'status': 'canceled',
+        })
+
         transfer = stripe.Transfer(id='tr_cancel')
-        transfer.cancel()
+
+        self.assertTrue(transfer is transfer.cancel(idempotency_key='idem-foo'))
+        self.assertEquals('canceled', transfer.status)
+        self.assertEquals('tr_cancel', transfer.id)
 
         self.requestor_mock.request.assert_called_with(
             'post',
             '/v1/transfers/tr_cancel/cancel',
             {},
-            None
+            {'Idempotency-Key': 'idem-foo'}
         )