From 05645d9b4c93f8edb53608eb6be39d5c213e19ab Mon Sep 17 00:00:00 2001
From: Olivier Bellone <ob@stripe.com>
Date: Wed, 8 Nov 2017 15:24:06 +0100
Subject: [PATCH] Added some tests

---
 tests/api_resources/test_invoice.py  | 22 +++++++++++++++++---
 tests/api_resources/test_order.py    | 31 +++++++++++++++++++++++++++-
 tests/api_resources/test_payout.py   | 12 +++++++++--
 tests/api_resources/test_source.py   | 11 ++++++++--
 tests/api_resources/test_transfer.py | 12 +++++++++--
 5 files changed, 78 insertions(+), 10 deletions(-)

diff --git a/tests/api_resources/test_invoice.py b/tests/api_resources/test_invoice.py
index 3f69610ad..8d45e07be 100644
--- 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
index 443ccea3a..23d87c8bd 100644
--- 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
index ba949edcc..bcc46891a 100644
--- 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
index 3eff83738..c83ad69e4 100644
--- 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
index fb9d41183..e84dd1421 100644
--- 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'}
         )