stripe-ruby-mock 2.5.0 → 2.5.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: ce6d1ae38721fc9f43fa69e3d82217593b0789d9
-  data.tar.gz: abbbfb0dd70438e23ac276215602a3d3a77705b1
+  metadata.gz: 27d9481d4fc8140fc2d42f733fc529fbb7a121cf
+  data.tar.gz: a8eeabea5e8b7d9181f597b9968e6e98b8fba4ed
 SHA512:
-  metadata.gz: 49d92a3efee46c78cd157fd84de5e8eb456b6686b06c00a23262648395b0dd2cb647620304de703e52e5919102ce6c782a4947b50c39aef90428a02af9cfe9b0
-  data.tar.gz: c87d09dd495bdca773e0ea39169f04900e7467f315f220ca556d907e9c0f294ddad399a73c2765e27e9ebe319b42f1b68b9ef825a02707c4877096777154a8c4
+  metadata.gz: b089c35e1b06284105b79baa151a5c61ebf54ec1e405406e3d0f827964c9e30f6e33b02a92dfc11ee83ec4a0e71a8a7fa35448c9a859476481bbbff2f60d15a1
+  data.tar.gz: e913cc264d89d765911c9de0066cc57a819f5235b3a2fb4d6266f7dd5561669e3ebe45e12faa5be8e12a74cc4a67ed51ca6adc2064bdcc02b1394dc5cf298899

data/README.md

@@ -12,7 +12,7 @@ This gem has unexpectedly grown in popularity and I've gotten pretty busy, so I'
 
 In your gemfile:
 
-    gem 'stripe-ruby-mock', '~> 2.5.0', :require => 'stripe_mock'
+    gem 'stripe-ruby-mock', '~> 2.5.1', :require => 'stripe_mock'
 
 ## Features
 

data/lib/stripe_mock.rb

@@ -26,6 +26,7 @@ require 'stripe_mock/api/client'
 require 'stripe_mock/api/server'
 
 require 'stripe_mock/api/bank_tokens'
+require 'stripe_mock/api/account_balance'
 require 'stripe_mock/api/conversion_rate'
 require 'stripe_mock/api/card_tokens'
 require 'stripe_mock/api/debug'
@@ -45,6 +46,7 @@ require 'stripe_mock/request_handlers/helpers/token_helpers.rb'
 require 'stripe_mock/request_handlers/validators/param_validators.rb'
 
 require 'stripe_mock/request_handlers/accounts.rb'
+require 'stripe_mock/request_handlers/balance.rb'
 require 'stripe_mock/request_handlers/balance_transactions.rb'
 require 'stripe_mock/request_handlers/charges.rb'
 require 'stripe_mock/request_handlers/cards.rb'
@@ -60,6 +62,7 @@ require 'stripe_mock/request_handlers/plans.rb'
 require 'stripe_mock/request_handlers/recipients.rb'
 require 'stripe_mock/request_handlers/refunds.rb'
 require 'stripe_mock/request_handlers/transfers.rb'
+require 'stripe_mock/request_handlers/payouts.rb'
 require 'stripe_mock/request_handlers/subscriptions.rb'
 require 'stripe_mock/request_handlers/tokens.rb'
 require 'stripe_mock/request_handlers/country_spec.rb'

data/lib/stripe_mock/api/account_balance.rb

@@ -0,0 +1,14 @@
+module StripeMock
+
+  def self.set_account_balance(value)
+    case @state
+      when 'local'
+      instance.account_balance = value
+    when 'remote'
+      client.set_account_balance(value)
+    else
+      raise UnstartedStateError
+    end
+  end
+
+end

data/lib/stripe_mock/api/webhooks.rb

@@ -18,6 +18,7 @@ module StripeMock
     json[:account] = params.delete(:account) if params.key?(:account)
     json[:data][:object] = Util.rmerge(json[:data][:object], params)
     json.delete(:id)
+    json[:created] = params[:created] || Time.now.to_i
 
     if @state == 'local'
       event_data = instance.generate_webhook_event(json)
@@ -43,6 +44,7 @@ module StripeMock
         'account.external_account.deleted',
         'balance.available',
         'charge.succeeded',
+        'charge.updated',
         'charge.failed',
         'charge.refunded',
         'charge.dispute.created',

data/lib/stripe_mock/client.rb

@@ -73,6 +73,10 @@ module StripeMock
       timeout_wrap { @pipe.set_conversion_rate(value) }
     end
 
