stripe-ruby-mock 1.10.1.7 → 2.0.0

data/lib/stripe_mock/request_handlers/customers.rb

@@ -23,8 +23,7 @@ module StripeMock
 
         if params[:plan]
           plan_id = params[:plan].to_s
-          plan = plans[plan_id]
-          assert_existance :plan, plan_id, plan
+          plan = assert_existance :plan, plan_id, plans[plan_id]
 
           if params[:default_card].nil? && plan[:trial_period_days].nil? && plan[:amount] != 0
             raise Stripe::InvalidRequestError.new('You must supply a valid card', nil, 400)
@@ -42,14 +41,12 @@ module StripeMock
 
       def update_customer(route, method_url, params, headers)
         route =~ method_url
-        assert_existance :customer, $1, customers[$1]
-
-        cus = customers[$1] ||= Data.mock_customer([], :id => $1)
+        cus = assert_existance :customer, $1, customers[$1]
         cus.merge!(params)
 
         if params[:card]
           new_card = get_card_by_token(params.delete(:card))
-          add_card_to_customer(new_card, cus)
+          add_card_to_object(:customer, new_card, cus, true)
           cus[:default_card] = new_card[:id]
         end
 
@@ -64,14 +61,11 @@ module StripeMock
           id: customers[$1][:id],
           deleted: true
         }
-
-        customers[$1]
       end
 
       def get_customer(route, method_url, params, headers)
         route =~ method_url
         assert_existance :customer, $1, customers[$1]
-        customers[$1] ||= Data.mock_customer([], :id => $1)
       end
 
       def list_customers(route, method_url, params, headers)

data/lib/stripe_mock/request_handlers/events.rb

@@ -8,9 +8,7 @@ module StripeMock
 
       def retrieve_event(route, method_url, params, headers)
         route =~ method_url
-        event = events[$1]
-        assert_existance :event, $1, event
-        event
+        assert_existance :event, $1, events[$1]
       end
 
     end

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

@@ -2,24 +2,31 @@ module StripeMock
   module RequestHandlers
     module Helpers
 
-      def get_customer_card(customer, token)
-        customer[:cards][:data].find{|cc| cc[:id] == token }
+      def get_card(object, card_id, class_name='Customer')
+        card = object[:cards][:data].find{|cc| cc[:id] == card_id }
+        if card.nil?
+          msg = "#{class_name} #{object[:id]} does not have card #{card_id}"
+          raise Stripe::InvalidRequestError.new(msg, 'card', 404)
+        end
+        card
       end
 
-      def add_card_to_customer(card, cus)
-        card[:customer] = cus[:id]
+      def add_card_to_object(type, card, object, replace_current=false)
+        card[type] = object[:id]
 
-        if cus[:cards][:count] == 0
-          cus[:cards][:count] += 1
+        if replace_current
+          object[:cards][:data].delete_if {|card| card[:id] == object[:default_card]}
+          object[:default_card] = card[:id]
         else
-          cus[:cards][:data].delete_if {|card| card[:id] == cus[:default_card]}
+          object[:cards][:count] += 1
         end
 
-        cus[:cards][:data] << card
+        object[:default_card] = card[:id] unless object[:default_card]
+        object[:cards][:data] << card
 
         card
       end
 
     end
   end
-end
+end

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

@@ -0,0 +1,16 @@
+module StripeMock
+  module RequestHandlers
+    module Helpers
+
+      def add_refund_to_charge(refund, charge)
+        refunds = charge[:refunds]
+        refunds[:data] << refund
+        refunds[:total_count] = refunds[:data].count
+
+        charge[:amount_refunded] = refunds[:data].reduce(0) {|sum, r| sum + r[:amount].to_i }
+        charge[:refunded] = true
+      end
+
+    end
+  end
+end

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

@@ -11,7 +11,7 @@ module StripeMock
 
         start_time = options[:current_period_start] || Time.now.utc.to_i
         params = { plan: plan, customer: cus[:id], current_period_start: start_time }
-        params.merge! options.select {|k,v| k =~ /application_fee_percent|quantity/}
+        params.merge! options.select {|k,v| k =~ /application_fee_percent|quantity|metadata/}
         # TODO: Implement coupon logic
 
         if (plan[:trial_period_days].nil? && options[:trial_end].nil?) || options[:trial_end] == "now"
@@ -27,7 +27,14 @@ module StripeMock
 
       def add_subscription_to_customer(cus, sub)
         cus[:subscriptions][:count] = (cus[:subscriptions][:count] || 0) + 1
