From 5eddcfdb5525ab97f7acfca143bed8ae07c2aa25 Mon Sep 17 00:00:00 2001
From: Remi Jannel <remi@stripe.com>
Date: Thu, 30 Aug 2018 20:14:37 -0400
Subject: [PATCH] Add support for the Reporting resources

---
 stripe/__init__.py                            |  1 +
 stripe/api_resources/reporting/__init__.py    |  6 +++
 stripe/api_resources/reporting/report_run.py  |  8 ++++
 stripe/api_resources/reporting/report_type.py |  7 ++++
 stripe/util.py                                |  4 ++
 .../reporting/test_report_run.py              | 38 +++++++++++++++++++
 .../reporting/test_report_type.py             | 25 ++++++++++++
 7 files changed, 89 insertions(+)
 create mode 100644 stripe/api_resources/reporting/__init__.py
 create mode 100644 stripe/api_resources/reporting/report_run.py
 create mode 100644 stripe/api_resources/reporting/report_type.py
 create mode 100644 tests/api_resources/reporting/test_report_run.py
 create mode 100644 tests/api_resources/reporting/test_report_type.py

diff --git a/stripe/__init__.py b/stripe/__init__.py
index c3d561b0a..b9f796834 100644
--- a/stripe/__init__.py
+++ b/stripe/__init__.py
@@ -26,6 +26,7 @@
 # API resources
 from stripe.api_resources import *  # noqa
 from stripe.api_resources import issuing  # noqa
+from stripe.api_resources import reporting  # noqa
 from stripe.api_resources import sigma  # noqa
 
 # OAuth
diff --git a/stripe/api_resources/reporting/__init__.py b/stripe/api_resources/reporting/__init__.py
new file mode 100644
index 000000000..0994171a7
--- /dev/null
+++ b/stripe/api_resources/reporting/__init__.py
@@ -0,0 +1,6 @@
+from __future__ import absolute_import, division, print_function
+
+# flake8: noqa
+
+from stripe.api_resources.reporting.report_run import ReportRun
+from stripe.api_resources.reporting.report_type import ReportType
diff --git a/stripe/api_resources/reporting/report_run.py b/stripe/api_resources/reporting/report_run.py
new file mode 100644
index 000000000..6127403b0
--- /dev/null
+++ b/stripe/api_resources/reporting/report_run.py
@@ -0,0 +1,8 @@
+from __future__ import absolute_import, division, print_function
+
+from stripe.api_resources.abstract import CreateableAPIResource
+from stripe.api_resources.abstract import ListableAPIResource
+
+
+class ReportRun(CreateableAPIResource, ListableAPIResource):
+    OBJECT_NAME = 'reporting.report_run'
diff --git a/stripe/api_resources/reporting/report_type.py b/stripe/api_resources/reporting/report_type.py
new file mode 100644
index 000000000..11df7943e
--- /dev/null
+++ b/stripe/api_resources/reporting/report_type.py
@@ -0,0 +1,7 @@
+from __future__ import absolute_import, division, print_function
+
+from stripe.api_resources.abstract import ListableAPIResource
+
+
+class ReportType(ListableAPIResource):
+    OBJECT_NAME = 'reporting.report_type'
diff --git a/stripe/util.py b/stripe/util.py
index da49e9c30..28e98825a 100644
--- a/stripe/util.py
+++ b/stripe/util.py
@@ -192,6 +192,10 @@ def load_object_classes():
         api_resources.RecipientTransfer.OBJECT_NAME:
             api_resources.RecipientTransfer,
         api_resources.Refund.OBJECT_NAME: api_resources.Refund,
+        api_resources.reporting.ReportRun.OBJECT_NAME:
+            api_resources.reporting.ReportRun,
+        api_resources.reporting.ReportType.OBJECT_NAME:
+            api_resources.reporting.ReportType,
         api_resources.Reversal.OBJECT_NAME: api_resources.Reversal,
         api_resources.sigma.ScheduledQueryRun.OBJECT_NAME:
             api_resources.sigma.ScheduledQueryRun,
diff --git a/tests/api_resources/reporting/test_report_run.py b/tests/api_resources/reporting/test_report_run.py
new file mode 100644
index 000000000..7cd7a0718
--- /dev/null
+++ b/tests/api_resources/reporting/test_report_run.py
@@ -0,0 +1,38 @@
+from __future__ import absolute_import, division, print_function
+
+import stripe
+
+
+TEST_RESOURCE_ID = 'frr_123'
+
+
+class TestReportRun(object):
+    def test_is_creatable(self, request_mock):
+        resource = stripe.reporting.ReportRun.create(
+            parameters={
+                'connected_account': 'acct_123',
+            },
+            report_type='activity.summary.1',
+        )
+        request_mock.assert_requested(
+            'post',
+            '/v1/reporting/report_runs'
+        )
+        assert isinstance(resource, stripe.reporting.ReportRun)
+
+    def test_is_listable(self, request_mock):
+        resources = stripe.reporting.ReportRun.list()
+        request_mock.assert_requested(
+            'get',
+            '/v1/reporting/report_runs'
+        )
+        assert isinstance(resources.data, list)
+        assert isinstance(resources.data[0], stripe.reporting.ReportRun)
+
+    def test_is_retrievable(self, request_mock):
+        resource = stripe.reporting.ReportRun.retrieve(TEST_RESOURCE_ID)
+        request_mock.assert_requested(
+            'get',
+            '/v1/reporting/report_runs/%s' % TEST_RESOURCE_ID
+        )
+        assert isinstance(resource, stripe.reporting.ReportRun)
diff --git a/tests/api_resources/reporting/test_report_type.py b/tests/api_resources/reporting/test_report_type.py
new file mode 100644
index 000000000..7d2869cff
--- /dev/null
+++ b/tests/api_resources/reporting/test_report_type.py
@@ -0,0 +1,25 @@
+from __future__ import absolute_import, division, print_function
+
+import stripe
+
+
+TEST_RESOURCE_ID = 'activity.summary.1'
+
+
+class TestReportType(object):
+    def test_is_listable(self, request_mock):
+        resources = stripe.reporting.ReportType.list()
+        request_mock.assert_requested(
+            'get',
+            '/v1/reporting/report_types'
+        )
+        assert isinstance(resources.data, list)
+        assert isinstance(resources.data[0], stripe.reporting.ReportType)
+
+    def test_is_retrievable(self, request_mock):
+        resource = stripe.reporting.ReportType.retrieve(TEST_RESOURCE_ID)
+        request_mock.assert_requested(
+            'get',
+            '/v1/reporting/report_types/%s' % TEST_RESOURCE_ID
+        )
+        assert isinstance(resource, stripe.reporting.ReportType)