stripe-ruby-mock 2.5.6 → 3.1.0.rc2

data/lib/stripe_mock/data/list.rb

@@ -1,19 +1,17 @@
 module StripeMock
   module Data
     class List
-      attr_reader :data, :limit, :offset, :starting_after, :ending_before
+      attr_reader :data, :limit, :offset, :starting_after, :ending_before, :active
 
       def initialize(data, options = {})
         @data = Array(data.clone)
         @limit = [[options[:limit] || 10, 100].min, 1].max # restrict @limit to 1..100
         @starting_after = options[:starting_after]
         @ending_before  = options[:ending_before]
-        if @data.first.is_a?(Hash) && @data.first[:created]
-          @data.sort_by! { |x| x[:created] }
-          @data.reverse!
-        elsif @data.first.respond_to?(:created)
-          @data.sort_by { |x| x.created }
-          @data.reverse!
+        @active = options[:active]
+        if contains_stripe_objects?
+          prune_deleted_data
+          sort_data
         end
       end
 
@@ -53,14 +51,21 @@ module StripeMock
           (index || raise("No such object id: #{starting_after}")) + 1
         when ending_before
           index = data.index { |datum| datum[:id] == ending_before }
-          (index || raise("No such object id: #{ending_before}")) - 1          
+          (index || raise("No such object id: #{ending_before}")) - 1
         else
           0
         end
       end
 
       def data_page
-        data[offset, limit]
+        filtered_data[offset, limit]
+      end
+
+      def filtered_data
+        filtered_data = data
+        filtered_data = filtered_data.select { |d| d[:active] == active } unless active.nil?
+
+        filtered_data
       end
 
       def object_types
@@ -68,6 +73,34 @@ module StripeMock
           "#{first_object.class.to_s.split('::')[-1].downcase}s"
         end
       end
+
+      def contains_stripe_objects?
+        return false if data.empty?
+
+        object = data.first
+        object.is_a?(Stripe::StripeObject) || (
+          object.is_a?(Hash) && [:created, :deleted].any? { |k| object.key?(k) }
+        )
+      end
+
+      def prune_deleted_data
+        data.reject! do |object|
+          (object.is_a?(Hash) && object[:deleted]) ||
+            (object.is_a?(Stripe::StripeObject) && object.deleted?)
+        end
+      end
+
+      def sort_data
+        # Reverse must follow sort to preserve existing test dependencies. The
+        # alternative would be to simply reverse lhs and rhs in the comparison,
+        # however, being a stable sort this breaks the existing dependency when
+        # more than one record share the same `created` value.
+        @data = data.sort { |lhs, rhs| sort_val(lhs) <=> sort_val(rhs) }.reverse
+      end
+
+      def sort_val(object)
+        object.is_a?(Stripe::StripeObject) ? object.created : object[:created]
+      end
     end
   end
 end

data/lib/stripe_mock/instance.rb

@@ -20,7 +20,12 @@ module StripeMock
       @@handlers.find {|h| method_url =~ h[:route] }
     end
 
+    include StripeMock::RequestHandlers::PaymentIntents
+    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
@@ -28,6 +33,7 @@ module StripeMock
     include StripeMock::RequestHandlers::Cards
     include StripeMock::RequestHandlers::Sources
     include StripeMock::RequestHandlers::Subscriptions # must be before Customers
+    include StripeMock::RequestHandlers::SubscriptionItems
     include StripeMock::RequestHandlers::Customers
     include StripeMock::RequestHandlers::Coupons
     include StripeMock::RequestHandlers::Disputes
@@ -36,6 +42,8 @@ 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
     include StripeMock::RequestHandlers::Transfers
@@ -43,10 +51,14 @@ module StripeMock
     include StripeMock::RequestHandlers::CountrySpec
     include StripeMock::RequestHandlers::Payouts
     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, :plans, :recipients,
-                :refunds, :transfers, :payouts, :subscriptions, :country_spec, :subscriptions_items
+                :disputes, :events, :invoices, :invoice_items, :orders, :payment_intents, :payment_methods,
+                :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
 
@@ -56,22 +68,30 @@ 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 = {}
+      @setup_intents = {}
       @coupons = {}
       @disputes = Data.mock_disputes(['dp_05RsQX2eZvKYlo2C0FRTGSSA','dp_15RsQX2eZvKYlo2C0ERTYUIA', 'dp_25RsQX2eZvKYlo2C0ZXCVBNM', 'dp_35RsQX2eZvKYlo2C0QAZXSWE', 'dp_45RsQX2eZvKYlo2C0EDCVFRT', 'dp_55RsQX2eZvKYlo2C0OIKLJUY', 'dp_65RsQX2eZvKYlo2C0ASDFGHJ', 'dp_75RsQX2eZvKYlo2C0EDCXSWQ', 'dp_85RsQX2eZvKYlo2C0UJMCDET', 'dp_95RsQX2eZvKYlo2C0EDFRYUI'])
       @events = {}
       @invoices = {}
       @invoice_items = {}
       @orders = {}
+      @payment_methods = {}
       @plans = {}
+      @prices = {}
+      @products = {}
       @recipients = {}
       @refunds = {}
       @transfers = {}
       @payouts = {}
       @subscriptions = {}
-      @subscriptions_items = []
+      @subscriptions_items = {}
       @country_spec = {}
+      @tax_rates = {}
+      @checkout_sessions = {}
 
       @debug = false
       @error_queue = ErrorQueue.new
