python3 compatibility + webapp to flask refactoring

2026-10-24 05:31:26 -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,113 @@
+import libraries.stripe
+from libraries.stripe.test.helper import (
+    StripeResourceTest, NOW, DUMMY_CHARGE
+)
+
+
+class ChargeTest(StripeResourceTest):
+
+    def test_charge_list(self):
+        libraries.stripe.Charge.list(created={'lt': NOW})
+
+        self.requestor_mock.request.assert_called_with(
+            'get',
+            '/v1/charges',
+            {
+                'created': {'lt': NOW},
+            }
+        )
+
+    def test_charge_create(self):
+        libraries.stripe.Charge.create(idempotency_key='foo', **DUMMY_CHARGE)
+
+        self.requestor_mock.request.assert_called_with(
+            'post',
+            '/v1/charges',
+            DUMMY_CHARGE,
+            {'Idempotency-Key': 'foo'},
+        )
+
+    def test_charge_retrieve(self):
+        libraries.stripe.Charge.retrieve('ch_test_id')
+
+        self.requestor_mock.request.assert_called_with(
+            'get',
+            '/v1/charges/ch_test_id',
+            {},
+            None
+        )
+
+    def test_charge_modify(self):
+        libraries.stripe.Charge.modify('ch_test_id', refund=True)
+
+        self.requestor_mock.request.assert_called_with(
+            'post',
+            '/v1/charges/ch_test_id',
+            {
+                'refund': True,
+            },
+            None
+        )
+
+    def test_charge_update_dispute(self):
+        charge = libraries.stripe.Charge(id='ch_update_id')
+        charge.update_dispute(idempotency_key='foo')
+
+        self.requestor_mock.request.assert_called_with(
+            'post',
+            '/v1/charges/ch_update_id/dispute',
+            {},
+            {'Idempotency-Key': 'foo'},
+        )
+
+    def test_charge_close_dispute(self):
+        charge = libraries.stripe.Charge(id='ch_update_id')
+        charge.close_dispute(idempotency_key='foo')
+
+        self.requestor_mock.request.assert_called_with(
+            'post',
+            '/v1/charges/ch_update_id/dispute/close',
+            {},
+            {'Idempotency-Key': 'foo'},
+        )
+
+    def test_mark_as_fraudulent(self):
+        charge = libraries.stripe.Charge(id='ch_update_id')
+        charge.mark_as_fraudulent(idempotency_key='foo')
+
+        self.requestor_mock.request.assert_called_with(
+            'post',
+            '/v1/charges/ch_update_id',
+            {
+                'fraud_details': {'user_report': 'fraudulent'}
+            },
+            {'Idempotency-Key': 'foo'},
+        )
+
+    def test_mark_as_safe(self):
+        charge = libraries.stripe.Charge(id='ch_update_id')
+        charge.mark_as_safe(idempotency_key='foo')
+
+        self.requestor_mock.request.assert_called_with(
+            'post',
+            '/v1/charges/ch_update_id',
+            {
+                'fraud_details': {'user_report': 'safe'}
+            },
+            {'Idempotency-Key': 'foo'},
+        )
+
+    def test_create_with_source_param(self):
+        libraries.stripe.Charge.create(amount=100, currency='usd',
+                             source='btcrcv_test_receiver')
+
+        self.requestor_mock.request.assert_called_with(
+            'post',
+            '/v1/charges',
+            {
+                'amount': 100,
+                'currency': 'usd',
+                'source': 'btcrcv_test_receiver'
+            },
+            None,
+        )