stripe-ruby-mock 3.0.1 → 3.1.0

data/lib/stripe_mock/data.rb

@@ -101,6 +101,27 @@ module StripeMock
       }.merge(params)
     end
 
+    def self.mock_account_link(params = {})
+      now = Time.now.to_i
+      {
+        object: 'account_link',
+        created: now,
+        expires_at: now + 300,
+        url: 'https://connect.stripe.com/setup/c/iB0ph1cPnRLY',
+        data: {}
+      }.merge(params)
+    end
+
+    def self.mock_express_login_link(params = {})
+      now = Time.now.to_i
+      {
+        object: 'login_link',
+        created: now,
+        url: 'https://connect.stripe.com/express/Ln7FfnNpUcCU',
+        data: {}
+      }.merge(params)
+    end
+
     def self.mock_tax_rate(params)
       {
         id: 'test_cus_default',
@@ -119,7 +140,7 @@ module StripeMock
 
     def self.mock_customer(sources, params)
       cus_id = params[:id] || "test_cus_default"
-      currency = params[:currency] || StripeMock.default_currency
+      currency = params[:currency]
       sources.each {|source| source[:customer] = cus_id}
       {
         email: 'stripe_mock@example.com',
@@ -142,12 +163,14 @@ module StripeMock
         sources: {
           object: "list",
           total_count: sources.size,
+          has_more: false,
           url: "/v1/customers/#{cus_id}/sources",
           data: sources
         },
         subscriptions: {
           object: "list",
           total_count: 0,
+          has_more: false,
           url: "/v1/customers/#{cus_id}/subscriptions",
           data: []
         },
@@ -166,11 +189,34 @@ module StripeMock
         paid: true,
         amount: 0,
         application_fee: nil,
+        application_fee_amount: nil,
         currency: currency,
         destination: nil,
         fraud_details: {},
+        payment_method_details: {
+          card: {
+            brand: "visa",
+            checks: {
+              address_line1_check: nil,
+              address_postal_code_check: nil,
+              cvc_check: "pass"
+            },
+            country: "US",
+            exp_month: 12,
+            exp_year: 2013,
+            fingerprint: "3TQGpK9JoY1GgXPw",
+            funding: "credit",
+            installments: nil,
+            last4: "4242",
+            network: "visa",
+            three_d_secure: nil,
+            wallet: nil
+          },
+          type: "card"
+        },
         receipt_email: nil,
         receipt_number: nil,
+        receipt_url: nil,
         refunded: false,
         shipping: {},
         statement_descriptor: "Charge #{charge_id}",
@@ -231,7 +277,8 @@ module StripeMock
         charge: "ch_4fWhYjzQ23UFWT",
         receipt_number: nil,
         status: "succeeded",
-        reason: "requested_by_customer"
+        reason: "requested_by_customer",
+        receipt_url: nil
       }.merge(params)
     end
 
@@ -339,8 +386,9 @@ module StripeMock
         },
         cancel_at_period_end: false,
         canceled_at: nil,
+        collection_method: 'charge_automatically',
         ended_at: nil,
-        start: 1308595038,
+        start_date: 1308595038,
         object: 'subscription',
         trial_start: 1308595038,
         trial_end: 1308681468,
@@ -350,8 +398,10 @@ module StripeMock
         discount: nil,
         metadata: {},
         default_tax_rates: nil,
+        default_payment_method: nil,
         pending_invoice_item_interval: nil,
-        next_pending_invoice_item_invoice: nil
+        next_pending_invoice_item_invoice: nil,
+        latest_invoice: nil
       }, params)
     end
 
@@ -371,6 +421,7 @@ module StripeMock
         lines: {
           object: "list",
           total_count: lines.count,
+          has_more: false,
           url: "/v1/invoices/#{in_id}/lines",
           data: lines
         },
@@ -399,7 +450,8 @@ module StripeMock
         next_payment_attempt: 1349825350,
         charge: nil,
         discount: nil,
-        subscription: nil
+        subscription: nil,
+        number: "6C41730-0001"
       }.merge(params)
       if invoice[:discount]
         invoice[:total] = [0, invoice[:subtotal] - invoice[:discount][:coupon][:amount_off]].max if invoice[:discount][:coupon][:amount_off]
