stripe-ruby-mock 2.2.4 → 2.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 907a296926746f9f85a5971db9272ffe4c7cd8e7
-  data.tar.gz: 41deab127c933031f4e08fe4b54d0e62b5268cdb
+  metadata.gz: e17b14c466a95e28e23f56859b0f20ce0fc0e6f2
+  data.tar.gz: 14b2121916a871814374262884df72a0501c32d9
 SHA512:
-  metadata.gz: 55104b30501886b7a11efbef32fb28eb1c5eb2c2d130bd9370e532f2ef417ee9a4338e0a08d1993f6bc62fe1a659ac7596aa51ff7a851c225b1054d228cd1b61
-  data.tar.gz: 430679dd30c713bd45a8ae8831c855862848706c01f46c871e66a2ae3e906a85944f11179dc596d6d729705287193ecb9146cc960636f4bf69434faf06794914
+  metadata.gz: 1cf3ee166d1a5e9cac2b1c61bf50822568986b84fdaff85166739137d936bfe34f0cdf08ddc25ee6781412cb8a84dd55b4b4beae2a440fecf4966f6c042a7316
+  data.tar.gz: 8237f3fbfbc338f16e64ac7671acf8715254fd35fd6ab992e55c34c68dd79bc15c4526831694e954df9327428c55cca9d32394d1d3b7fa89e2e7d4846e69f5e1

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.2.4', :require => 'stripe_mock'
+    gem 'stripe-ruby-mock', '~> 2.3.0', :require => 'stripe_mock'
 
 ## Features
 

data/lib/stripe_mock/api/bank_tokens.rb

@@ -1,13 +1,13 @@
 module StripeMock
 
-  def self.generate_bank_token(bank_params)
-    if @state == 'local'
+  def self.generate_bank_token(bank_params = {})
+    case @state
+    when 'local'
       instance.generate_bank_token(bank_params)
-    elsif @state == 'remote'
+    when 'remote'
       client.generate_bank_token(bank_params)
     else
       raise UnstartedStateError
     end
   end
-
-end
+end

data/lib/stripe_mock/api/card_tokens.rb

@@ -1,13 +1,13 @@
 module StripeMock
 
-  def self.generate_card_token(card_params)
-    if @state == 'local'
+  def self.generate_card_token(card_params = {})
+    case @state
+    when 'local'
       instance.generate_card_token(card_params)
-    elsif @state == 'remote'
+    when 'remote'
       client.generate_card_token(card_params)
     else
       raise UnstartedStateError
     end
   end
-
 end

data/lib/stripe_mock/api/client.rb

@@ -1,6 +1,8 @@
 module StripeMock
 
-  def self.client; @client; end
+  def self.client
+    @client
+  end
 
   def self.start_client(port=4999)
     return false if @state == 'live'
@@ -27,11 +29,12 @@ module StripeMock
 
   def self.redirect_to_mock_server(method, url, api_key, params={}, headers={}, api_base_url=nil)
     handler = Instance.handler_for_method_url("#{method} #{url}")
-    mock_error = client.error_queue.error_for_handler_name(handler[:name])
-    if mock_error
+
+    if mock_error = client.error_queue.error_for_handler_name(handler[:name])
       client.error_queue.dequeue
       raise mock_error
     end
+
     Stripe::Util.symbolize_names client.mock_request(method, url, api_key, params, headers)
   end
 

data/lib/stripe_mock/api/errors.rb

@@ -4,12 +4,12 @@ module StripeMock
     handler_names.push(:all) if handler_names.count == 0
 
     if @state == 'local'
-      instance.error_queue.queue(stripe_error, handler_names)
+      instance
     elsif @state == 'remote'
-      client.error_queue.queue(stripe_error, handler_names)
+      client
     else
       raise UnstartedStateError
-    end
+    end.error_queue.queue stripe_error, handler_names
   end
 
   def self.prepare_card_error(code, *handler_names)
@@ -17,7 +17,7 @@ module StripeMock
 
     args = CardErrors.argument_map[code]
     raise StripeMockError.new("Unrecognized stripe card error code: #{code}") if args.nil?
-    self.prepare_error  Stripe::CardError.new(*args), *handler_names
+    self.prepare_error Stripe::CardError.new(*args), *handler_names
   end
 
   module CardErrors

data/lib/stripe_mock/api/server.rb

@@ -1,5 +1,4 @@
 module StripeMock
-
   @default_server_pid_path = './stripe-mock-server.pid'
   @default_server_log_path = './stripe-mock-server.log'
 

data/lib/stripe_mock/client.rb

