Update auto_churn_subscription_setting response format

Author: wscourge

chartmogul/api/data_source.py

@@ -37,6 +37,16 @@ def make(self, data, **kwargs):
             return InvoiceHandlingSetting(**data)
 
 
+class AutoChurnSubscriptionSetting(DataObject):
+    class _Schema(Schema):
+        enabled = fields.Boolean()
+        interval = fields.Integer(allow_none=True)
+
+        @post_load
+        def make(self, data, **kwargs):
+            return AutoChurnSubscriptionSetting(**data)
+
+
 class DataSource(Resource):
     """
     https://dev.chartmogul.com/v1.0/reference#data-sources
@@ -62,7 +72,7 @@ class _Schema(Schema):
         status = fields.Str()
         system = fields.Str()
         processing_status = fields.Nested(ProcessingStatus._Schema, many=False, unknown=EXCLUDE, allow_none=True)
-        auto_churn_subscription_setting = fields.Boolean(allow_none=True)
+        auto_churn_subscription_setting = fields.Nested(AutoChurnSubscriptionSetting._Schema, many=False, unknown=EXCLUDE, allow_none=True)
         invoice_handling_setting = fields.Nested(InvoiceHandlingSetting._Schema, many=False, unknown=EXCLUDE, allow_none=True)
 
         @post_load

test/api/test_data_source.py

@@ -53,7 +53,10 @@ def test_retrieve(self, mock_requests):
                     "failed": 3,
                     "pending": 8
                 },
-                "auto_churn_subscription_setting": True,
+                "auto_churn_subscription_setting": {
+                    "enabled": True,
+                    "interval": 30
+                },
                 "invoice_handling_setting": {
                     "manual": {
                         "create_subscription_when_invoice_is": "open",
@@ -92,7 +95,10 @@ def test_retrieve(self, mock_requests):
                     "failed": 3,
                     "pending": 8
                 },
-                "auto_churn_subscription_setting": True,
+                "auto_churn_subscription_setting": {
+                    "enabled": True,
+                    "interval": 30
+                },
                 "invoice_handling_setting": {
                     "manual": {
                         "create_subscription_when_invoice_is": "open",
@@ -141,7 +147,10 @@ def test_all(self, mock_requests):
                             "failed": 3,
                             "pending": 8
                         },
-                        "auto_churn_subscription_setting": True,
+                        "auto_churn_subscription_setting": {
+                            "enabled": True,
+                            "interval": 30
+                        },
                         "invoice_handling_setting": {
                             "manual": {
                                 "create_subscription_when_invoice_is": "open",
@@ -183,7 +192,10 @@ def test_all(self, mock_requests):
                             "failed": 3,
                             "pending": 8
                         },
-                        "auto_churn_subscription_setting": True,
+                        "auto_churn_subscription_setting": {
+                            "enabled": True,
+                            "interval": 30
+                        },
                         "invoice_handling_setting": {
                             "manual": {
                                 "create_subscription_when_invoice_is": "open",