@@ -554,6 +606,34 @@ module StripeMock
       }.merge(params)
     end
 
+    def self.mock_price(params={})
+      currency = params[:currency] || StripeMock.default_currency
+      {
+        id: "mock_price_123",
+        object: "price",
+        active: true,
+        billing_scheme: "per_unit",
+        created: 1593044959,
+        currency: currency,
+        livemode: false,
+        lookup_key: nil,
+        metadata: {},
+        nickname: 'My Mock Price',
+        product: "mock_prod_NONEXIST",  # override this with your own existing product id
+        recurring: {
+          aggregate_usage: nil,
+          interval: "month",
+          interval_count: 1,
+          usage_type: "licensed"
+        },
+        tiers_mode: nil,
+        transform_quantity: nil,
+        type: "recurring",
+        unit_amount: 2000,
+        unit_amount_decimal: "2000"
+      }.merge(params)
+    end
+
     def self.mock_product(params={})
       {
         id: "mock_prod_abc123",
@@ -578,6 +658,44 @@ module StripeMock
       }.merge(params)
     end
 
+    def self.mock_promotion_code(params={})
+      {
+        id: "mock_promo_abc123",
+        object: "promotion_code",
+        active: true,
+        code: "TESTCODE",
+        coupon: {
+          id: "mock_coupon_abc123",
+          object: "coupon",
+          amount_off: nil,
+          created: 1665773498,
+          currency: "usd",
+          duration: "repeating",
+          duration_in_months: 1,
+          livemode: false,
+          max_redemptions: nil,
+          metadata: {},
+          name: "Mock Coupon",
+          percent_off: 10.0,
+          redeem_by: nil,
+          times_redeemed: 0,
+          valid: true
+        },
+        created: 1665773499,
+        customer: nil,
+        expires_at: nil,
+        livemode: false,
+        max_redemptions: nil,
+        metadata: {},
+        restrictions: {
+          first_time_transaction: false,
+          minimum_amount: nil,
+          minimum_amount_currency: nil
+        },
+        times_redeemed: 0
+      }.merge(params)
+    end
+
     def self.mock_recipient(cards, params={})
       rp_id = params[:id] || "test_rp_default"
       cards.each {|card| card[:recipient] = rp_id}
@@ -601,6 +719,7 @@ module StripeMock
           object: "list",
           url: "/v1/recipients/#{rp_id}/cards",
           data: cards,
+          has_more: false,
           total_count: cards.count
         },
         default_card: nil
@@ -1008,6 +1127,16 @@ module StripeMock
               bitcoin_receiver: 1545182
             }
           }],
