[Subscriptions] Add support for collection_method parameter

Author: rpietraszko

lib/stripe_mock/data.rb

@@ -350,6 +350,7 @@ def self.mock_subscription(params={})
         },
         cancel_at_period_end: false,
         canceled_at: nil,
+        collection_method: 'charge_automatically',
         ended_at: nil,
         start: 1308595038,
         object: 'subscription',

lib/stripe_mock/request_handlers/helpers/subscription_helpers.rb

@@ -32,7 +32,7 @@ def custom_subscription_params(plans, cus, options = {})
         start_time = options[:current_period_start] || now
         params = { customer: cus[:id], current_period_start: start_time, created: created_time }
         params.merge!({ :plan => (plans.size == 1 ? plans.first : nil) })
-        keys_to_merge = /application_fee_percent|quantity|metadata|tax_percent|billing|days_until_due|default_tax_rates|pending_invoice_item_interval|default_payment_method/
+        keys_to_merge = /application_fee_percent|quantity|metadata|tax_percent|billing|days_until_due|default_tax_rates|pending_invoice_item_interval|default_payment_method|collection_method/
         params.merge! options.select {|k,v| k =~ keys_to_merge}
 
         if options[:cancel_at_period_end] == true

lib/stripe_mock/request_handlers/subscriptions.rb

@@ -102,7 +102,7 @@ def create_subscription(route, method_url, params, headers)
           customer[:default_source] = new_card[:id]
         end
 
-        allowed_params = %w(customer application_fee_percent coupon items metadata plan quantity source tax_percent trial_end trial_period_days current_period_start created prorate billing_cycle_anchor billing days_until_due idempotency_key enable_incomplete_payments cancel_at_period_end default_tax_rates payment_behavior pending_invoice_item_interval default_payment_method)
+        allowed_params = %w(customer application_fee_percent coupon items metadata plan quantity source tax_percent trial_end trial_period_days current_period_start created prorate billing_cycle_anchor billing days_until_due idempotency_key enable_incomplete_payments cancel_at_period_end default_tax_rates payment_behavior pending_invoice_item_interval default_payment_method collection_method)
         unknown_params = params.keys - allowed_params.map(&:to_sym)
         if unknown_params.length > 0
           raise Stripe::InvalidRequestError.new("Received unknown parameter: #{unknown_params.join}", unknown_params.first.to_s, http_status: 400)

spec/shared_stripe_examples/subscription_examples.rb

@@ -133,12 +133,14 @@
         plan: 'silver',
         customer: customer,
         metadata: { foo: "bar", example: "yes" },
+        collection_method: 'send_invoice',
         default_payment_method: payment_method.id,
       )
 
       subscriptions = Stripe::Subscription.list(customer: customer.id)
       expect(subscriptions.count).to eq(1)
       expect(subscriptions.data.first.id).to eq(sub.id)
+      expect(subscriptions.data.first.collection_method).to eq('send_invoice')
       expect(subscriptions.data.first.default_payment_method).to eq(payment_method.id)
     end
 
@@ -808,9 +810,11 @@
       Stripe::Subscription.update(
         subscription.id,
         default_payment_method: payment_method_sepa.id,
+        collection_method: 'send_invoice',
       )
 
       subscriptions = Stripe::Subscription.list(customer: customer)
+      expect(subscriptions.data.first.collection_method).to eq('send_invoice')
       expect(subscriptions.data.first.default_payment_method).to eq(payment_method_sepa.id)
     end