stripe-ruby-mock 3.0.1 → 3.1.0.rc2

data/lib/stripe_mock/instance.rb

@@ -24,6 +24,8 @@ module StripeMock
     include StripeMock::RequestHandlers::PaymentMethods
     include StripeMock::RequestHandlers::SetupIntents
     include StripeMock::RequestHandlers::ExternalAccounts
+    include StripeMock::RequestHandlers::AccountLinks
+    include StripeMock::RequestHandlers::ExpressLoginLinks
     include StripeMock::RequestHandlers::Accounts
     include StripeMock::RequestHandlers::Balance
     include StripeMock::RequestHandlers::BalanceTransactions
@@ -40,6 +42,7 @@ module StripeMock
     include StripeMock::RequestHandlers::InvoiceItems
     include StripeMock::RequestHandlers::Orders
     include StripeMock::RequestHandlers::Plans
+    include StripeMock::RequestHandlers::Prices
     include StripeMock::RequestHandlers::Products
     include StripeMock::RequestHandlers::Refunds
     include StripeMock::RequestHandlers::Recipients
@@ -50,10 +53,11 @@ module StripeMock
     include StripeMock::RequestHandlers::EphemeralKey
     include StripeMock::RequestHandlers::TaxRates
     include StripeMock::RequestHandlers::Checkout
+    include StripeMock::RequestHandlers::Checkout::Session
 
     attr_reader :accounts, :balance, :balance_transactions, :bank_tokens, :charges, :coupons, :customers,
                 :disputes, :events, :invoices, :invoice_items, :orders, :payment_intents, :payment_methods,
-                :setup_intents, :plans, :recipients, :refunds, :transfers, :payouts, :subscriptions, :country_spec,
+                :setup_intents, :plans, :prices, :recipients, :refunds, :transfers, :payouts, :subscriptions, :country_spec,
                 :subscriptions_items, :products, :tax_rates, :checkout_sessions
 
     attr_accessor :error_queue, :debug, :conversion_rate, :account_balance
@@ -64,7 +68,7 @@ module StripeMock
       @balance_transactions = Data.mock_balance_transactions(['txn_05RsQX2eZvKYlo2C0FRTGSSA','txn_15RsQX2eZvKYlo2C0ERTYUIA', 'txn_25RsQX2eZvKYlo2C0ZXCVBNM', 'txn_35RsQX2eZvKYlo2C0QAZXSWE', 'txn_45RsQX2eZvKYlo2C0EDCVFRT', 'txn_55RsQX2eZvKYlo2C0OIKLJUY', 'txn_65RsQX2eZvKYlo2C0ASDFGHJ', 'txn_75RsQX2eZvKYlo2C0EDCXSWQ', 'txn_85RsQX2eZvKYlo2C0UJMCDET', 'txn_95RsQX2eZvKYlo2C0EDFRYUI'])
       @bank_tokens = {}
       @card_tokens = {}
-      @customers = {}
+      @customers = { Stripe.api_key => {} }
       @charges = {}
       @payment_intents = {}
       @payment_methods = {}
@@ -77,6 +81,7 @@ module StripeMock
       @orders = {}
       @payment_methods = {}
       @plans = {}
+      @prices = {}
       @products = {}
       @recipients = {}
       @refunds = {}

data/lib/stripe_mock/request_handlers/account_links.rb

@@ -0,0 +1,15 @@
+module StripeMock
+  module RequestHandlers
+    module AccountLinks
+
+      def AccountLinks.included(klass)
+        klass.add_handler 'post /v1/account_links',      :new_account_link
+      end
+
+      def new_account_link(route, method_url, params, headers)
+        route =~ method_url
+        Data.mock_account_link(params)
+      end
+    end
+  end
+end

data/lib/stripe_mock/request_handlers/charges.rb

@@ -12,7 +12,9 @@ module StripeMock
         klass.add_handler 'post /v1/charges/(.*)',          :update_charge
       end
 