-        cus[:subscriptions][:data] << sub
+        cus[:subscriptions][:data].unshift sub
+      end
+
+      def delete_subscription_from_customer(cus, subscription)
+        cus[:subscriptions][:data].reject!{|sub|
+          sub[:id] == subscription[:id]
+        }
+        cus[:subscriptions][:count] -=1
       end
 
       # `intervals` is set to 1 when calculating current_period_end from current_period_start & plan

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

@@ -25,7 +25,9 @@ module StripeMock
 
       def get_card_by_token(token)
         if token.nil? || @card_tokens[token].nil?
-          Data.mock_card :id => new_id('cc')
+          # TODO: Make this strict
+          msg = "Invalid token id: #{token}"
+          raise Stripe::InvalidRequestError.new(msg, 'tok', 404)
         else
           @card_tokens.delete(token)
         end

data/lib/stripe_mock/request_handlers/invoice_items.rb

@@ -3,11 +3,41 @@ module StripeMock
     module InvoiceItems
 
       def InvoiceItems.included(klass)
-        klass.add_handler 'post /v1/invoiceitems',  :new_invoice_item
+        klass.add_handler 'post /v1/invoiceitems',        :new_invoice_item
+        klass.add_handler 'post /v1/invoiceitems/(.*)',   :update_invoice_item
+        klass.add_handler 'get /v1/invoiceitems/(.*)',    :get_invoice_item
+        klass.add_handler 'get /v1/invoiceitems',         :list_invoice_items
+        klass.add_handler 'delete /v1/invoiceitems/(.*)', :delete_invoice_item
       end
 
       def new_invoice_item(route, method_url, params, headers)
-        Data.mock_invoice_item(params)
+        params[:id] ||= new_id('ii')
+        invoice_items[params[:id]] = Data.mock_invoice_item(params)
+      end
+
+      def update_invoice_item(route, method_url, params, headers)
+        route =~ method_url
+        list_item = assert_existance :list_item, $1, invoice_items[$1]
+        list_item.merge!(params)
+      end
+
+      def delete_invoice_item(route, method_url, params, headers)
+        route =~ method_url
+        assert_existance :list_item, $1, invoice_items[$1]
+
+        invoice_items[$1] = {
+          id: invoice_items[$1][:id],
+          deleted: true
+        }
+      end
+
+      def list_invoice_items(route, method_url, params, headers)
+        invoice_items.values
+      end
+
+      def get_invoice_item(route, method_url, params, headers)
+        route =~ method_url
+        assert_existance :invoice_item, $1, invoice_items[$1]
       end
 
     end

data/lib/stripe_mock/request_handlers/invoices.rb

@@ -8,6 +8,7 @@ module StripeMock
         klass.add_handler 'get /v1/invoices/(.*)',           :get_invoice
         klass.add_handler 'get /v1/invoices',                :list_invoices
         klass.add_handler 'post /v1/invoices/(.*)/pay',      :pay_invoice
+        klass.add_handler 'post /v1/invoices/(.*)',          :update_invoice
       end
 
       def new_invoice(route, method_url, params, headers)
@@ -16,6 +17,12 @@ module StripeMock
         invoices[id] = Data.mock_invoice([invoice_item], params.merge(:id => id))
       end
 
+      def update_invoice(route, method_url, params, headers)
+        route =~ method_url
+        assert_existance :invoice, $1, invoices[$1]
+        invoices[$1].merge!(params)
+      end
+
       def list_invoices(route, method_url, params, headers)
         params[:offset] ||= 0
         params[:count] ||= 10
@@ -32,13 +39,11 @@ module StripeMock
       def get_invoice(route, method_url, params, headers)
         route =~ method_url
         assert_existance :invoice, $1, invoices[$1]
-        invoices[$1] ||= Data.mock_invoice([], :id => $1)
       end
 
       def pay_invoice(route, method_url, params, headers)
         route =~ method_url
         assert_existance :invoice, $1, invoices[$1]
-        invoices[$1] ||= Data.mock_invoice([], :id => $1)
         invoices[$1].merge!(:paid => true, :attempted => true, :charge => 'ch_1fD6uiR9FAA2zc')
       end
 
@@ -48,7 +53,6 @@ module StripeMock
 
         customer = customers[params[:customer]]
         assert_existance :customer, params[:customer], customer
-        customer ||= Data.mock_customer([], :id => params[:customer])
 
         raise Stripe::InvalidRequestError.new("No upcoming invoices for customer: #{customer[:id]}", nil, 404) if customer[:subscriptions][:data].length == 0
 