@@ -1,23 +1,24 @@
 module StripeMock
-
   class Client
-    attr_reader :port, :state, :error_queue
+    attr_reader :port, :state
 
     def initialize(port)
       @port = port
-      @pipe = Jimson::Client.new("http://0.0.0.0:#{port}")
+
+      DRb.start_service
+      @pipe = DRbObject.new_with_uri "druby://localhost:#{port}"
+
       # Ensure client can connect to server
       timeout_wrap(5) { @pipe.ping }
       @state = 'ready'
-      @error_queue = ErrorQueue.new
     end
 
     def mock_request(method, url, api_key, params={}, headers={})
       timeout_wrap do
         @pipe.mock_request(method, url, api_key, params, headers).tap {|result|
           response, api_key = result
-          if response.is_a?(Hash) && response['error_raised'] == 'invalid_request'
-            raise Stripe::InvalidRequestError.new(*response['error_params'])
+          if response.is_a?(Hash) && response[:error_raised] == 'invalid_request'
+            raise Stripe::InvalidRequestError.new(*response[:error_params])
           end
         }
       end
@@ -32,6 +33,10 @@ module StripeMock
       }
     end
 
+    def error_queue
+      timeout_wrap { @pipe.error_queue }
+    end
+
     def set_server_debug(toggle)
       timeout_wrap { @pipe.set_debug(toggle) }
     end
@@ -86,7 +91,7 @@ module StripeMock
         yield
       rescue ClosedClientConnectionError
         raise
-      rescue Errno::ECONNREFUSED => e
+      rescue Errno::ECONNREFUSED, DRb::DRbConnError => e
         tries -= 1
         if tries > 0
           if tries == original_tries - 1

data/lib/stripe_mock/data.rb

@@ -192,7 +192,9 @@ module StripeMock
         object: "refund",
         balance_transaction: "txn_4fWh2RKvgxcXqV",
         metadata: {},
-        charge: "ch_4fWhYjzQ23UFWT"
+        charge: "ch_4fWhYjzQ23UFWT",
+        receipt_number: nil,
+        status: "succeeded"
       }.merge(params)
     end
 
@@ -695,9 +697,53 @@ module StripeMock
       }
     end
 
-    def self.mock_list_object(data, params = {})
+    def self.mock_list_object(data, params={})
       list = StripeMock::Data::List.new(data, params)
       list.to_h
     end
+
+    def self.mock_balance_transactions(ids=[])
+      bts = {}
+      ids.each do |id|
+        bts[id] = self.mock_balance_transaction(id: id)
+      end
+      bts
+    end
+
+    def self.mock_balance_transaction(params = {})
+      bt_id = params[:id] || 'test_txn_default'
+      source = params[:source] || 'ch_test_charge'
+      {
+        id: bt_id,
+        object: "balance_transaction",
+        amount: 10000,
+        available_on: 1462406400,
+        created: 1461880226,
+        currency: "usd",
+        description: nil,
+        fee: 320,
+        fee_details: [
+          {
+            amount: 320,
+            application: nil,
+            currency: "usd",
+            description: "Stripe processing fees",
+            type: "stripe_fee"
+          }
+        ],
+        net: 9680,
+        source: source,
+        sourced_transfers: {
+          object: "list",
+          data: [],
+          has_more: false,
+          total_count: 0,
+          url: "/v1/transfers?source_transaction=#{source}"
+        },
+        status: "pending",
+        type: "charge"
+      }.merge(params)
+    end
+
   end
 end

data/lib/stripe_mock/error_queue.rb

@@ -1,5 +1,9 @@
+require 'drb/drb'
+
 module StripeMock
   class ErrorQueue
+    include DRb::DRbUndumped
+    extend DRb::DRbUndumped
 
     def initialize
       @queue = []
@@ -20,4 +24,4 @@ module StripeMock
     end
 
   end
-end
+end

data/lib/stripe_mock/instance.rb

@@ -21,6 +21,7 @@ module StripeMock
     end
 
     include StripeMock::RequestHandlers::Accounts
+    include StripeMock::RequestHandlers::BalanceTransactions
     include StripeMock::RequestHandlers::Charges
     include StripeMock::RequestHandlers::Cards
     include StripeMock::RequestHandlers::Sources
@@ -38,14 +39,15 @@ module StripeMock
     include StripeMock::RequestHandlers::Tokens
 
 