+        instant_available: [
+          {
+            currency: "usd",
+            amount: usd_balance,
+            source_types: {
+              card: 25907032203,
+              bank_account: 108476658,
+              bitcoin_receiver: 1545182
+            }
+          }],
         connect_reserved: [
           {
             currency: "usd",
@@ -1072,13 +1201,12 @@ module StripeMock
     end
 
     def self.mock_subscription_item(params = {})
-      iid = params[:id] || 'test_txn_default'
+      id = params[:id] || 'test_si_default'
       {
-        id: iid,
+        id: id,
         object: 'subscription_item',
         created: 1504716183,
-        metadata: {
-      },
+        metadata: {},
         plan: {
           id: 'PER_USER_PLAN1',
           object: 'plan',
@@ -1167,29 +1295,72 @@ module StripeMock
     end
 
     def self.mock_payment_method(params = {})
-      payment_method_id = params[:id] || "pm_1ExEuFL2DI6wht39WNJgbybl"
-      {
-          id: payment_method_id,
-          object: "payment_method",
-          type: "card",
-          billing_details: {},
-          card: {
-              brand: "visa",
-              checks: { address_line1_check: nil, address_postal_code_check: nil, cvc_check: "pass" },
-              country: "FR",
-              exp_month: 2,
-              exp_year: 2022,
-              fingerprint: "Hr3Ly5z5IYxsokWA",
-              funding: "credit",
-              last4: "3155",
-              three_d_secure_usage: { supported: true }
+      payment_method_id = params[:id] || 'pm_1ExEuFL2DI6wht39WNJgbybl'
+
+      type = params[:type].to_sym
+      last4 = params.dig(:card, :number)
+      data = {
+        card: {
+          brand: case last4&.to_s
+          when /^4/, nil
+            'visa'
+          when /^5[1-5]/
+            'mastercard'
+          else
+            'unknown'
+          end,
+          checks: {
+            address_line1_check: nil,
+            address_postal_code_check: nil,
+            cvc_check: 'pass'
           },
-          customer: params[:customer] || nil,
-          metadata: {
-            order_id: "123456789"
-          }
+          country: 'FR',
+          exp_month: params.dig(:card, :exp_month) || 2,
+          exp_year: params.dig(:card, :exp_year) || 2022,
+          fingerprint: 'Hr3Ly5z5IYxsokWA',
+          funding: 'credit',
+          generated_from: nil,
+          last4: last4.nil? ? '3155' : last4.to_s[-4..],
+          three_d_secure_usage: { supported: true },
+          wallet: nil
+        },
+        ideal: {
+          bank: 'ing',
+          bic: 'INGBNL2A',
+          iban_last4: '****',
+          verified_name: 'JENNY ROSEN'
+        },
+        sepa_debit: {
+          bank_code: '37040044',
+          branch_code: '',
+          country: 'DE',
+          fingerprint: 'FD81kbVPe7M05BMj',
+          last4: params.dig(:sepa_debit, :iban)&.[](-4..) || '3000'
+        }
+      }
 
-      }.merge(params)
+      {
+        id: payment_method_id,
+        object: 'payment_method',
+        type: params[:type],
+        billing_details: {
+          address: {
+            city: 'New Orleans',
+            country: 'US',
+            line1: 'Bourbon Street 23',
+            line2: nil,
+            postal_code: '10000',
+            state: nil
+          },
+          email: 'foo@bar.com',
+          name: 'John Dolton',
+          phone: nil
+        },
+        customer: params[:customer] || nil,
+        metadata: {
+          order_id: '123456789'
+        }
+      }.merge(params).merge(type => data[type])
     end
 
     def self.mock_setup_intent(params = {})

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,7 +42,9 @@ 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::PromotionCodes
     include StripeMock::RequestHandlers::Refunds
     include StripeMock::RequestHandlers::Recipients
     include StripeMock::RequestHandlers::Transfers
@@ -50,11 +54,13 @@ 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,
-                :subscriptions_items, :products, :tax_rates, :checkout_sessions
+                :setup_intents, :plans, :prices, :promotion_codes, :recipients, :refunds, :transfers, :payouts,
+                :subscriptions, :country_spec, :subscriptions_items, :products, :tax_rates, :checkout_sessions,
+                :checkout_session_line_items
 
     attr_accessor :error_queue, :debug, :conversion_rate, :account_balance
 
@@ -64,7 +70,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,7 +83,9 @@ module StripeMock
       @orders = {}
       @payment_methods = {}
       @plans = {}
+      @prices = {}
       @products = {}
+      @promotion_codes = {}
       @recipients = {}
       @refunds = {}
       @transfers = {}
@@ -87,6 +95,7 @@ module StripeMock
       @country_spec = {}
       @tax_rates = {}
       @checkout_sessions = {}
+      @checkout_session_line_items = {}
 
       @debug = false
       @error_queue = ErrorQueue.new

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

@@ -4,12 +4,13 @@ module StripeMock
       VALID_START_YEAR = 2009
 
       def Accounts.included(klass)
-        klass.add_handler 'post /v1/accounts',      :new_account
-        klass.add_handler 'get /v1/account',        :get_account
-        klass.add_handler 'get /v1/accounts/(.*)',  :get_account
-        klass.add_handler 'post /v1/accounts/(.*)', :update_account
-        klass.add_handler 'get /v1/accounts',       :list_accounts
-        klass.add_handler 'post /oauth/deauthorize',:deauthorize
+        klass.add_handler 'post /v1/accounts',        :new_account
+        klass.add_handler 'get /v1/account',          :get_account
+        klass.add_handler 'get /v1/accounts/(.*)',    :get_account
+        klass.add_handler 'post /v1/accounts/(.*)',   :update_account
+        klass.add_handler 'get /v1/accounts',         :list_accounts
+        klass.add_handler 'post /oauth/deauthorize',  :deauthorize
+        klass.add_handler 'delete /v1/accounts/(.*)', :delete_account
       end
 
       def new_account(route, method_url, params, headers)
@@ -48,6 +49,16 @@ module StripeMock
         Stripe::StripeObject.construct_from(:stripe_user_id => params[:stripe_user_id])
       end
 
+      def delete_account(route, method_url, params, headers)
+        init_account
+        route =~ method_url
+        assert_existence :account, $1, accounts[$1]
+        accounts[$1] = {
+          id: accounts[$1][:id],
+          deleted: true
+        }
+      end
+
       private
 
       def init_account

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
 
@@ -169,6 +171,11 @@ module StripeMock
            params[:refunds].has_key?(:data) && params[:refunds][:data].nil?)
           allowed << :refunds
         end
+        if params.has_key?(:payment_method_details) && (params[:payment_method_details].empty? ||
+           params[:payment_method_details].has_key?(:card) && (params[:payment_method_details][:card].empty? ||
+           params[:payment_method_details][:card].has_key?(:checks) && params[:payment_method_details][:card][:checks].empty?))
+          allowed << :payment_method_details
+        end
 
         allowed
       end

data/lib/stripe_mock/request_handlers/checkout_session.rb

@@ -0,0 +1,179 @@
+module StripeMock
+  module RequestHandlers
+    module Checkout
+      module Session
+        def Session.included(klass)
+          klass.add_handler 'post /v1/checkout/sessions', :new_session
+          klass.add_handler 'get /v1/checkout/sessions', :list_checkout_sessions
+          klass.add_handler 'get /v1/checkout/sessions/([^/]*)', :get_checkout_session
+          klass.add_handler 'get /v1/checkout/sessions/([^/]*)/line_items', :list_line_items
+        end
+
+        def new_session(route, method_url, params, headers)
+          id = params[:id] || new_id('cs')
+
+          [:cancel_url, :success_url].each do |p|
+            require_param(p) if params[p].nil? || params[p].empty?
+          end
+
+          line_items = nil
+          if params[:line_items]
+            line_items = params[:line_items].each_with_index.map do |line_item, i|
+              throw Stripe::InvalidRequestError("Quantity is required. Add `quantity` to `line_items[#{i}]`") unless line_item[:quantity]
+              unless line_item[:price] || line_item[:price_data] || (line_item[:amount] && line_item[:currency] && line_item[:name])
+                throw Stripe::InvalidRequestError("Price or amount and currency is required. Add `price`, `price_data`, or `amount`, `currency` and `name` to `line_items[#{i}]`")
+              end
+              {
+                id: new_id("li"),
+                price: if line_item[:price]
+                  line_item[:price]
+                elsif line_item[:price_data]
+                  new_price(nil, nil, line_item[:price_data], nil)[:id]
+                else
+                  new_price(nil, nil, {
+                    unit_amount: line_item[:amount],
+                    currency: line_item[:currency],
+                    product_data: {
+                      name: line_item[:name]
+                    }
+                  }, nil)[:id]
+                end,
+                quantity: line_item[:quantity]
+              }
+            end
+          end
+
+          amount = nil
+          currency = nil
+          if line_items
+            amount = 0
+
+            line_items.each do |line_item| 
+              price = prices[line_item[:price]]
+
+              if price.nil?
+                raise StripeMock::StripeMockError.new("Price not found for ID: #{line_item[:price]}")
+              end
+
+              amount += (price[:unit_amount] * line_item[:quantity])
+            end
+
+            currency = prices[line_items.first[:price]][:currency]
+          end
+
+          payment_status = "unpaid"
+          payment_intent = nil
+          setup_intent = nil
+          case params[:mode]
+          when nil, "payment"
+            params[:customer] ||= new_customer(nil, nil, {email: params[:customer_email]}, nil)[:id]
+            require_param(:line_items) if params[:line_items].nil? || params[:line_items].empty?
+            payment_intent = new_payment_intent(nil, nil, {
+              amount: amount,
+              currency: currency,
+              customer: params[:customer],
+              payment_method_options: params[:payment_method_options],
+              payment_method_types: params[:payment_method_types]
+            }.merge(params[:payment_intent_data] || {}), nil)[:id]
+            checkout_session_line_items[id] = line_items
+          when "setup"
+            if !params[:line_items].nil? && !params[:line_items].empty?
+              throw Stripe::InvalidRequestError.new("You cannot pass `line_items` in `setup` mode", :line_items, http_status: 400)
+            end
+            setup_intent = new_setup_intent(nil, nil, {
+              customer: params[:customer],
+              payment_method_options: params[:payment_method_options],
+              payment_method_types: params[:payment_method_types]
+            }.merge(params[:setup_intent_data] || {}), nil)[:id]
+            payment_status = "no_payment_required"
+          when "subscription"
+            params[:customer] ||= new_customer(nil, nil, {email: params[:customer_email]}, nil)[:id]
+            require_param(:line_items) if params[:line_items].nil? || params[:line_items].empty?
+            checkout_session_line_items[id] = line_items
+          else
+            throw Stripe::InvalidRequestError.new("Invalid mode: must be one of payment, setup, or subscription", :mode, http_status: 400)
+          end
+
+          checkout_sessions[id] = {
+            id: id,
+            object: "checkout.session",
+            allow_promotion_codes: nil,
+            amount_subtotal: amount,
+            amount_total: amount,
+            automatic_tax: {
+              enabled: false,
+              status: nil
+            },
+            billing_address_collection: nil,
+            cancel_url: params[:cancel_url],
+            client_reference_id: nil,
+            currency: currency,
+            customer: params[:customer],
+            customer_details: nil,
+            customer_email: params[:customer_email],
+            livemode: false,
+            locale: nil,
+            metadata: params[:metadata],
+            mode: params[:mode],
+            payment_intent: payment_intent,
+            payment_method_options: params[:payment_method_options],
+            payment_method_types: params[:payment_method_types],
+            payment_status: payment_status,
+            setup_intent: setup_intent,
+            shipping: nil,
+            shipping_address_collection: nil,
+            submit_type: nil,
+            subscription: nil,
+            success_url: params[:success_url],
+            total_details: nil,
+            url: URI.join(StripeMock.checkout_base, id).to_s
+          }
+        end
+
+        def list_checkout_sessions(route, method_url, params, headers)
+          Data.mock_list_object(checkout_sessions.values)
+        end
+
+        def get_checkout_session(route, method_url, params, headers)
+          route =~ method_url
+          checkout_session = assert_existence :checkout_session, $1, checkout_sessions[$1]
+
+          checkout_session = checkout_session.clone
+          if params[:expand]&.include?('setup_intent') && checkout_session[:setup_intent]
+            checkout_session[:setup_intent] = setup_intents[checkout_session[:setup_intent]]
+          end
+          checkout_session
+        end
+
+        def list_line_items(route, method_url, params, headers)
+          route =~ method_url
+          checkout_session = assert_existence :checkout_session, $1, checkout_sessions[$1]
+
+          case checkout_session[:mode]
+          when "payment", "subscription"
+            line_items = assert_existence :checkout_session_line_items, $1, checkout_session_line_items[$1]
+            line_items.map do |line_item|
+              price = prices[line_item[:price]].clone
+
+              if price.nil?
+                raise StripeMock::StripeMockError.new("Price not found for ID: #{line_item[:price]}")
+              end
+
+              {
+                id: line_item[:id],
+                object: "item",
+                amount_subtotal: price[:unit_amount] * line_item[:quantity],
+                amount_total: price[:unit_amount] * line_item[:quantity],
+                currency: price[:currency],
+                price: price.clone,
+                quantity: line_item[:quantity]
+              }
+            end
+          else
+            throw Stripe::InvalidRequestError("Only payment and subscription sessions have line items")
+          end
+        end
+      end
+    end
+  end
+end