data/lib/stripe_mock/request_handlers/plans.rb

@@ -11,27 +11,24 @@ module StripeMock
       end
 
       def new_plan(route, method_url, params, headers)
-        params[:id] = ( params[:id] || new_id('plan') ).to_s
+        validate_create_plan_params(params)
         plans[ params[:id] ] = Data.mock_plan(params)
       end
 
       def update_plan(route, method_url, params, headers)
         route =~ method_url
         assert_existance :plan, $1, plans[$1]
-        plans[$1] ||= Data.mock_plan(:id => $1)
         plans[$1].merge!(params)
       end
 
       def get_plan(route, method_url, params, headers)
         route =~ method_url
         assert_existance :plan, $1, plans[$1]
-        plans[$1] ||= Data.mock_plan(:id => $1)
       end
 
       def delete_plan(route, method_url, params, headers)
         route =~ method_url
-        assert_existance :plan, $1, plans[$1]
-        plans.delete($1)
+        assert_existance :plan, $1, plans.delete($1)
       end
 
       def list_plans(route, method_url, params, headers)

data/lib/stripe_mock/request_handlers/recipients.rb

@@ -4,22 +4,44 @@ module StripeMock
 
       def Recipients.included(klass)
         klass.add_handler 'post /v1/recipients',            :new_recipient
+        klass.add_handler 'post /v1/recipients/(.*)',       :update_recipient
         klass.add_handler 'get /v1/recipients/(.*)',        :get_recipient
       end
 
       def new_recipient(route, method_url, params, headers)
-        id = new_id('rp')
+        params[:id] ||= new_id('rp')
+        cards = []
+
         if params[:bank_account]
           params[:active_account] = get_bank_by_token(params.delete(:bank_account))
         end
-        recipients[id] = Data.mock_recipient(params.merge :id => id)
-        recipients[id]
+
+        if params[:card]
+          cards << get_card_by_token(params.delete(:card))
+          params[:default_card] = cards.first[:id]
+        end
+
+        recipients[ params[:id] ] = Data.mock_recipient(cards, params)
+        recipients[ params[:id] ]
+      end
+
+      def update_recipient(route, method_url, params, headers)
+        route =~ method_url
+        recipient = assert_existance :recipient, $1, recipients[$1]
+        recipient.merge!(params)
+
+        if params[:card]
+          new_card = get_card_by_token(params.delete(:card))
+          add_card_to_object(:recipient, new_card, recipient, true)
+          recipient[:default_card] = new_card[:id]
+        end
+
+        recipient
       end
 
       def get_recipient(route, method_url, params, headers)
         route =~ method_url
         assert_existance :recipient, $1, recipients[$1]
-        recipients[$1] ||= Data.mock_recipient(:id => $1)
       end
     end
   end

data/lib/stripe_mock/request_handlers/subscriptions.rb

@@ -12,61 +12,51 @@ module StripeMock
 
       def create_subscription(route, method_url, params, headers)
         route =~ method_url
+        customer = assert_existance :customer, $1, customers[$1]
 
-        customer = customers[$1]
-        assert_existance :customer, $1, customer
-
-        plan = plans[params[:plan]]
-        assert_existance :plan, params[:plan], plan
+        plan_id = params[:plan]
+        plan = assert_existance :plan, plan_id, plans[plan_id]
 
         if params[:card]
           new_card = get_card_by_token(params.delete(:card))
-          add_card_to_customer(new_card, customer)
+          add_card_to_object(:customer, new_card, customer)
           customer[:default_card] = new_card[:id]
         end
 
         # Ensure customer has card to charge if plan has no trial and is not free
-        verify_card_present(customer, plan)
+        verify_card_present(customer, plan, params)
 
         subscription = Data.mock_subscription({ id: (params[:id] || new_id('su')) })
         subscription.merge!(custom_subscription_params(plan, customer, params))
         add_subscription_to_customer(customer, subscription)
 
-        # oddly, subscription returned from 'create_subscription' does not expand plan
-        subscription.merge(plan: params[:plan])
+        subscription
       end
 
       def retrieve_subscription(route, method_url, params, headers)
         route =~ method_url
 
-        customer = customers[$1]
-        assert_existance :customer, $1, customer
-        subscription = get_customer_subscription(customer, $2)
-        assert_existance :subscription, $2, subscription
-
-        subscription
+        customer = assert_existance :customer, $1, customers[$1]
+        assert_existance :subscription, $2, get_customer_subscription(customer, $2)
       end
 
       def retrieve_subscriptions(route, method_url, params, headers)
         route =~ method_url
 
