Add new API usage for Customer Subscriptions connect and disconnect

lib/chartmogul/metrics/customers/subscription.rb

@@ -22,12 +22,23 @@ class Subscription < ChartMogul::Object
         def self.all(customer_uuid, options = {})
           ChartMogul::Metrics::Customers::Subscriptions.all(customer_uuid, options)
         end
+
+        def connect(data_source_uuid, customer_uuid, subscriptions)
+          subscriptions.unshift(self)
+          ChartMogul::Metrics::Customers::Subscriptions.connect(data_source_uuid, customer_uuid, subscriptions)
+        end
+
+        def disconnect(data_source_uuid, customer_uuid, subscriptions)
+          subscriptions.unshift(self)
+          ChartMogul::Metrics::Customers::Subscriptions.disconnect(data_source_uuid, customer_uuid, subscriptions)
+        end
       end
 
       class Subscriptions < APIResource
         set_resource_name 'Subscriptions'
         set_resource_path '/v1/customers/:customer_uuid/subscriptions'
 
+        include API::Actions::Custom
         include Concerns::Entries
         include Concerns::Pageable
         include Concerns::PageableWithCursor
@@ -38,6 +49,29 @@ def self.all(customer_uuid, options = {})
           super(options.merge(customer_uuid: customer_uuid))
         end
 
+        def self.connect(data_source_uuid, customer_uuid, subscriptions)
+          custom!(:post,
+                  "/v1/customers/#{customer_uuid}/connect_subscriptions",
+                  connect_disconnect_body(data_source_uuid, subscriptions))
+        end
+
+        def self.disconnect(data_source_uuid, customer_uuid, subscriptions)
+          custom!(:post,
+                  "/v1/customers/#{customer_uuid}/disconnect_subscriptions",
+                  connect_disconnect_body(data_source_uuid, subscriptions))
+        end
+
+        def self.connect_disconnect_body(data_source_uuid, subscriptions)
+          {
+            subscriptions: subscriptions.uniq.map do |subscription|
+              {
+                data_source_uuid: data_source_uuid,
+                uuid: subscription.uuid
+              }
+            end
+          }
+        end
+
         def next(customer_uuid, options = {})
           Subscriptions.all(customer_uuid, options.merge(cursor: cursor))
         end

lib/chartmogul/subscription.rb

@@ -26,13 +26,19 @@ def cancel(cancelled_at)
     end
 
     def connect(customer_uuid, subscriptions)
+      warn 'DEPRECATION WARNING: the method ChartMogul::Subscription#connect is deprecated. Use ChartMogul::Metrics::Customers::Subscription#connect instead.'
       subscriptions.unshift(self)
-      custom!(:post, "/v1/customers/#{customer_uuid}/connect_subscriptions", subscriptions: subscriptions.map(&:serialize_for_write))
+      custom!(:post,
+              "/v1/customers/#{customer_uuid}/connect_subscriptions",
+              subscriptions: subscriptions.map(&:serialize_for_write))
     end
 
     def disconnect(customer_uuid, subscriptions)
+      warn 'DEPRECATION WARNING: the method ChartMogul::Subscription#disconnect is deprecated. Use ChartMogul::Metrics::Customers::Subscription#disconnect instead.'
       subscriptions.unshift(self)
-      custom!(:post, "/v1/customers/#{customer_uuid}/disconnect_subscriptions", subscriptions: subscriptions.map(&:serialize_for_write))
+      custom!(:post,
+              "/v1/customers/#{customer_uuid}/disconnect_subscriptions",
+              subscriptions: subscriptions.map(&:serialize_for_write))
     end
 
     def self.all(customer_uuid, options = {})

spec/chartmogul/metrics/customers/subscriptions_spec.rb

@@ -41,4 +41,18 @@
     next_subscriptions = subscriptions.next(customer_uuid, per_page: 1)
     expect(next_subscriptions.size).to eq(0)
   end
+
+  it 'should connect and disconnect subscriptions ' do
+    data_source_uuid = 'ds_20bbb506-d69a-11f0-a876-535e2d78a156'
+    customer_uuid = 'cus_c9352490-8d66-403c-90c5-01663aae2959'
+    subscriptions = ChartMogul::Metrics::Customers::Subscription.all(customer_uuid).entries
+    expect(subscriptions.size).to eq(3)
+    subscription = subscriptions[0]
+    subscription.connect(data_source_uuid, customer_uuid, subscriptions)
+    subscriptions = ChartMogul::Metrics::Customers::Subscription.all(customer_uuid).entries
+    expect(subscriptions.size).to eq(1)
+    subscription.disconnect(data_source_uuid, customer_uuid, subscriptions)
+    subscriptions = ChartMogul::Metrics::Customers::Subscription.all(customer_uuid).entries
+    expect(subscriptions.size).to eq(3)
+  end
 end