stripe-ruby-mock 2.3.1 → 2.4.0

data/lib/stripe_mock/instance.rb

@@ -34,14 +34,16 @@ module StripeMock
     include StripeMock::RequestHandlers::InvoiceItems
     include StripeMock::RequestHandlers::Orders
     include StripeMock::RequestHandlers::Plans
+    include StripeMock::RequestHandlers::Refunds
     include StripeMock::RequestHandlers::Recipients
     include StripeMock::RequestHandlers::Transfers
     include StripeMock::RequestHandlers::Tokens
+    include StripeMock::RequestHandlers::CountrySpec
 
 
     attr_reader :accounts, :balance_transactions, :bank_tokens, :charges, :coupons, :customers,
                 :disputes, :events, :invoices, :invoice_items, :orders, :plans, :recipients,
-                :transfers, :subscriptions
+                :refunds, :transfers, :subscriptions, :country_spec
 
     attr_accessor :error_queue, :debug
 
@@ -60,8 +62,10 @@ module StripeMock
       @orders = {}
       @plans = {}
       @recipients = {}
+      @refunds = {}
       @transfers = {}
       @subscriptions = {}
+      @country_spec = {}
 
       @debug = false
       @error_queue = ErrorQueue.new
@@ -124,9 +128,16 @@ module StripeMock
       "#{StripeMock.global_id_prefix}#{prefix}_#{@id_counter += 1}"
     end
 
-    def new_balance_transaction(prefix)
+    def new_balance_transaction(prefix, params = {})
       # balance transaction ids must be strings
-      "#{StripeMock.global_id_prefix}#{prefix}_#{@balance_transaction_counter += 1}"
+      id = "#{StripeMock.global_id_prefix}#{prefix}_#{@balance_transaction_counter += 1}"
+      amount = params[:amount]
+      unless amount.nil?
+        # Fee calculation
+        params[:fee] ||= (30 + (amount.abs * 0.029).ceil) * (amount > 0 ? 1 : -1)
+      end
+      @balance_transactions[id] = Data.mock_balance_transaction(params.merge(id: id))
+      id
     end
 
     def symbolize_names(hash)

data/lib/stripe_mock/request_handlers/accounts.rb

@@ -8,6 +8,7 @@ module StripeMock
         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
       end
 
       def new_account(route, method_url, params, headers)
@@ -34,6 +35,12 @@ module StripeMock
         Data.mock_list_object(accounts.values, params)
       end
 
+      def deauthorize(route, method_url, params, headers)
+        init_account
+        route =~ method_url
+        Stripe::StripeObject.construct_from(:stripe_user_id => params[:stripe_user_id])
+      end
+
       private
 
       def init_account

data/lib/stripe_mock/request_handlers/charges.rb

@@ -8,7 +8,7 @@ module StripeMock
         klass.add_handler 'get /v1/charges/(.*)',           :get_charge
         klass.add_handler 'post /v1/charges/(.*)/capture',  :capture_charge
         klass.add_handler 'post /v1/charges/(.*)/refund',   :refund_charge
-        klass.add_handler 'post /v1/charges/(.*)/refunds',  :create_refund
+        klass.add_handler 'post /v1/charges/(.*)/refunds',  :refund_charge
         klass.add_handler 'post /v1/charges/(.*)',          :update_charge
       end
 
@@ -28,7 +28,7 @@ module StripeMock
             if params[:customer]
               params[:source] = get_card(customers[params[:customer]], params[:source])
             else
-              params[:source] = get_card_by_token(params[:source])
+              params[:source] = get_card_or_bank_by_token(params[:source])
             end
           elsif params[:source][:id]
             raise Stripe::InvalidRequestError.new("Invalid token id: #{params[:source]}", 'card', 400)
@@ -41,8 +41,20 @@ module StripeMock
         end
 
         ensure_required_params(params)
+        bal_trans_params = { amount: params[:amount], source: params[:source] }
 
-        charges[id] = Data.mock_charge(params.merge :id => id, :balance_transaction => new_balance_transaction('txn'))
+        balance_transaction_id = new_balance_transaction('txn', bal_trans_params)
+
+        charges[id] = Data.mock_charge(
+            params.merge :id => id,
+            :balance_transaction => balance_transaction_id)
+
+        if params[:expand] == ['balance_transaction']
+          charges[id][:balance_transaction] =
+            balance_transactions[balance_transaction_id]
+        end
+
+        charges[id]
       end
 
       def update_charge(route, method_url, params, headers)