-        customer = customers[$1]
-        assert_existance :customer, $1, customer
-
+        customer = assert_existance :customer, $1, customers[$1]
         customer[:subscriptions]
       end
 
       def update_subscription(route, method_url, params, headers)
         route =~ method_url
+        customer = assert_existance :customer, $1, customers[$1]
 
-        customer = customers[$1]
-        assert_existance :customer, $1, customer
         subscription = get_customer_subscription(customer, $2)
         assert_existance :subscription, $2, subscription
 
         if params[:card]
           new_card = get_card_by_token(params.delete(:card))
-          add_card_to_customer(new_card, customer)
+          add_card_to_object(:customer, new_card, customer)
           customer[:default_card] = new_card[:id]
         end
 
@@ -78,26 +68,30 @@ module StripeMock
         params[:plan] = plan if params[:plan]
         verify_card_present(customer, plan)
 
+        if subscription[:cancel_at_period_end]
+          subscription[:cancel_at_period_end] = false
+          subscription[:canceled_at] = nil
+        end
+
         subscription.merge!(custom_subscription_params(plan, customer, params))
 
         # delete the old subscription, replace with the new subscription
         customer[:subscriptions][:data].reject! { |sub| sub[:id] == subscription[:id] }
         customer[:subscriptions][:data] << subscription
 
-        # oddly, subscription returned from 'create_subscription' does not expand plan
-        subscription.merge(plan: plan_name)
+        subscription
       end
 
       def cancel_subscription(route, method_url, params, headers)
         route =~ method_url
+        customer = assert_existance :customer, $1, customers[$1]
 
-        customer = customers[$1]
-        assert_existance :customer, $1, customer
         subscription = get_customer_subscription(customer, $2)
         assert_existance :subscription, $2, subscription
 
         cancel_params = { canceled_at: Time.now.utc.to_i }
-        if params[:at_period_end] == true
+        cancelled_at_period_end = (params[:at_period_end] == true)
+        if cancelled_at_period_end
           cancel_params[:cancel_at_period_end] = true
         else
           cancel_params[:status] = "canceled"
@@ -107,19 +101,18 @@ module StripeMock
 
         subscription.merge!(cancel_params)
 
-        customer[:subscriptions][:data].reject!{|sub|
-          sub[:id] == subscription[:id]
-        }
+        unless cancelled_at_period_end
+          delete_subscription_from_customer customer, subscription
+        end
 
-        customer[:subscriptions][:data] << subscription
         subscription
       end
 
       private
 
-      def verify_card_present(customer, plan)
-        if customer[:default_card].nil? && plan[:trial_period_days].nil? && plan[:amount] != 0
-          raise Stripe::InvalidRequestError.new('You must supply a valid card', nil, 400)
+      def verify_card_present(customer, plan, params={})
+        if customer[:default_card].nil? && plan[:trial_period_days].nil? && plan[:amount] != 0 && plan[:trial_end].nil? && params[:trial_end].nil?
+          raise Stripe::InvalidRequestError.new('You must supply a valid card xoxo', nil, 400)
         end
       end
 

data/lib/stripe_mock/request_handlers/tokens.rb

@@ -8,9 +8,29 @@ module StripeMock
       end
 
       def create_token(route, method_url, params, headers)
-        # "Sanitize" card number
-        params[:card][:last4] = params[:card][:number][-4,4]
-        token_id = generate_card_token(params[:card])
+        if params[:customer].nil? && params[:card].nil?
+          raise Stripe::InvalidRequestError.new('You must supply either a card, customer, or bank account to create a token.', nil, 400)
+        end
+
+        cus_id = params[:customer]
+
+        if cus_id && params[:card]
+          customer = assert_existance :customer, cus_id, customers[cus_id]
+
+          # params[:card] is an id; grab it from the db
+          customer_card = get_card(customer, params[:card])
+          assert_existance :card, params[:card], customer_card
+        elsif params[:card]
+          # params[:card] is a hash of cc info; "Sanitize" the card number
+          params[:card][:fingerprint] = StripeMock::Util.fingerprint(params[:card][:number])
+          params[:card][:last4] = params[:card][:number][-4,4]
+          customer_card = params[:card]
+        else
+          customer = assert_existance :customer, cus_id, customers[cus_id]
+          customer_card = get_card(customer, customer[:default_card])
+        end
+
+        token_id = generate_card_token(customer_card)
         card = @card_tokens[token_id]
 
         Data.mock_token(params.merge :id => token_id, :card => card)