-      def new_charge(route, method_url, params, headers)
+      def new_charge(route, method_url, params, headers = {})
+        stripe_account = headers && headers[:stripe_account] || Stripe.api_key
+
         if headers && headers[:idempotency_key]
           params[:idempotency_key] = headers[:idempotency_key]
           if charges.any?
@@ -29,7 +31,7 @@ module StripeMock
             # card id, not a token. in this case we'll find the card in the customer
             # object and return that.
             if params[:customer]
-              params[:source] = get_card(customers[params[:customer]], params[:source])
+              params[:source] = get_card(customers[stripe_account][params[:customer]], params[:source])
             else
               params[:source] = get_card_or_bank_by_token(params[:source])
             end
@@ -37,7 +39,7 @@ module StripeMock
             raise Stripe::InvalidRequestError.new("Invalid token id: #{params[:source]}", 'card', http_status: 400)
           end
         elsif params[:customer]
-          customer = customers[params[:customer]]
+          customer = customers[stripe_account][params[:customer]]
           if customer && customer[:default_source]
             params[:source] = get_card(customer, customer[:default_source])
           end
@@ -146,7 +148,7 @@ module StripeMock
         elsif non_positive_charge_amount?(params)
           raise Stripe::InvalidRequestError.new('Invalid positive integer', 'amount', http_status: 400)
         elsif params[:source].nil? && params[:customer].nil?
-          raise Stripe::InvalidRequestError.new('Must provide source or customer.', http_status: nil)
+          raise Stripe::InvalidRequestError.new('Must provide source or customer.', nil, http_status: nil)
         end
       end
 

data/lib/stripe_mock/request_handlers/checkout_session.rb

@@ -0,0 +1,16 @@
+module StripeMock
+  module RequestHandlers
+    module Checkout
+      module Session
+        def Session.included(klass)
+          klass.add_handler 'get /v1/checkout/sessions/(.*)', :get_checkout_session
+        end
+
+        def get_checkout_session(route, method_url, params, headers)
+          route =~ method_url
+          assert_existence :checkout_session, $1, checkout_sessions[$1]
+        end
+      end
+    end
+  end
+end

data/lib/stripe_mock/request_handlers/customers.rb

@@ -12,6 +12,7 @@ module StripeMock
       end
 
       def new_customer(route, method_url, params, headers)
+        stripe_account = headers && headers[:stripe_account] || Stripe.api_key
         params[:id] ||= new_id('cus')
         sources = []
 
@@ -29,7 +30,8 @@ module StripeMock
           params[:default_source] = sources.first[:id]
         end
 
-        customers[params[:id]] = Data.mock_customer(sources, params)
+        customers[stripe_account] ||= {}
+        customers[stripe_account][params[:id]] = Data.mock_customer(sources, params)
 
         if params[:plan]
           plan_id = params[:plan].to_s
@@ -40,8 +42,8 @@ module StripeMock
           end
 
           subscription = Data.mock_subscription({ id: new_id('su') })
-          subscription = resolve_subscription_changes(subscription, [plan], customers[ params[:id] ], params)
-          add_subscription_to_customer(customers[ params[:id] ], subscription)
+          subscription = resolve_subscription_changes(subscription, [plan], customers[stripe_account][params[:id]], params)
+          add_subscription_to_customer(customers[stripe_account][params[:id]], subscription)
           subscriptions[subscription[:id]] = subscription
         elsif params[:trial_end]
           raise Stripe::InvalidRequestError.new('Received unknown parameter: trial_end', nil, http_status: 400)
@@ -50,15 +52,16 @@ module StripeMock
         if params[:coupon]
           coupon = coupons[params[:coupon]]
           assert_existence :coupon, params[:coupon], coupon
-          add_coupon_to_object(customers[params[:id]], coupon)
+          add_coupon_to_object(customers[stripe_account][params[:id]], coupon)
         end
 
-        customers[params[:id]]
+        customers[stripe_account][params[:id]]
       end
 
       def update_customer(route, method_url, params, headers)
+        stripe_account = headers && headers[:stripe_account] || Stripe.api_key
         route =~ method_url