@@ -74,7 +86,8 @@ module StripeMock
 
       def get_charge(route, method_url, params, headers)
         route =~ method_url
-        assert_existence :charge, $1, charges[$1]
+        charge_id = $1 || params[:charge]
+        assert_existence :charge, charge_id, charges[charge_id]
       end
 
       def capture_charge(route, method_url, params, headers)
@@ -101,24 +114,12 @@ module StripeMock
       def refund_charge(route, method_url, params, headers)
         charge = get_charge(route, method_url, params, headers)
 
-        refund = Data.mock_refund params.merge(
-          :balance_transaction => new_balance_transaction('txn'),
-          :id => new_id('re')
-        )
-        add_refund_to_charge(refund, charge)
-        charge
-      end
-
-      def create_refund(route, method_url, params, headers)
-        charge = get_charge(route, method_url, params, headers)
-
-        refund = Data.mock_refund params.merge(
-          :balance_transaction => new_balance_transaction('txn'),
-          :id => new_id('re'),
-          :charge => charge[:id]
+        new_refund(
+          route,
+          method_url,
+          params.merge(:charge => charge[:id]),
+          headers
         )
-        add_refund_to_charge(refund, charge)
-        refund
       end
 
       private
@@ -132,6 +133,8 @@ module StripeMock
           raise Stripe::InvalidRequestError.new("Invalid integer: #{params[:amount]}", 'amount', 400)
         elsif non_positive_charge_amount?(params)
           raise Stripe::InvalidRequestError.new('Invalid positive integer', 'amount', 400)
+        elsif params[:source].nil? && params[:customer].nil?
+          raise Stripe::InvalidRequestError.new('Must provide source or customer.', nil)
         end
       end
 

data/lib/stripe_mock/request_handlers/country_spec.rb

@@ -0,0 +1,22 @@
+module StripeMock
+  module RequestHandlers
+    module CountrySpec
+
+      def CountrySpec.included(klass)
+        klass.add_handler 'get /v1/country_specs/(.*)', :retrieve_country_spec
+      end
+
+      def retrieve_country_spec(route, method_url, params, headers)
+        route =~ method_url
+
+        unless ["AT", "AU", "BE", "CA", "DE", "DK", "ES", "FI", "FR", "GB", "IE", "IT", "JP", "LU", "NL", "NO", "SE", "SG", "US"].include?($1)
+          raise Stripe::InvalidRequestError.new("#{$1} is not currently supported by Stripe.", $1.to_s)
+        end
+
+        country_spec[$1] ||= Data.mock_country_spec($1)
+
+        assert_existence :country_spec, $1, country_spec[$1]
+      end
+    end
+  end
+end

data/lib/stripe_mock/request_handlers/customers.rb

@@ -4,9 +4,9 @@ module StripeMock
 
       def Customers.included(klass)
         klass.add_handler 'post /v1/customers',                     :new_customer
-        klass.add_handler 'post /v1/customers/(.*)',                :update_customer
-        klass.add_handler 'get /v1/customers/(.*)',                 :get_customer
-        klass.add_handler 'delete /v1/customers/(.*)',              :delete_customer
+        klass.add_handler 'post /v1/customers/([^/]*)',             :update_customer
+        klass.add_handler 'get /v1/customers/([^/]*)',              :get_customer
+        klass.add_handler 'delete /v1/customers/([^/]*)',           :delete_customer
         klass.add_handler 'get /v1/customers',                      :list_customers
       end
 

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

@@ -26,8 +26,11 @@ module StripeMock
       end
 
       def add_subscription_to_customer(cus, sub)
-        id = new_id('ch')
-        charges[id] = Data.mock_charge(:id => id, :customer => cus[:id], :amount => sub[:plan][:amount])
+        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])
+        end
+
         if cus[:currency].nil?
           cus[:currency] = sub[:plan][:currency]
         elsif cus[:currency] != sub[:plan][:currency]

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

@@ -35,7 +35,8 @@ module StripeMock
       end
 
       def get_card_or_bank_by_token(token)
-        @card_tokens[token] || @bank_tokens[token] || raise(Stripe::InvalidRequestError.new("Invalid token id: #{token}", 'tok', 404))
+        token_id = token['id'] || token
+         @card_tokens[token_id] || @bank_tokens[token_id] || raise(Stripe::InvalidRequestError.new("Invalid token id: #{token_id}", 'tok', 404))
       end
 
     end

data/lib/stripe_mock/request_handlers/recipients.rb

@@ -12,6 +12,18 @@ module StripeMock
         params[:id] ||= new_id('rp')
         cards = []
 