+    def set_account_balance(value)
+      timeout_wrap { @pipe.set_account_balance(value) }
+    end
+
     def destroy_resource(type, id)
       timeout_wrap { @pipe.destroy_resource(type, id) }
     end

data/lib/stripe_mock/data.rb

@@ -12,7 +12,7 @@ module StripeMock
         timezone: "US/Pacific",
         details_submitted: false,
         charges_enabled: false,
-        transfers_enabled: false,
+        payouts_enabled: false,
         currencies_supported: [
           "usd"
         ],
@@ -103,7 +103,7 @@ module StripeMock
 
     def self.mock_customer(sources, params)
       cus_id = params[:id] || "test_cus_default"
-      currency = params[:currency] || nil
+      currency = params[:currency] || 'usd'
       sources.each {|source| source[:customer] = cus_id}
       {
         email: 'stripe_mock@example.com',
@@ -207,7 +207,8 @@ module StripeMock
         metadata: {},
         charge: "ch_4fWhYjzQ23UFWT",
         receipt_number: nil,
-        status: "succeeded"
+        status: "succeeded",
+        reason: "requested_by_customer"
       }.merge(params)
     end
 
@@ -284,6 +285,7 @@ module StripeMock
     def self.mock_subscription(params={})
       StripeMock::Util.rmerge({
         created: 1478204116,
+        billing: 'charge_automatically',
         current_period_start: 1308595038,
         current_period_end: 1308681468,
         status: 'trialing',
@@ -620,6 +622,21 @@ module StripeMock
       }.merge(params)
     end
 
+    def self.mock_payout(params={})
+      currency = params[:currency] || 'usd'
+      id = params[:id] || 'po_test_payout'
+      {
+        :amount => 100,
+        :id => id,
+        :livemode => false,
+        :metadata => {},
+        :currency => currency,
+        :object => "payout",
+        :date => 1304114826,
+        :description => "Payout description",
+      }.merge(params)
+    end
+
     def self.mock_disputes(ids=[])
       disputes = {}
       ids.each do |id|
@@ -909,6 +926,38 @@ module StripeMock
       }
     end
 
+    def self.mock_balance(usd_balance = 10000)
+      {
+        object: "balance",
+        available: [
+          {
+            currency: "usd",
+            amount: usd_balance,
+            source_types: {
+              card: 25907032203,
+              bank_account: 108476658,
+              bitcoin_receiver: 1545182
+            }
+          }],
+        connect_reserved: [
+          {
+            currency: "usd",
+            amount: 4700
+          }],
+        livemode: false,
+        pending: [
+          {
+            currency: "usd",
+            amount: 22738833554,
+            source_types: {
+              card: 22738826610,
+              bank_account: 0,
+              bitcoin_receiver: 6944
+            }
+          }]
+      }
+    end
+
     def self.mock_balance_transactions(ids=[])
       bts = {}
       ids.each do |id|
@@ -952,5 +1001,31 @@ module StripeMock
         type: "charge"
       }.merge(params)
     end
+
+    def self.mock_subscription_item(params = {})
+      iid = params[:id] || 'test_txn_default'
+      {
+        id: iid,
+        object: 'subscription_item',
+        created: 1504716183,
+        metadata: {
+      },
+        plan: {
+          id: 'PER_USER_PLAN1',
+          object: 'plan',
+          amount: 1337,
+          created: 1504716177,
+          currency: 'usd',
+          interval: 'month',
+          interval_count: 1,
+          livemode: false,
+          metadata: {},
+          name: 'StripeMock Default Plan ID',
+          statement_descriptor: nil,
+          trial_period_days: nil
+        },
+        quantity: 2
+      }.merge(params)
+    end
   end
 end

data/lib/stripe_mock/instance.rb

@@ -21,6 +21,7 @@ module StripeMock
     end
 
     include StripeMock::RequestHandlers::Accounts
+    include StripeMock::RequestHandlers::Balance
     include StripeMock::RequestHandlers::BalanceTransactions
     include StripeMock::RequestHandlers::Charges
     include StripeMock::RequestHandlers::Cards
@@ -39,16 +40,17 @@ module StripeMock
     include StripeMock::RequestHandlers::Transfers
     include StripeMock::RequestHandlers::Tokens
     include StripeMock::RequestHandlers::CountrySpec
+    include StripeMock::RequestHandlers::Payouts
 
-
-    attr_reader :accounts, :balance_transactions, :bank_tokens, :charges, :coupons, :customers,
+    attr_reader :accounts, :balance, :balance_transactions, :bank_tokens, :charges, :coupons, :customers,
                 :disputes, :events, :invoices, :invoice_items, :orders, :plans, :recipients,
