Merge pull request #23 from payjp/add-statement-master

add statement API (top level resource)

Author: darai0512

.github/workflows/test.yml

@@ -9,7 +9,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        python-version: ["pypy3", 3.6, 3.7, 3.8, 3.9]
+        python-version: ["3.7", "3.8", "3.9", "3.10", "3.11"]
 
     steps:
     - uses: actions/checkout@v2

payjp/__init__.py

@@ -11,9 +11,9 @@
 retry_max_delay = 32
 
 # TODO include Card?
-__all__ = ['Account', 'Card', 'Charge', 'Customer', 'Event', 'Plan', 'Subscription', 'Token', 'Transfer']
+__all__ = ['Account', 'Card', 'Charge', 'Customer', 'Event', 'Plan', 'Subscription', 'Token', 'Transfer', 'Statement']
 
 # Resource
 from payjp.resource import (  # noqa
-    Account, Charge, Customer, Event, Plan, Subscription, Token, Transfer)
+    Account, Charge, Customer, Event, Plan, Subscription, Token, Transfer, Statement)
 

payjp/resource.py

@@ -16,7 +16,7 @@ def convert_to_payjp_object(resp, api_key, account, api_base=None):
              'charge': Charge, 'customer': Customer,
              'event': Event, 'plan': Plan,
              'subscription': Subscription, 'token': Token,
-             'transfer': Transfer, 'list': ListObject}
+             'transfer': Transfer, 'statement': Statement, 'list': ListObject}
 
     if isinstance(resp, list):
         return [convert_to_payjp_object(i, api_key, account, api_base) for i in resp]
@@ -422,3 +422,10 @@ def cancel(self, **kwargs):
 class Transfer(ListableAPIResource):
     pass
 
+
+class Statement(ListableAPIResource):
+
+    def statement_urls(self, **kwargs):
+        url = self.instance_url() + '/statement_urls'
+        self.refresh_from(self.request('post', url, kwargs))
+        return self

payjp/test/test_resources.py

@@ -1,10 +1,7 @@
 # coding: utf-8
 
-import json
 import pickle
 import sys
-import time
-import datetime
 import unittest
 
 import payjp
@@ -668,10 +665,11 @@ def test_delete(self):
 
 
 class PayjpResourceTest(PayjpApiTestCase):
+    response = {}
 
     def setUp(self):
         super(PayjpResourceTest, self).setUp()
-        self.mock_response({})
+        self.mock_response(self.response)
 
 
 class ChargeTest(PayjpResourceTest):
@@ -1121,5 +1119,29 @@ def test_individual_delete(self):
             None
         )
 
+
+class StatementTest(PayjpResourceTest):
+    response = {
+        'object': 'statement',
+        "id": "st_xxx",
+    }
+
+    def test_statement_urls(self):
+        statement = payjp.Statement.retrieve('st_xxx')
+        self.requestor_mock.request.assert_called_with(
+            'get', '/v1/statements/st_xxx', {}, None)
+        self.assertTrue(isinstance(statement, payjp.Statement))
+
+        statement.statement_urls(platformer=True)
+        self.requestor_mock.request.assert_called_with(
+            'post',
+            '/v1/statements/st_xxx/statement_urls',
+            {
+                'platformer': True,
+            },
+            None
+        )
+
+
 if __name__ == '__main__':
     unittest.main()