+        if params[:name].nil?
+          raise StripeMock::StripeMockError.new("Missing required parameter name for recipients.")
+        end
+
+        if params[:type].nil?
+          raise StripeMock::StripeMockError.new("Missing required parameter type for recipients.")
+        end
+
+        unless %w(individual corporation).include?(params[:type])
+          raise StripeMock::StripeMockError.new("Type must be either individual or corporation..")
+        end
+
         if params[:bank_account]
           params[:active_account] = get_bank_by_token(params.delete(:bank_account))
         end

data/lib/stripe_mock/request_handlers/refunds.rb

@@ -0,0 +1,88 @@
+module StripeMock
+  module RequestHandlers
+    module Refunds
+
+      def Refunds.included(klass)
+        klass.add_handler 'post /v1/refunds',               :new_refund
+        klass.add_handler 'get /v1/refunds',                :get_refunds
+        klass.add_handler 'get /v1/refunds/(.*)',           :get_refund
+        klass.add_handler 'post /v1/refunds/(.*)',          :update_refund
+      end
+
+      def new_refund(route, method_url, params, headers)
+        if params[:idempotency_key] && refunds.any?
+          original_refund = refunds.values.find { |c| c[:idempotency_key] == params[:idempotency_key]}
+          return refunds[original_refund[:id]] if original_refund
+        end
+
+        charge = get_charge(route, method_url, params, headers)
+        params[:amount] ||= charge[:amount]
+        id = new_id('re')
+        bal_trans_params = {
+          amount: params[:amount] * -1,
+          source: id,
+          type: 'refund'
+        }
+        balance_transaction_id = new_balance_transaction('txn', bal_trans_params)
+        refund = Data.mock_refund params.merge(
+          :balance_transaction => balance_transaction_id,
+          :id => id,
+          :charge => charge[:id],
+        )
+        add_refund_to_charge(refund, charge)
+        refunds[id] = refund
+
+        if params[:expand] == ['balance_transaction']
+          refunds[id][:balance_transaction] =
+            balance_transactions[balance_transaction_id]
+        end
+        refund
+      end
+
+      def update_refund(route, method_url, params, headers)
+        route =~ method_url
+        id = $1
+
+        refund = assert_existence :refund, id, refunds[id]
+        allowed = allowed_refund_params(params)
+        disallowed = params.keys - allowed
+        if disallowed.count > 0
+          raise Stripe::InvalidRequestError.new("Received unknown parameters: #{disallowed.join(', ')}" , '', 400)
+        end
+
+        refunds[id] = Util.rmerge(refund, params)
+      end
+
+      def get_refunds(route, method_url, params, headers)
+        params[:offset] ||= 0
+        params[:limit] ||= 10
+
+        clone = refunds.clone
+
+        Data.mock_list_object(clone.values, params)
+      end
+
+      def get_refund(route, method_url, params, headers)
+        route =~ method_url
+        refund_id = $1 || params[:refund]
+        assert_existence :refund, refund_id, refunds[refund_id]
+      end
+
+      private
+
+      def ensure_refund_required_params(params)
+        if non_integer_charge_amount?(params)
+          raise Stripe::InvalidRequestError.new("Invalid integer: #{params[:amount]}", 'amount', 400)
+        elsif non_positive_charge_amount?(params)
+          raise Stripe::InvalidRequestError.new('Invalid positive integer', 'amount', 400)
+        elsif params[:charge].nil?
+          raise Stripe::InvalidRequestError.new('Must provide the identifier of the charge to refund.', nil)
+        end
+      end
+
+      def allowed_refund_params(params)
+        [:metadata]
+      end
+    end
+  end
+end

data/lib/stripe_mock/request_handlers/subscriptions.rb

@@ -9,11 +9,11 @@ module StripeMock
         klass.add_handler 'post /v1/subscriptions/(.*)', :update_subscription
         klass.add_handler 'delete /v1/subscriptions/(.*)', :cancel_subscription
 
-        klass.add_handler 'post /v1/customers/(.*)/subscriptions', :create_customer_subscription
-        klass.add_handler 'get /v1/customers/(.*)/subscriptions/(.*)', :retrieve_customer_subscription
-        klass.add_handler 'get /v1/customers/(.*)/subscriptions', :retrieve_customer_subscriptions
-        klass.add_handler 'post /v1/customers/(.*)subscriptions/(.*)', :update_subscription
-        klass.add_handler 'delete /v1/customers/(.*)/subscriptions/(.*)', :cancel_subscription
+        klass.add_handler 'post /v1/customers/(.*)/subscription(?:s)?', :create_customer_subscription
+        klass.add_handler 'get /v1/customers/(.*)/subscription(?:s)?/(.*)', :retrieve_customer_subscription
+        klass.add_handler 'get /v1/customers/(.*)/subscription(?:s)?', :retrieve_customer_subscriptions
+        klass.add_handler 'post /v1/customers/(.*)subscription(?:s)?/(.*)', :update_subscription
+        klass.add_handler 'delete /v1/customers/(.*)/subscription(?:s)?/(.*)', :cancel_subscription
       end
 
       def retrieve_customer_subscription(route, method_url, params, headers)