-                :refunds, :transfers, :subscriptions, :country_spec
+                :refunds, :transfers, :payouts, :subscriptions, :country_spec, :subscriptions_items
 
-    attr_accessor :error_queue, :debug, :conversion_rate
+    attr_accessor :error_queue, :debug, :conversion_rate, :account_balance
 
     def initialize
       @accounts = {}
+      @balance = Data.mock_balance
       @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 = {}
@@ -64,14 +66,18 @@ module StripeMock
       @recipients = {}
       @refunds = {}
       @transfers = {}
+      @payouts = {}
       @subscriptions = {}
+      @subscriptions_items = []
       @country_spec = {}
 
       @debug = false
       @error_queue = ErrorQueue.new
       @id_counter = 0
       @balance_transaction_counter = 0
+      @dispute_counter = 0
       @conversion_rate = 1.0
+      @account_balance = 10000
 
       # This is basically a cache for ParamValidators
       @base_strategy = TestStrategies::Base.new
@@ -125,6 +131,8 @@ module StripeMock
         case object
           when :balance_transaction
             id = new_balance_transaction('txn', attributes)
+          when :dispute
+            id = new_dispute('dp', attributes)
           else
             raise UnsupportedRequestError.new "Unsupported stripe object `#{object}`"
         end
@@ -134,6 +142,9 @@ module StripeMock
           when :balance_transaction
             btxn = assert_existence :balance_transaction, id, @balance_transactions[id]
             btxn.merge!(attributes)
+          when :dispute
+            dispute = assert_existence :dispute, id, @disputes[id]
+            dispute.merge!(attributes)
           else
             raise UnsupportedRequestError.new "Unsupported stripe object `#{object}`"
         end
@@ -169,6 +180,12 @@ module StripeMock
       id
     end
 
+    def new_dispute(prefix, params = {})
+      id = "#{StripeMock.global_id_prefix}#{prefix}_#{@dispute_counter += 1}"
+      @disputes[id] = Data.mock_dispute(params.merge(id: id))
+      id
+    end
+
     def symbolize_names(hash)
       Stripe::Util.symbolize_names(hash)
     end

data/lib/stripe_mock/request_handlers/accounts.rb

@@ -26,8 +26,12 @@ module StripeMock
 
       def update_account(route, method_url, params, headers)
         route =~ method_url
-        assert_existence :account, $1, accounts[$1]
-        accounts[$1].merge!(params)
+        account = assert_existence :account, $1, accounts[$1]
+        account.merge!(params)
+        if blank_value?(params[:tos_acceptance], :date)
+          raise Stripe::InvalidRequestError.new("Invalid integer: ", "tos_acceptance[date]", http_status: 400)
+        end
+        account
       end
 
       def list_accounts(route, method_url, params, headers)
@@ -49,6 +53,18 @@ module StripeMock
           accounts[acc[:id]] = acc
         end
       end
+
+      # Checks if setting a blank value
+      #
+      # returns true if the key is included in the hash
+      # and its value is empty or nil
+      def blank_value?(hash, key)
+        if hash.key?(key)
+          value = hash[key]
+          return true if value.nil? || "" == value
+        end
+        false
+      end
     end
   end
 end

data/lib/stripe_mock/request_handlers/balance.rb

@@ -0,0 +1,17 @@
+module StripeMock
+  module RequestHandlers
+    module Balance
+
+      def Balance.included(klass)
+        klass.add_handler 'get /v1/balance',                        :get_balance
+      end
+
+      def get_balance(route, method_url, params, headers)
+        route =~ method_url
+
+        return_balance = Data.mock_balance(account_balance)
+        return_balance
+      end
+    end
+  end
+end

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

@@ -9,12 +9,14 @@ module StripeMock
       def custom_subscription_params(plan, cus, options = {})
         verify_trial_end(options[:trial_end]) if options[:trial_end]
 
-        start_time = options[:current_period_start] || Time.now.utc.to_i
-        params = { plan: plan, customer: cus[:id], current_period_start: start_time }
+        now = Time.now.utc.to_i
+        created_time = options[:created] || now
+        start_time = options[:current_period_start] || now
+        params = { plan: plan, customer: cus[:id], current_period_start: start_time, created: created_time }
         params.merge! options.select {|k,v| k =~ /application_fee_percent|quantity|metadata|tax_percent/}
         # TODO: Implement coupon logic
 