-    attr_reader :accounts, :bank_tokens, :charges, :coupons, :customers, :disputes, :events,
-                :invoices, :invoice_items, :orders, :plans, :recipients, :transfers,
-                :subscriptions
+    attr_reader :accounts, :balance_transactions, :bank_tokens, :charges, :coupons, :customers,
+                :disputes, :events, :invoices, :invoice_items, :orders, :plans, :recipients,
+                :transfers, :subscriptions
 
     attr_accessor :error_queue, :debug
 
     def initialize
       @accounts = {}
+      @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 = {}

data/lib/stripe_mock/request_handlers/balance_transactions.rb

@@ -0,0 +1,21 @@
+module StripeMock
+  module RequestHandlers
+    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
+      end
+
+      def get_balance_transaction(route, method_url, params, headers)
+        route =~ method_url
+        assert_existence :balance_transaction, $1, balance_transactions[$1]
+      end
+
+      def list_balance_transactions(route, method_url, params, headers)
+        Data.mock_list_object(balance_transactions.values, params)
+      end
+
+    end
+  end
+end

data/lib/stripe_mock/request_handlers/charges.rb

@@ -15,17 +15,24 @@ module StripeMock
       def new_charge(route, method_url, params, headers)
         id = new_id('ch')
 
-        if params[:source] && params[:source].is_a?(String)
-          # if a customer is provided, the card parameter is assumed to be the actual
-          # 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])
-          else
-            params[:source] = get_card_by_token(params[:source])
+        if params[:source]
+          if params[:source].is_a?(String)
+            # if a customer is provided, the card parameter is assumed to be the actual
+            # 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])
+            else
+              params[:source] = get_card_by_token(params[:source])
+            end
+          elsif params[:source][:id]
+            raise Stripe::InvalidRequestError.new("Invalid token id: #{params[:source]}", 'card', 400)
+          end
+        elsif params[:customer]
+          customer = customers[params[:customer]]
+          if customer && customer[:default_source]
+            params[:source] = get_card(customer, customer[:default_source])
           end
-        elsif params[:source] && params[:source][:id]
-          raise Stripe::InvalidRequestError.new("Invalid token id: #{params[:card]}", 'card', 400)
         end
 
         ensure_required_params(params)

data/lib/stripe_mock/request_handlers/customers.rb

@@ -41,6 +41,7 @@ module StripeMock
           subscription = Data.mock_subscription({ id: new_id('su') })
           subscription.merge!(custom_subscription_params(plan, customers[ params[:id] ], params))
           add_subscription_to_customer(customers[ params[:id] ], subscription)
+          subscriptions[subscription[:id]] = subscription
         elsif params[:trial_end]
           raise Stripe::InvalidRequestError.new('Received unknown parameter: trial_end', nil, 400)
         end

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

@@ -17,6 +17,23 @@ module StripeMock
         card
       end
 
+      def add_source_to_object(type, source, object, replace_current=false)
+        source[type] = object[:id]
+        sources = object[:sources]
+
+        if replace_current && sources[:data]
+          sources[:data].delete_if {|source| source[:id] == object[:default_source]}
+          object[:default_source] = source[:id]
+          sources[:data] = [source]
+        else
+          sources[:total_count] = (sources[:total_count] || 0) + 1
+          (sources[:data] ||= []) << source
+        end
+        object[:default_source] = source[:id] if object[:default_source].nil?
+
+        source
+      end
+
       def add_card_to_object(type, card, object, replace_current=false)
         card[type] = object[:id]
         cards_or_sources = object[:cards] || object[:sources]
@@ -64,6 +81,24 @@ module StripeMock
         card
       end
 
+      def add_source_to(type, type_id, params, objects)
+        resource = assert_existence type, type_id, objects[type_id]
+
+        source =
+          if params[:card]
+            card_from_params(params[:card])
+          elsif params[:bank_account]
+            get_bank_by_token(params[:bank_account])
+          else
+            begin
+              get_card_by_token(params[:source])
+            rescue Stripe::InvalidRequestError
+              get_bank_by_token(params[:source])
+            end
+          end
+        add_source_to_object(type, source, resource)
+      end
+
       def add_card_to(type, type_id, params, objects)
         resource = assert_existence type, type_id, objects[type_id]
 

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

@@ -2,14 +2,14 @@ module StripeMock
   module RequestHandlers
     module Helpers
 
-      def generate_bank_token(bank_params)
+      def generate_bank_token(bank_params = {})
         token = new_id 'btok'
         bank_params[:id] = new_id 'bank_account'
         @bank_tokens[token] = Data.mock_bank_account bank_params
         token
       end
 
-      def generate_card_token(card_params)
+      def generate_card_token(card_params = {})
         token = new_id 'tok'
         card_params[:id] = new_id 'cc'
         @card_tokens[token] = Data.mock_card symbolize_names(card_params)

