python3 compatibility + webapp to flask refactoring

2026-10-23 21:51:28 -05:00 · 2024-01-27 20:37:44 +03:00
parent 5180a9c31c
commit 53b3de28ab
81 changed files with 11213 additions and 1085 deletions
@@ -0,0 +1,161 @@
+import libraries.stripe
+from libraries.stripe.test.helper import StripeResourceTest
+
+
+class RefundTest(StripeResourceTest):
+
+    def test_create_refund(self):
+        libraries.stripe.Refund.create(charge='ch_foo')
+
+        self.requestor_mock.request.assert_called_with(
+            'post',
+            '/v1/refunds',
+            {'charge': 'ch_foo'},
+            None
+        )
+
+    def test_fetch_refund(self):
+        libraries.stripe.Refund.retrieve('re_foo')
+
+        self.requestor_mock.request.assert_called_with(
+            'get',
+            '/v1/refunds/re_foo',
+            {},
+            None
+        )
+
+    def test_list_refunds(self):
+        libraries.stripe.Refund.list(limit=3, charge='ch_foo')
+
+        self.requestor_mock.request.assert_called_with(
+            'get',
+            '/v1/refunds',
+            {'limit': 3, 'charge': 'ch_foo'}
+        )
+
+    def test_update_refund(self):
+        refund = libraries.stripe.resource.Refund.construct_from({
+            'id': "ref_update",
+            'charge': "ch_update",
+            'metadata': {},
+        }, 'api_key')
+        refund.metadata["key"] = "value"
+        refund.save()
+
+        self.requestor_mock.request.assert_called_with(
+            'post',
+            '/v1/refunds/ref_update',
+            {
+                'metadata': {
+                    'key': 'value',
+                }
+            },
+            None
+        )
+
+
+class ChargeRefundTest(StripeResourceTest):
+
+    def test_create_refund(self):
+        charge = libraries.stripe.Charge.construct_from({
+            'id': 'ch_foo',
+            'refunds': {
+                'object': 'list',
+                'url': '/v1/charges/ch_foo/refunds',
+            }
+        }, 'api_key')
+
+        charge.refunds.create()
+
+        self.requestor_mock.request.assert_called_with(
+            'post',
+            '/v1/charges/ch_foo/refunds',
+            {},
+            None
+        )
+
+    def test_non_recursive_save(self):
+        charge = libraries.stripe.Charge.construct_from({
+            'id': 'ch_nested_update',
+            'customer': {
+                'object': 'customer',
+                'description': 'foo',
+            },
+            'refunds': {
+                'object': 'list',
+                'url': '/v1/charges/ch_foo/refunds',
+                'data': [{
+                    'id': 'ref_123',
+                }],
+            },
+        }, 'api_key')
+
+        charge.customer.description = 'bar'
+        charge.refunds.has_more = True
+        charge.refunds.data[0].description = 'bar'
+        charge.save()
+
+        # Note: ideally, we'd want the library to NOT issue requests in this
+        # case (i.e. the assert should actually be `assert_not_called()`).
+        self.requestor_mock.request.assert_called_with(
+            'post',
+            '/v1/charges/ch_nested_update',
+            {'refunds': {'has_more': True}},
+            None
+        )
+
+    def test_fetch_refund(self):
+        charge = libraries.stripe.Charge.construct_from({
+            'id': 'ch_get_refund',
+            'refunds': {
+                'object': 'list',
+                'url': '/v1/charges/ch_get_refund/refunds',
+            }
+        }, 'api_key')
+
+        charge.refunds.retrieve("ref_get")
+
+        self.requestor_mock.request.assert_called_with(
+            'get',
+            '/v1/charges/ch_get_refund/refunds/ref_get',
+            {},
+            None
+        )
+
+    def test_list_refunds(self):
+        charge = libraries.stripe.Charge.construct_from({
+            'id': 'ch_get_refund',
+            'refunds': {
+                'object': 'list',
+                'url': '/v1/charges/ch_get_refund/refunds',
+            }
+        }, 'api_key')
+
+        charge.refunds.list()
+
+        self.requestor_mock.request.assert_called_with(
+            'get',
+            '/v1/charges/ch_get_refund/refunds',
+            {},
+            None
+        )
+
+    def test_update_refund(self):
+        refund = libraries.stripe.resource.Refund.construct_from({
+            'id': "ref_update",
+            'charge': "ch_update",
+            'metadata': {},
+        }, 'api_key')
+        refund.metadata["key"] = "value"
+        refund.save()
+
+        self.requestor_mock.request.assert_called_with(
+            'post',
+            '/v1/refunds/ref_update',
+            {
+                'metadata': {
+                    'key': 'value',
+                }
+            },
+            None
+        )