Add fetching additional DataSource fields

lib/chartmogul.rb

@@ -50,6 +50,8 @@
 require 'chartmogul/concerns/pageable2'
 require 'chartmogul/concerns/pageable_with_anchor'
 require 'chartmogul/concerns/pageable_with_cursor'
+require 'chartmogul/concerns/auto_churn_subscription_setting'
+require 'chartmogul/concerns/processing_status'
 
 require 'chartmogul/subscription'
 require 'chartmogul/invoice'
@@ -58,6 +60,8 @@
 require 'chartmogul/note'
 require 'chartmogul/contact'
 require 'chartmogul/data_source'
+require 'chartmogul/data_source_settings/auto_churn_subscription_setting'
+require 'chartmogul/data_source_settings/processing_status'
 require 'chartmogul/ping'
 require 'chartmogul/plan'
 require 'chartmogul/plan_group'

lib/chartmogul/concerns/auto_churn_subscription_setting.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+module ChartMogul
+  module Concerns
+    module AutoChurnSubscriptionSetting
+      def self.included(base)
+        base.send :prepend, InstanceMethods
+
+        base.instance_eval do
+          readonly_attr :auto_churn_subscription_setting
+        end
+      end
+
+      module InstanceMethods
+        def set_auto_churn_subscription_setting(auto_churn_subscription_setting_attributes)
+          @auto_churn_subscription_setting = ChartMogul::DataSourceSettings::AutoChurnSubscriptionSetting.new_from_json(auto_churn_subscription_setting_attributes) if auto_churn_subscription_setting_attributes
+        end
+      end
+    end
+  end
+end

lib/chartmogul/concerns/processing_status.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+module ChartMogul
+  module Concerns
+    module ProcessingStatus
+      def self.included(base)
+        base.send :prepend, InstanceMethods
+
+        base.instance_eval do
+          readonly_attr :processing_status
+        end
+      end
+
+      module InstanceMethods
+        def set_processing_status(processing_status_attributes)
+          @processing_status = ChartMogul::DataSourceSettings::ProcessingStatus.new_from_json(processing_status_attributes) if processing_status_attributes
+        end
+      end
+    end
+  end
+end

lib/chartmogul/data_source.rb

@@ -9,6 +9,7 @@ class DataSource < APIResource
     readonly_attr :status
     readonly_attr :system
     readonly_attr :created_at, type: :time
+    readonly_attr :invoice_handling_setting
 
     writeable_attr :name
 
@@ -17,10 +18,17 @@ class DataSource < APIResource
     include API::Actions::Custom
     include API::Actions::Destroy
     include API::Actions::Retrieve
+    include Concerns::AutoChurnSubscriptionSetting
+    include Concerns::ProcessingStatus
 
     def self.all(options = {})
       DataSources.all(options)
     end
+
+    def self.retrieve(uuid, options = {})
+      path = ChartMogul::ResourcePath.new('/v1/data_sources/:uuid')
+      custom!(:get, path.apply_with_get_params(options.merge(uuid: uuid)))
+    end
   end
 
   class DataSources < APIResource

lib/chartmogul/data_source_settings/auto_churn_subscription_setting.rb

@@ -0,0 +1,10 @@
+# frozen_string_literal: true
+
+module ChartMogul
+  module DataSourceSettings
+    class AutoChurnSubscriptionSetting < ChartMogul::Object
+      readonly_attr :enabled
+      readonly_attr :interval
+    end
+  end
+end

lib/chartmogul/data_source_settings/processing_status.rb

@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+module ChartMogul
+  module DataSourceSettings
+    class ProcessingStatus < ChartMogul::Object
+      readonly_attr :processed
+      readonly_attr :pending
+      readonly_attr :failed
+    end
+  end
+end

spec/chartmogul/data_source_spec.rb

@@ -44,13 +44,68 @@
       expect(ds.uuid).not_to be_nil
       expect(ds.name).to eq('Test Data Source')
 
-      data_sources = ChartMogul::DataSource.all
+      data_sources = ChartMogul::DataSource.all(
+        with_processing_status: true,
+        with_auto_churn_subscription_setting: true,
+        with_invoice_handling_setting: true,
+      )
 
       expect(data_sources.size).to eq(1)