data/lib/stripe_mock/request_handlers/sources.rb

@@ -13,7 +13,7 @@ module StripeMock
 
       def create_source(route, method_url, params, headers)
         route =~ method_url
-        add_card_to(:customer, $1, params, customers)
+        add_source_to(:customer, $1, params, customers)
       end
 
       def retrieve_sources(route, method_url, params, headers)

data/lib/stripe_mock/request_handlers/subscriptions.rb

@@ -3,20 +3,41 @@ module StripeMock
     module Subscriptions
 
       def Subscriptions.included(klass)
-        klass.add_handler 'get /v1/customers/(.*)/subscriptions', :retrieve_subscriptions
-        klass.add_handler 'post /v1/customers/(.*)/subscriptions', :create_subscription
-        klass.add_handler 'get /v1/customers/(.*)/subscriptions/(.*)', :retrieve_subscription
-        klass.add_handler 'post /v1/customers/(.*)/subscriptions/(.*)', :update_subscription
-        klass.add_handler 'delete /v1/customers/(.*)/subscriptions/(.*)', :cancel_subscription
+        klass.add_handler 'get /v1/subscriptions', :retrieve_subscriptions
+        klass.add_handler 'post /v1/subscriptions', :create_subscription
+        klass.add_handler 'get /v1/subscriptions/(.*)', :retrieve_subscription
+        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
       end
 
-      def create_subscription(route, method_url, params, headers)
+      def retrieve_customer_subscription(route, method_url, params, headers)
+        route =~ method_url
+
+        customer = :customer, $1, customers[$1]
+        subscription = get_customer_subscription(customer, $2)
+
+        assert_existence :subscription, $2, subscription
+      end
+
+      def retrieve_customer_subscriptions(route, method_url, params, headers)
         route =~ method_url
+
         customer = assert_existence :customer, $1, customers[$1]
+        customer[:subscriptions]
+      end
+
+      def create_customer_subscription(route, method_url, params, headers)
+        route =~ method_url
 
         plan_id = params[:plan].to_s
         plan = assert_existence :plan, plan_id, plans[plan_id]
 
+        customer = assert_existence :customer, $1, customers[$1]
+
         if params[:source]
           new_card = get_card_by_token(params.delete(:source))
           add_card_to_object(:customer, new_card, customer)
@@ -42,32 +63,72 @@ module StripeMock
           subscription[:discount] = Stripe::Util.convert_to_stripe_object({ coupon: coupon }, {})
         end
 
+        subscriptions[subscription[:id]] = subscription
         add_subscription_to_customer(customer, subscription)
 
+        subscriptions[subscription[:id]]
+      end
 
-        subscription
+      def create_subscription(route, method_url, params, headers)
+        route =~ method_url
+
+        plan_id = params[:plan].to_s
+        plan = assert_existence :plan, plan_id, plans[plan_id]
+
+        customer_id = params[:customer].to_s
+        customer = assert_existence :customer, customer_id, customers[customer_id]
+
+        if params[:source]
+          new_card = get_card_by_token(params.delete(:source))
+          add_card_to_object(:customer, new_card, customer)
+          customer[:default_source] = new_card[:id]
+        end
+
+        # Ensure customer has card to charge if plan has no trial and is not free
+        verify_card_present(customer, plan, params)
+
+        subscription = Data.mock_subscription({ id: (params[:id] || new_id('su')) })
+        subscription.merge!(custom_subscription_params(plan, customer, params))
+
+        if params[:coupon]
+          coupon_id = params[:coupon]
+
+          raise Stripe::InvalidRequestError.new("No such coupon: #{coupon_id}", 'coupon', 400) unless coupons[coupon_id]
+
+          # FIXME assert_existence returns 404 error code but Stripe returns 400
+          # coupon = assert_existence :coupon, coupon_id, coupons[coupon_id]
+
+          coupon = Data.mock_coupon({ id: coupon_id })
+
+          subscription[:discount] = Stripe::Util.convert_to_stripe_object({ coupon: coupon }, {})
+        end
+
+        subscriptions[subscription[:id]] = subscription
+        add_subscription_to_customer(customer, subscription)
+
+        subscriptions[subscription[:id]]
       end
 
       def retrieve_subscription(route, method_url, params, headers)
         route =~ method_url
 
-        customer = assert_existence :customer, $1, customers[$1]
-        assert_existence :subscription, $2, get_customer_subscription(customer, $2)
+        assert_existence :subscription, $1, subscriptions[$1]
       end
 
       def retrieve_subscriptions(route, method_url, params, headers)
         route =~ method_url
 