@@ -175,7 +195,8 @@ module StripeMock
       amount = params[:amount]
       unless amount.nil?
         # Fee calculation
-        params[:fee] ||= (30 + (amount.abs * 0.029).ceil) * (amount > 0 ? 1 : -1)
+        calculate_fees(params) unless params[:fee]
+        params[:net] = amount - params[:fee]
         params[:amount] = amount * @conversion_rate
       end
       @balance_transactions[id] = Data.mock_balance_transaction(params.merge(id: id))
@@ -196,5 +217,31 @@ module StripeMock
       response = Struct.new(:data)
       response.new(hash)
     end
+
+    def calculate_fees(params)
+      application_fee = params[:application_fee] || 0
+      params[:fee] = processing_fee(params[:amount]) + application_fee
+      params[:fee_details] = [
+        {
+          amount: processing_fee(params[:amount]),
+          application: nil,
+          currency: params[:currency] || StripeMock.default_currency,
+          description: "Stripe processing fees",
+          type: "stripe_fee"
+        }
+      ]
+      if application_fee
+        params[:fee_details] << {
+          amount: application_fee,
+          currency: params[:currency] || StripeMock.default_currency,
+          description: "Application fee",
+          type: "application_fee"
+        }
+      end
+    end
+
+    def processing_fee(amount)
+      (30 + (amount.abs * 0.029).ceil) * (amount > 0 ? 1 : -1)
+    end
   end
 end

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/balance_transactions.rb

@@ -3,8 +3,8 @@ module StripeMock
     module BalanceTransactions
 
       def BalanceTransactions.included(klass)
-        klass.add_handler 'get /v1/balance/history/(.*)',  :get_balance_transaction
-        klass.add_handler 'get /v1/balance/history',       :list_balance_transactions
+        klass.add_handler 'get /v1/balance_transactions/(.*)',  :get_balance_transaction
+        klass.add_handler 'get /v1/balance_transactions',       :list_balance_transactions
       end
 
       def get_balance_transaction(route, method_url, params, headers)

data/lib/stripe_mock/request_handlers/charges.rb

@@ -12,10 +12,15 @@ module StripeMock
         klass.add_handler 'post /v1/charges/(.*)',          :update_charge
       end
 
-      def new_charge(route, method_url, params, headers)
-        if params[:idempotency_key] && charges.any?
-          original_charge = charges.values.find { |c| c[:idempotency_key] == params[:idempotency_key]}
-          return charges[original_charge[:id]] if original_charge
+      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?
+            original_charge = charges.values.find { |c| c[:idempotency_key] == headers[:idempotency_key]}
+            return charges[original_charge[:id]] if original_charge
+          end
         end
 
         id = new_id('ch')
@@ -26,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
@@ -34,14 +39,14 @@ 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
         end
 
         ensure_required_params(params)
-        bal_trans_params = { amount: params[:amount], source: id }
+        bal_trans_params = { amount: params[:amount], source: id, application_fee: params[:application_fee] }
 
         balance_transaction_id = new_balance_transaction('txn', bal_trans_params)
 
@@ -143,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
 
@@ -155,10 +160,6 @@ module StripeMock
         params[:amount] && params[:amount] < 1
       end
 
-      def require_param(param)
-        raise Stripe::InvalidRequestError.new("Missing required param: #{param}", param.to_s, http_status: 400)
-      end
-
       def allowed_params(params)
         allowed = [:description, :metadata, :receipt_email, :fraud_details, :shipping, :destination]
 

data/lib/stripe_mock/request_handlers/checkout.rb

@@ -0,0 +1,15 @@
+module StripeMock
+  module RequestHandlers
+    module Checkout
+      def Checkout.included(klass)
+        klass.add_handler 'post /v1/checkout/sessions', :new_session
+      end
+
+      def new_session(route, method_url, params, headers)
+        params[:id] ||= new_id('cs')
+
+        checkout_sessions[params[:id]] = Data.mock_checkout_session(params)
+      end
+    end
+  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,26 +42,30 @@ 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)
         end
 
         if params[:coupon]
-          coupon = coupons[ params[:coupon] ]
+          coupon = coupons[params[:coupon]]
           assert_existence :coupon, params[:coupon], coupon
-
-          add_coupon_to_customer(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) || {}
+        metadata_updates = params.delete(:metadata) || {}
 
         # Delete those params if their value is nil. Workaround of the problematic way Stripe serialize objects
         params.delete(:sources) if params[:sources] && params[:sources][:data].nil?
@@ -72,10 +78,13 @@ module StripeMock
           params.delete(:subscriptions) unless params[:subscriptions][:data].any?{ |v| !!v[:type]}
         end
         cus.merge!(params)
+        cus[:metadata] = {**metadata, **metadata_updates}
 
         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)
@@ -87,31 +96,37 @@ module StripeMock
         end
 
         if params[:coupon]
-          coupon = coupons[ params[:coupon] ]
-          assert_existence :coupon, params[:coupon], coupon
+          if params[:coupon] == ''
+            delete_coupon_from_object(cus)
+          else
+            coupon = coupons[params[:coupon]]
+            assert_existence :coupon, params[:coupon], coupon
 
-          add_coupon_to_customer(cus, coupon)
+            add_coupon_to_object(cus, coupon)
+          end
         end
 
         cus
       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
@@ -121,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