@@ -30,4 +50,4 @@ module StripeMock
       end
     end
   end
-end
+end

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

@@ -0,0 +1,18 @@
+module StripeMock
+  module RequestHandlers
+    module ParamValidators
+
+      def validate_create_plan_params(params)
+        params[:id] = params[:id].to_s
+
+        @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
+      end
+
+    end
+  end
+end

data/lib/stripe_mock/server.rb

@@ -35,6 +35,10 @@ module StripeMock
       @instance.send(key)
     end
 
+    def destroy_resource(type, id)
+      @instance.send(type).delete(id)
+    end
+
     def clear_data
       @instance = Instance.new
     end
@@ -43,10 +47,6 @@ module StripeMock
       @instance.debug = toggle
     end
 
-    def set_strict(toggle)
-      @instance.strict = toggle
-    end
-
     def set_global_id_prefix(value)
       StripeMock.global_id_prefix = value
     end
@@ -68,7 +68,6 @@ module StripeMock
     end
 
     def debug?; @instance.debug; end
-    def strict?; @instance.strict; end
     def ping; true; end
   end
 

data/lib/stripe_mock/test_strategies/base.rb

@@ -0,0 +1,27 @@
+module StripeMock
+  module TestStrategies
+    class Base
+
+      def create_plan_params(params={})
+        {
+          :id => 'stripe_mock_default_plan_id',
+          :name => 'StripeMock Default Plan ID',
+          :amount => 1337,
+          :currency => 'usd',
+          :interval => 'month'
+        }.merge(params)
+      end
+
+      def generate_card_token(card_params={})
+        card = {
+          :number => "4242424242424242", :exp_month => 9, :exp_year => 2018, :cvc => "999"
+        }.merge(card_params)
+        card[:fingerprint] = StripeMock::Util.fingerprint(card[:number])
+
+        stripe_token = Stripe::Token.create(:card => card)
+        stripe_token.id
+      end
+
+    end
+  end
+end

data/lib/stripe_mock/test_strategies/live.rb

@@ -0,0 +1,22 @@
+module StripeMock
+  module TestStrategies
+    class Live < Base
+
+      def create_plan(params={})
+        raise "create_plan requires an :id" if params[:id].nil?
+        delete_plan(params[:id])
+        Stripe::Plan.create create_plan_params(params)
+      end
+
+      def delete_plan(plan_id)
+        begin
+          plan = Stripe::Plan.retrieve(plan_id)
+          plan.delete
+        rescue Stripe::StripeError => e
+          # Do nothing; we just want to make sure this plan ceases to exists
+        end
+      end
+
+    end
+  end
+end

data/lib/stripe_mock/test_strategies/mock.rb

@@ -0,0 +1,19 @@
+module StripeMock
+  module TestStrategies
+    class Mock < Base
+
+      def create_plan(params={})
+        Stripe::Plan.create create_plan_params(params)
+      end
+
+      def delete_plan(plan_id)
+        if StripeMock.state == 'remote'
+          StripeMock.client.destroy_resource('plans', plan_id)
+        elsif StripeMock.state == 'local'
+          StripeMock.instance.plans.delete(plan_id)
+        end
+      end
+
+    end
+  end
+end

data/lib/stripe_mock/util.rb

@@ -7,6 +7,7 @@ module StripeMock
 
       desh_hash.merge(source_hash) do |key, oldval, newval|
         if oldval.is_a?(Array) && newval.is_a?(Array)
+          oldval.fill(nil, oldval.length...newval.length)
           oldval.zip(newval).map {|elems|
             elems[1].nil? ? elems[0] : rmerge(elems[0], elems[1])
           }
@@ -18,5 +19,9 @@ module StripeMock
       end
     end
 
+    def self.fingerprint(source)
+      Digest::SHA1.base64digest(source).gsub(/[^a-z]/i, '')[0..15]
+    end
+
   end
 end

data/lib/stripe_mock/version.rb

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

data/lib/stripe_mock/webhook_fixtures/charge.failed.json

@@ -19,6 +19,7 @@
         "object": "card",
         "last4": "4242",
         "type": "Visa",
+        "brand": "Visa",
         "exp_month": 12,
         "exp_year": 2013,
         "fingerprint": "wXWJT135mEK107G8",
@@ -36,9 +37,9 @@
         "address_zip_check": null
       },
       "captured": true,
-      "refunds": [
+      "refunds": {
 
-      ],
+      },
       "balance_transaction": "txn_00000000000000",
       "failure_message": null,
       "failure_code": null,