-        cus = assert_existence :customer, $1, customers[$1]
+        cus = assert_existence :customer, $1, customers[stripe_account][$1]
 
         # get existing and pending metadata
         metadata = cus.delete(:metadata) || {}
@@ -80,6 +83,8 @@ module StripeMock
         if params[:source]
           if params[:source].is_a?(String)
             new_card = get_card_or_bank_by_token(params.delete(:source))
+          elsif params[:source].is_a?(Stripe::Token)
+            new_card = get_card_or_bank_by_token(params[:source][:id])
           elsif params[:source].is_a?(Hash)
             unless params[:source][:object] && params[:source][:number] && params[:source][:exp_month] && params[:source][:exp_year]
               raise Stripe::InvalidRequestError.new('You must supply a valid card', nil, http_status: 400)
@@ -105,21 +110,23 @@ module StripeMock
       end
 
       def delete_customer(route, method_url, params, headers)
+        stripe_account = headers && headers[:stripe_account] || Stripe.api_key
         route =~ method_url
-        assert_existence :customer, $1, customers[$1]
+        assert_existence :customer, $1, customers[stripe_account][$1]
 
-        customers[$1] = {
-          id: customers[$1][:id],
+        customers[stripe_account][$1] = {
+          id: customers[stripe_account][$1][:id],
           deleted: true
         }
       end
 
       def get_customer(route, method_url, params, headers)
+        stripe_account = headers && headers[:stripe_account] || Stripe.api_key
         route =~ method_url
-        customer = assert_existence :customer, $1, customers[$1]
+        customer = assert_existence :customer, $1, customers[stripe_account][$1]
 
         customer = customer.clone
-        if params[:expand] == ['default_source']
+        if params[:expand] == ['default_source'] && customer[:sources][:data]
           customer[:default_source] = customer[:sources][:data].detect do |source|
             source[:id] == customer[:default_source]
           end
@@ -129,12 +136,14 @@ module StripeMock
       end
 
       def list_customers(route, method_url, params, headers)
-        Data.mock_list_object(customers.values, params)
+        stripe_account = headers && headers[:stripe_account] || Stripe.api_key
+        Data.mock_list_object(customers[stripe_account]&.values, params)
       end
 
       def delete_customer_discount(route, method_url, params, headers)
+        stripe_account = headers && headers[:stripe_account] || Stripe.api_key
         route =~ method_url
-        cus = assert_existence :customer, $1, customers[$1]
+        cus = assert_existence :customer, $1, customers[stripe_account][$1]
 
         cus[:discount] = nil
 

data/lib/stripe_mock/request_handlers/ephemeral_key.rb

@@ -6,7 +6,7 @@ module StripeMock
       end
 
       def create_ephemeral_key(route, method_url, params, headers)
-        Data.mock_ephemeral_key(params)
+        Data.mock_ephemeral_key(**params)
       end
     end
   end

data/lib/stripe_mock/request_handlers/express_login_links.rb

@@ -0,0 +1,15 @@
+module StripeMock
+  module RequestHandlers
+    module ExpressLoginLinks
+
+      def ExpressLoginLinks.included(klass)
+        klass.add_handler 'post /v1/accounts/(.*)/login_links', :new_account_login_link
+      end
+
+      def new_account_login_link(route, method_url, params, headers)
+        route =~ method_url
+        Data.mock_express_login_link(params)
+      end
+    end
+  end
+end

data/lib/stripe_mock/request_handlers/helpers/subscription_helpers.rb

@@ -11,12 +11,13 @@ module StripeMock
         items = options[:items]
         items = items.values if items.respond_to?(:values)
         subscription[:items][:data] = plans.map do |plan|
-          if items && items.size == plans.size
-            quantity = items &&
-              items.detect { |item| item[:plan] == plan[:id] }[:quantity] || 1
-            Data.mock_subscription_item({ plan: plan, quantity: quantity })
+          matching_item = items && items.detect { |item| [item[:price], item[:plan]].include? plan[:id] }
+          if matching_item
+            quantity = matching_item[:quantity] || 1
+            id = matching_item[:id] || new_id('si')
+            Data.mock_subscription_item({ plan: plan, quantity: quantity, id: id })
           else
-            Data.mock_subscription_item({ plan: plan })
+            Data.mock_subscription_item({ plan: plan, id: new_id('si') })
           end
         end
         subscription
@@ -32,7 +33,7 @@ module StripeMock
         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/
+        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
@@ -111,7 +112,11 @@ module StripeMock
 
       def total_items_amount(items)
         total = 0
-        items.each { |i| total += (i[:quantity] || 1) * i[:plan][:amount] }
+        items.each do |item|
+          quantity = item[:quantity] || 1
+          amount = item[:plan][:unit_amount] || item[:plan][:amount]
+          total += quantity * amount
+        end
         total
       end
     end

data/lib/stripe_mock/request_handlers/invoices.rb

@@ -56,13 +56,14 @@ module StripeMock
         invoices[$1].merge!(:paid => true, :attempted => true, :charge => charge[:id])
       end
 
-      def upcoming_invoice(route, method_url, params, headers)
+      def upcoming_invoice(route, method_url, params, headers = {})
+        stripe_account = headers && headers[:stripe_account] || Stripe.api_key
         route =~ method_url
-        raise Stripe::InvalidRequestError.new('Missing required param: customer', nil, http_status: 400) if params[:customer].nil?
+        raise Stripe::InvalidRequestError.new('Missing required param: customer if subscription is not provided', nil, http_status: 400) if params[:customer].nil? && params[:subscription].nil?
         raise Stripe::InvalidRequestError.new('When previewing changes to a subscription, you must specify either `subscription` or `subscription_items`', nil, http_status: 400) if !params[:subscription_proration_date].nil? && params[:subscription].nil? && params[:subscription_plan].nil?
         raise Stripe::InvalidRequestError.new('Cannot specify proration date without specifying a subscription', nil, http_status: 400) if !params[:subscription_proration_date].nil? && params[:subscription].nil?
 
-        customer = customers[params[:customer]]
+        customer = customers[stripe_account][params[:customer]]
         assert_existence :customer, params[:customer], customer
 
         raise Stripe::InvalidRequestError.new("No upcoming invoices for customer: #{customer[:id]}", nil, http_status: 404) if customer[:subscriptions][:data].length == 0

data/lib/stripe_mock/request_handlers/payment_methods.rb

@@ -51,14 +51,15 @@ module StripeMock
 
         Data.mock_list_object(clone.values, params)
       end
-      
+
       # post /v1/payment_methods/:id/attach
       def attach_payment_method(route, method_url, params, headers)
+        stripe_account = headers && headers[:stripe_account] || Stripe.api_key
         allowed_params = [:customer]
 
         id = method_url.match(route)[1]
 
-        assert_existence :customer, params[:customer], customers[params[:customer]]
+        assert_existence :customer, params[:customer], customers[stripe_account][params[:customer]]
 
         payment_method = assert_existence :payment_method, id, payment_methods[id]
         payment_methods[id] = Util.rmerge(payment_method, params.select { |k, _v| allowed_params.include?(k) })
@@ -77,7 +78,7 @@ module StripeMock
 
       # post /v1/payment_methods/:id
       def update_payment_method(route, method_url, params, headers)
-        allowed_params = [:billing_details, :card, :metadata]
+        allowed_params = [:billing_details, :card, :ideal, :sepa_debit, :metadata]
 
         id = method_url.match(route)[1]
 
@@ -86,6 +87,7 @@ module StripeMock
         if payment_method[:customer].nil?
           raise Stripe::InvalidRequestError.new(
             'You must save this PaymentMethod to a customer before you can update it.',
+            nil,
             http_status: 400
           )
         end
@@ -103,14 +105,15 @@ module StripeMock
 
         if invalid_type?(params[:type])
           raise Stripe::InvalidRequestError.new(
-            'Invalid type: must be one of card or card_present',
+            'Invalid type: must be one of card, ideal or sepa_debit',
+            nil,
             http_status: 400
           )
         end
       end
 
       def invalid_type?(type)
-        !['card', 'card_present'].include?(type)
+        !%w(card ideal sepa_debit).include?(type)
       end
     end
   end

data/lib/stripe_mock/request_handlers/prices.rb

@@ -0,0 +1,44 @@
+module StripeMock
+  module RequestHandlers
+    module Prices
+
+      def Prices.included(klass)
+        klass.add_handler 'post /v1/prices',        :new_price
+        klass.add_handler 'post /v1/prices/(.*)',   :update_price
+        klass.add_handler 'get /v1/prices/(.*)',    :get_price
+        klass.add_handler 'get /v1/prices',         :list_prices
+      end
+
+      def new_price(route, method_url, params, headers)
+        params[:id] ||= new_id('price')
+        validate_create_price_params(params)
+        prices[ params[:id] ] = Data.mock_price(params)
+      end
+
+      def update_price(route, method_url, params, headers)
+        route =~ method_url
+        assert_existence :price, $1, prices[$1]
+        prices[$1].merge!(params)
+      end
+
+      def get_price(route, method_url, params, headers)
+        route =~ method_url
+        assert_existence :price, $1, prices[$1]
+      end
+
+      def list_prices(route, method_url, params, headers)
+        limit = params[:limit] ? params[:limit] : 10
+        price_data = prices.values
+        validate_list_prices_params(params)
+
+        if params.key?(:lookup_keys)
+          price_data.select! do |price|
+            params[:lookup_keys].include?(price[:lookup_key])
+          end
+        end
+
+        Data.mock_list_object(price_data.first(limit), params.merge!(limit: limit))
+      end
+    end
+  end
+end

data/lib/stripe_mock/request_handlers/sources.rb

@@ -12,30 +12,35 @@ module StripeMock
       end
 
       def create_source(route, method_url, params, headers)
+        stripe_account = headers && headers[:stripe_account] || Stripe.api_key
         route =~ method_url
-        add_source_to(:customer, $1, params, customers)
+        add_source_to(:customer, $1, params, customers[stripe_account])
       end
 
       def retrieve_sources(route, method_url, params, headers)
+        stripe_account = headers && headers[:stripe_account] || Stripe.api_key
         route =~ method_url
-        retrieve_object_cards(:customer, $1, customers)
+        retrieve_object_cards(:customer, $1, customers[stripe_account])
       end
 
       def retrieve_source(route, method_url, params, headers)
+        stripe_account = headers && headers[:stripe_account] || Stripe.api_key
         route =~ method_url
-        customer = assert_existence :customer, $1, customers[$1]
+        customer = assert_existence :customer, $1, customers[stripe_account][$1]
 
         assert_existence :card, $2, get_card(customer, $2)
       end
 
       def delete_source(route, method_url, params, headers)
+        stripe_account = headers && headers[:stripe_account] || Stripe.api_key
         route =~ method_url
-        delete_card_from(:customer, $1, $2, customers)
+        delete_card_from(:customer, $1, $2, customers[stripe_account])
       end
 
       def update_source(route, method_url, params, headers)
+        stripe_account = headers && headers[:stripe_account] || Stripe.api_key
         route =~ method_url
-        customer = assert_existence :customer, $1, customers[$1]
+        customer = assert_existence :customer, $1, customers[stripe_account][$1]
 
         card = assert_existence :card, $2, get_card(customer, $2)
         card.merge!(params)
@@ -43,8 +48,9 @@ module StripeMock
       end
 
       def verify_source(route, method_url, params, headers)
+        stripe_account = headers && headers[:stripe_account] || Stripe.api_key
         route =~ method_url
-        customer = assert_existence :customer, $1, customers[$1]
+        customer = assert_existence :customer, $1, customers[stripe_account][$1]
 
         bank_account = assert_existence :bank_account, $2, verify_bank_account(customer, $2)
         bank_account