@@ -158,9 +158,10 @@ module StripeMock
           # coupon = assert_existence :coupon, coupon_id, coupons[coupon_id]
 
           coupon = coupons[coupon_id]
-
           if coupon
             subscription[:discount] = Stripe::Util.convert_to_stripe_object({ coupon: coupon }, {})
+          elsif coupon_id == ""
+            subscription[:discount] = Stripe::Util.convert_to_stripe_object(nil, {})
           else
             raise Stripe::InvalidRequestError.new("No such coupon: #{coupon_id}", 'coupon', 400)
           end

data/lib/stripe_mock/request_handlers/validators/param_validators.rb

@@ -8,9 +8,14 @@ module StripeMock
         @base_strategy.create_plan_params.keys.each do |name|
           raise Stripe::InvalidRequestError.new("Missing required param: #{name}.", name) if params[name].nil?
         end
+
         if plans[ params[:id] ]
           raise Stripe::InvalidRequestError.new("Plan already exists.", :id)
         end
+
+        unless params[:amount].integer?
+          raise Stripe::InvalidRequestError.new("Invalid integer: #{params[:amount]}", :amount)
+        end
       end
 
     end

data/lib/stripe_mock/test_strategies/base.rb

@@ -3,17 +3,18 @@ module StripeMock
     class Base
 
       def create_plan_params(params={})
+        currency = params[:currency] || 'usd'
         {
           :id => 'stripe_mock_default_plan_id',
           :name => 'StripeMock Default Plan ID',
           :amount => 1337,
-          :currency => 'usd',
+          :currency => currency,
           :interval => 'month'
         }.merge(params)
       end
 
       def generate_card_token(card_params={})
-        card_data = { :number => "4242424242424242", :exp_month => 9, :exp_year => 2018, :cvc => "999" }
+        card_data = { :number => "4242424242424242", :exp_month => 9, :exp_year => 2018, :cvc => "999", :tokenization_method => nil }
         card = StripeMock::Util.card_merge(card_data, card_params)
         card[:fingerprint] = StripeMock::Util.fingerprint(card[:number]) if StripeMock.state == 'local'
 
@@ -22,9 +23,10 @@ module StripeMock
       end
 
       def generate_bank_token(bank_account_params={})
+        currency = bank_account_params[:currency] || 'usd'
         bank_account = {
           :country => "US",
-          :currency => "usd",
+          :currency => currency,
           :account_holder_name => "Jane Austen",
           :account_holder_type => "individual",
           :routing_number => "110000000",
@@ -37,10 +39,11 @@ module StripeMock
       end
 
       def create_coupon_params(params = {})
+        currency = params[:currency] || 'usd'
         {
           id: '10BUCKS',
           amount_off: 1000,
-          currency: 'usd',
+          currency: currency,
           max_redemptions: 100,
           metadata: {
             created_by: 'admin_acct_1'

data/lib/stripe_mock/version.rb

@@ -1,4 +1,4 @@
 module StripeMock
   # stripe-ruby-mock version
-  VERSION = "2.3.1"
+  VERSION = "2.4.0"
 end

data/lib/stripe_mock/webhook_fixtures/account.external_account.created.json

@@ -0,0 +1,27 @@
+{
+   "created":1326853478,
+   "livemode":false,
+   "id":"evt_00000000000000",
+   "type":"account.external_account.created",
+   "object":"event",
+   "data":{
+      "object":{
+         "id":"ba_00000000000000",
+         "object":"bank_account",
+         "account":"acct_00000000000000",
+         "account_holder_name":"Jane Austen",
+         "account_holder_type":"individual",
+         "bank_name":"STRIPE TEST BANK",
+         "country":"US",
+         "currency":"eur",
+         "default_for_currency":false,
+         "fingerprint":"efGCBmiwp56O1lsN",
+         "last4":"6789",
+         "metadata":{
+
+         },
+         "routing_number":"110000000",
+         "status":"new"
+      }
+   }
+}