-        if ((plan[:trial_period_days]||0) == 0 && options[:trial_end].nil?) || options[:trial_end] == "now"
+        if (((plan && plan[:trial_period_days]) || 0) == 0 && options[:trial_end].nil?) || options[:trial_end] == "now"
           end_time = get_ending_time(start_time, plan)
           params.merge!({status: 'active', current_period_end: end_time, trial_start: nil, trial_end: nil})
         else
@@ -28,12 +30,16 @@ module StripeMock
       def add_subscription_to_customer(cus, sub)
         if sub[:trial_end].nil? || sub[:trial_end] == "now"
           id = new_id('ch')
-          charges[id] = Data.mock_charge(:id => id, :customer => cus[:id], :amount => sub[:plan][:amount])
+          charges[id] = Data.mock_charge(
+            :id => id,
+            :customer => cus[:id],
+            :amount => (sub[:plan] ? sub[:plan][:amount] : total_items_amount(sub[:items][:data]))
+          )
         end
 
         if cus[:currency].nil?
-          cus[:currency] = sub[:plan][:currency]
-        elsif cus[:currency] != sub[:plan][:currency]
+          cus[:currency] = sub[:items][:data][0][:plan][:currency]
+        elsif cus[:currency] != sub[:items][:data][0][:plan][:currency]
           raise Stripe::InvalidRequestError.new( "Can't combine currencies on a single customer. This customer has had a subscription, coupon, or invoice item with currency #{cus[:currency]}", 'currency', http_status: 400)
         end
         cus[:subscriptions][:total_count] = (cus[:subscriptions][:total_count] || 0) + 1
@@ -50,6 +56,8 @@ module StripeMock
       # `intervals` is set to 1 when calculating current_period_end from current_period_start & plan
       # `intervals` is set to 2 when calculating Stripe::Invoice.upcoming end from current_period_start & plan
       def get_ending_time(start_time, plan, intervals = 1)
+        return start_time unless plan
+
         case plan[:interval]
         when "week"
           start_time + (604800 * (plan[:interval_count] || 1) * intervals)
@@ -74,6 +82,19 @@ module StripeMock
         end
       end
 
+      def total_items_amount(items)
+        total = 0
+        items.each { |i| total += (i[:quantity] || 1) * i[:plan][:amount] }
+        total
+      end
+
+      def mock_subscription_items(items = [])
+        data = []
+        items.each do |i|
+          data << Data.mock_subscription_item(i.merge(plan: plans[i[:plan].to_s]))
+        end
+        data
+      end
     end
   end
 end

data/lib/stripe_mock/request_handlers/invoices.rb

@@ -73,7 +73,7 @@ module StripeMock
           else
             customer[:subscriptions][:data].min_by { |sub| sub[:current_period_end] }
           end
-        
+
         if params[:subscription_proration_date] && !((subscription[:current_period_start]..subscription[:current_period_end]) === params[:subscription_proration_date])
           raise Stripe::InvalidRequestError.new('Cannot specify proration date outside of current subscription period', nil, http_status: 400)
         end
@@ -112,14 +112,30 @@ module StripeMock
                                    quantity: subscription[:quantity],
                                    proration: true
           )
+
+          preview_plan = assert_existence :plan, params[:subscription_plan], plans[params[:subscription_plan]]
+          if preview_plan[:interval] == subscription[:plan][:interval] && preview_plan[:interval_count] == subscription[:plan][:interval_count] && params[:subscription_trial_end].nil?
+            remaining_amount = preview_plan[:amount] * subscription_quantity * (subscription[:current_period_end] - subscription_proration_date.to_i) / (subscription[:current_period_end] - subscription[:current_period_start])
+            invoice_lines << Data.mock_line_item(
+                                     id: new_id('ii'),
+                                     amount: remaining_amount,
+                                     description: 'Remaining time',
+                                     plan: preview_plan,
+                                     period: {
+                                         start: subscription_proration_date.to_i,
+                                         end: subscription[:current_period_end]
+                                     },
+                                     quantity: subscription_quantity,
+                                     proration: true
+            )
+          end
         end
 
         subscription_line = get_mock_subscription_line_item(preview_subscription)
         invoice_lines << subscription_line
 
-        id = new_id('in')
-        invoices[id] = Data.mock_invoice(invoice_lines,
-          id: id,
+        Data.mock_invoice(invoice_lines,
+          id: new_id('in'),
           customer: customer[:id],
           discount: customer[:discount],
           date: invoice_date,