-        customer = assert_existence :customer, $1, customers[$1]
-        customer[:subscriptions]
+        Data.mock_list_object(subscriptions.values, params)
+        #customer = assert_existence :customer, $1, customers[$1]
+        #customer[:subscriptions]
       end
 
       def update_subscription(route, method_url, params, headers)
         route =~ method_url
-        customer = assert_existence :customer, $1, customers[$1]
+        subscription = assert_existence :subscription, $1, subscriptions[$1]
 
-        subscription = get_customer_subscription(customer, $2)
-        assert_existence :subscription, $2, subscription
+        customer_id = subscription[:customer]
+        customer = assert_existence :customer, customer_id, customers[customer_id]
 
         if params[:source]
           new_card = get_card_by_token(params.delete(:source))
@@ -111,10 +172,11 @@ module StripeMock
 
       def cancel_subscription(route, method_url, params, headers)
         route =~ method_url
-        customer = assert_existence :customer, $1, customers[$1]
 
-        subscription = get_customer_subscription(customer, $2)
-        assert_existence :subscription, $2, subscription
+        subscription = assert_existence :subscription, $1, subscriptions[$1]
+
+        customer_id = subscription[:customer]
+        customer = assert_existence :customer, customer_id, customers[customer_id]
 
         cancel_params = { canceled_at: Time.now.utc.to_i }
         cancelled_at_period_end = (params[:at_period_end] == true)

data/lib/stripe_mock/request_handlers/tokens.rb

@@ -8,7 +8,7 @@ module StripeMock
       end
 
       def create_token(route, method_url, params, headers)
-        if params[:customer].nil? && params[:card].nil?
+        if params[:customer].nil? && params[:card].nil? && params[:bank_account].nil?
           raise Stripe::InvalidRequestError.new('You must supply either a card, customer, or bank account to create a token.', nil, 400)
         end
 
@@ -31,15 +31,25 @@ module StripeMock
           params[:card][:fingerprint] = StripeMock::Util.fingerprint(params[:card][:number])
           params[:card][:last4] = params[:card][:number][-4,4]
           customer_card = params[:card]
+        elsif params[:bank_account]
+          # params[:card] is a hash of cc info; "Sanitize" the card number
+          bank_account = params[:bank_account]
         else
           customer = assert_existence :customer, cus_id, customers[cus_id]
           customer_card = get_card(customer, customer[:default_source])
         end
 
-        token_id = generate_card_token(customer_card)
-        card = @card_tokens[token_id]
+        if bank_account
+          token_id = generate_bank_token(bank_account)
+          bank_account = @bank_tokens[token_id]
+
+          Data.mock_bank_account_token(params.merge :id => token_id, :bank_account => bank_account)
+        else
+          token_id = generate_card_token(customer_card)
+          card = @card_tokens[token_id]
 
-        Data.mock_card_token(params.merge :id => token_id, :card => card)
+          Data.mock_card_token(params.merge :id => token_id, :card => card)
+        end
       end
 
       def get_token(route, method_url, params, headers)

data/lib/stripe_mock/server.rb

@@ -1,19 +1,15 @@
-require 'jimson-temp'
+require 'drb/drb'
 
 module StripeMock
-
   class Server
-    extend Jimson::Handler
-
     def self.start_new(opts)
       puts "Starting StripeMock server on port #{opts[:port] || 4999}"
-      server = Jimson::Server.new(Server.new,
-        :host => opts[:host] || '0.0.0.0',
-        :port => opts[:port] || 4999,
-        :server => opts[:server] || :thin,
-        :show_errors => true
-      )
-      server.start
+
+      host = opts.fetch :host,'0.0.0.0'
+      port = opts.fetch :port, 4999
+
+      DRb.start_service "druby://#{host}:#{port}", Server.new
+      DRb.thread.join
     end
 
     def initialize
@@ -26,7 +22,9 @@ module StripeMock
       rescue Stripe::InvalidRequestError => e
         {
           :error_raised => 'invalid_request',
-          :error_params => [e.message, e.param, e.http_status, e.http_body, e.json_body]
+          :error_params => [
+            e.message, e.param, e.http_status, e.http_body, e.json_body
+          ]
         }
       end
     end
@@ -67,8 +65,16 @@ module StripeMock
       @instance.generate_webhook_event(event_data)
     end
 
-    def debug?; @instance.debug; end
-    def ping; true; end
-  end
+    def error_queue
+      @instance.error_queue
+    end
+
+    def debug?
+      @instance.debug
+    end
 
+    def ping
+      true
+    end
+  end
 end