-      expect(data_sources[0].uuid).to eq(ds.uuid)
-      expect(data_sources[0].name).to eq(ds.name)
-      expect(data_sources[0].created_at).to eq(ds.created_at)
-      expect(data_sources[0].status).to eq(ds.status)
+      data_source = data_sources[0]
+      expect(data_source.uuid).to eq(ds.uuid)
+      expect(data_source.name).to eq(ds.name)
+      expect(data_source.created_at).to eq(ds.created_at)
+      expect(data_source.status).to eq(ds.status)
+      expect(data_source.processing_status).to be_a(ChartMogul::DataSourceSettings::ProcessingStatus)
+      expect(data_source.processing_status.processed).to eq(12)
+      expect(data_source.processing_status.failed).to eq(8)
+      expect(data_source.processing_status.pending).to eq(37)
+      expect(data_source.auto_churn_subscription_setting).to be_a(ChartMogul::DataSourceSettings::AutoChurnSubscriptionSetting)
+      expect(data_source.auto_churn_subscription_setting.enabled).to eq(true)
+      expect(data_source.auto_churn_subscription_setting.interval).to eq(30)
+      expect(data_source.invoice_handling_setting).to be_a(Hash)
+      expect(data_source.invoice_handling_setting[:automatic]).to be_a(Hash)
+      expect(data_source.invoice_handling_setting[:automatic][:create_subscription_when_invoice_is]).to eq('open')
+      expect(data_source.invoice_handling_setting[:automatic][:update_subscription_when_invoice_is]).to eq('open')
+      expect(data_source.invoice_handling_setting[:automatic][:prevent_subscription_for_invoice_refunded]).to eq(false)
+      expect(data_source.invoice_handling_setting[:automatic][:prevent_subscription_for_invoice_voided]).to eq(true)
+      expect(data_source.invoice_handling_setting[:automatic][:prevent_subscription_for_invoice_written_off]).to eq(true)
+      expect(data_source.invoice_handling_setting[:manual]).to be_a(Hash)
+      expect(data_source.invoice_handling_setting[:manual][:create_subscription_when_invoice_is]).to eq('open')
+      expect(data_source.invoice_handling_setting[:manual][:update_subscription_when_invoice_is]).to eq('open')
+      expect(data_source.invoice_handling_setting[:manual][:prevent_subscription_for_invoice_refunded]).to eq(false)
+      expect(data_source.invoice_handling_setting[:manual][:prevent_subscription_for_invoice_voided]).to eq(true)
+      expect(data_source.invoice_handling_setting[:manual][:prevent_subscription_for_invoice_written_off]).to eq(true)
+
+      data_source = ChartMogul::DataSource.retrieve(ds.uuid,
+        with_processing_status: true,
+        with_auto_churn_subscription_setting: true,
+        with_invoice_handling_setting: true,
+      )
+      expect(data_source.uuid).to eq(ds.uuid)
+      expect(data_source.name).to eq(ds.name)
+      expect(data_source.created_at).to eq(ds.created_at)
+      expect(data_source.status).to eq(ds.status)
+      expect(data_source.processing_status).to be_a(ChartMogul::DataSourceSettings::ProcessingStatus)
+      expect(data_source.processing_status.processed).to eq(12)
+      expect(data_source.processing_status.failed).to eq(8)
+      expect(data_source.processing_status.pending).to eq(37)
+      expect(data_source.auto_churn_subscription_setting).to be_a(ChartMogul::DataSourceSettings::AutoChurnSubscriptionSetting)
+      expect(data_source.auto_churn_subscription_setting.enabled).to eq(true)
+      expect(data_source.auto_churn_subscription_setting.interval).to eq(30)
+      expect(data_source.invoice_handling_setting).to be_a(Hash)
+      expect(data_source.invoice_handling_setting[:automatic]).to be_a(Hash)
+      expect(data_source.invoice_handling_setting[:automatic][:create_subscription_when_invoice_is]).to eq('open')
+      expect(data_source.invoice_handling_setting[:automatic][:update_subscription_when_invoice_is]).to eq('open')
+      expect(data_source.invoice_handling_setting[:automatic][:prevent_subscription_for_invoice_refunded]).to eq(false)
+      expect(data_source.invoice_handling_setting[:automatic][:prevent_subscription_for_invoice_voided]).to eq(true)
+      expect(data_source.invoice_handling_setting[:automatic][:prevent_subscription_for_invoice_written_off]).to eq(true)
+      expect(data_source.invoice_handling_setting[:manual]).to be_a(Hash)
+      expect(data_source.invoice_handling_setting[:manual][:create_subscription_when_invoice_is]).to eq('open')
+      expect(data_source.invoice_handling_setting[:manual][:update_subscription_when_invoice_is]).to eq('open')
+      expect(data_source.invoice_handling_setting[:manual][:prevent_subscription_for_invoice_refunded]).to eq(false)
+      expect(data_source.invoice_handling_setting[:manual][:prevent_subscription_for_invoice_voided]).to eq(true)
+      expect(data_source.invoice_handling_setting[:manual][:prevent_subscription_for_invoice_written_off]).to eq(true)
 
       data_sources[0].destroy!