mustwin-stripe-ruby-mock 1.8.4.10

data/lib/stripe_mock/error_queue.rb

@@ -0,0 +1,23 @@
+module StripeMock
+  class ErrorQueue
+
+    def initialize
+      @queue = []
+    end
+
+    def queue(error, handler_names)
+      @queue << handler_names.map {|n| [n, error]}
+    end
+
+    def error_for_handler_name(handler_name)
+      return nil if @queue.count == 0
+      triggers = @queue.first
+      (triggers.assoc(:all) || triggers.assoc(handler_name) || [])[1]
+    end
+
+    def dequeue
+      @queue.shift
+    end
+
+  end
+end

data/lib/stripe_mock/errors/closed_client_connection_error.rb

@@ -0,0 +1,9 @@
+module StripeMock
+  class ClosedClientConnectionError < StripeMockError
+
+    def initialize
+      super("This StripeMock client has already been closed.")
+    end
+
+  end
+end

data/lib/stripe_mock/errors/server_timeout_error.rb

@@ -0,0 +1,12 @@
+module StripeMock
+  class ServerTimeoutError < StripeMockError
+
+    attr_reader :associated_error
+
+    def initialize(associated_error)
+      @associated_error = associated_error
+      super("Unable to connect to stripe mock server (did you forget to run `$ stripe-mock-server`?)")
+    end
+
+  end
+end

data/lib/stripe_mock/errors/stripe_mock_error.rb

@@ -0,0 +1,15 @@
+module StripeMock
+  class StripeMockError < StandardError
+
+    attr_reader :message
+
+    def initialize(message)
+      @message = message
+    end
+
+    def to_s
+      @message
+    end
+
+  end
+end

data/lib/stripe_mock/errors/uninitialized_instance_error.rb

@@ -0,0 +1,9 @@
+module StripeMock
+  class UninitializedInstanceError < StripeMockError
+
+    def initialize
+      super("StripeMock instance is nil (did you forget to call `StripeMock.start`?)")
+    end
+
+  end
+end

data/lib/stripe_mock/errors/unstarted_state_error.rb

@@ -0,0 +1,9 @@
+module StripeMock
+  class UnstartedStateError < StripeMockError
+
+    def initialize
+      super("StripeMock has not been started. Please call StripeMock.start or StripeMock.start_client")
+    end
+
+  end
+end

data/lib/stripe_mock/errors/unsupported_request_error.rb

@@ -0,0 +1,4 @@
+module StripeMock
+  class UnsupportedRequestError < StripeMockError
+  end
+end

data/lib/stripe_mock/instance.rb

@@ -0,0 +1,108 @@
+module StripeMock
+  class Instance
+
+    # Handlers are ordered by priority
+    @@handlers = []
+
+    def self.add_handler(route, name)
+      @@handlers << {
+        :route => %r{^#{route}$},
+        :name => name
+      }
+    end
+
+    def self.handler_for_method_url(method_url)
+      @@handlers.find {|h| method_url =~ h[:route] }
+    end
+
+    include StripeMock::RequestHandlers::Charges
+    include StripeMock::RequestHandlers::Customers
+    include StripeMock::RequestHandlers::InvoiceItems
+    include StripeMock::RequestHandlers::Plans
+
+
+    attr_reader :charges, :customers, :plans, :error_queue
+    attr_accessor :debug, :strict
+
+    def initialize
+      @customers = {}
+      @charges = {}
+      @plans = {}
+      @recipient_tokens = {}
+      @card_tokens = {}
+
+      @id_counter = 0
+      @error_queue = ErrorQueue.new
+      @debug = false
+      @strict = true
+    end
+
+    def mock_request(method, url, api_key, params={}, headers={})
+      return {} if method == :xtest
+
+      # Ensure params hash has symbols as keys
+      params = Stripe::Util.symbolize_names(params)
+
+      method_url = "#{method} #{url}"
+
+      if handler = Instance.handler_for_method_url(method_url)
+        if @debug == true
+          puts "[StripeMock req]::#{handler[:name]} #{method} #{url}"
+          puts "                  #{params}"
+        end
+
+        if mock_error = @error_queue.error_for_handler_name(handler[:name])
+          @error_queue.dequeue
+          raise mock_error
+        else
+          res = self.send(handler[:name], handler[:route], method_url, params, headers)
+          puts "           [res]  #{res}" if @debug == true
+          [res, api_key]
+        end
+      else
+        puts "WARNING: Unrecognized method + url: [#{method} #{url}]"
+        puts " params: #{params}"
+        [{}, api_key]
+      end
+    end
+
+    def generate_recipient_token(recipient_params)
+      token = new_id 'tok'
+      recipient_params[:id] = new_id 'rec'
+      @recipient_tokens[token] = Data.mock_card(recipient_params)
+      token
+    end
+
+    def generate_card_token(card_params)
+      token = new_id 'tok'
+      card_params[:id] = new_id 'cc'
+      @card_tokens[token] = Data.mock_card(card_params)
+      token
+    end
+
+    def get_card_by_token(token)
+      if token.nil? || @card_tokens[token].nil?
+        Data.mock_card :id => new_id('cc')
+      else
+        @card_tokens.delete(token)
+      end
+    end
+
+    private
+
+    def assert_existance(type, id, obj, message=nil)
+      return unless @strict == true
+
+      if obj.nil?
+        msg = message || "No such #{type}: #{id}"
+        raise Stripe::InvalidRequestError.new(msg, type.to_s, 404)
+      end
+    end
+
+    def new_id(prefix)
+      # Stripe ids must be strings
+      "test_#{prefix}_#{@id_counter += 1}"
+    end
+
+  end
+end

data/lib/stripe_mock/request_handlers/charges.rb

@@ -0,0 +1,33 @@
+module StripeMock
+  module RequestHandlers
+    module Charges
+
+      def Charges.included(klass)
+        klass.add_handler 'post /v1/charges',               :new_charge
+        klass.add_handler 'get /v1/charges/(.*)',           :get_charge
+        klass.add_handler 'post /v1/charges/(.*)/capture',  :capture_charge
+      end
+
+      def new_charge(route, method_url, params, headers)
+        id = new_id('ch')
+        charges[id] = Data.mock_charge(params.merge :id => id)
+      end
+
+      def get_charge(route, method_url, params, headers)
+        route =~ method_url
+        assert_existance :charge, $1, charges[$1]
+        charges[$1] ||= Data.mock_charge(:id => $1)
+      end
+
+      def capture_charge(route, method_url, params, headers)
+        route =~ method_url
+        charge = charges[$1]
+        assert_existance :charge, $1, charge
+
+        charge[:captured] = true
+        charge
+      end
+
+    end
+  end
+end

data/lib/stripe_mock/request_handlers/customers.rb

@@ -0,0 +1,107 @@
+module StripeMock
+  module RequestHandlers
+    module Customers
+
+      def Customers.included(klass)
+        klass.add_handler 'post /v1/customers',                     :new_customer
+        klass.add_handler 'post /v1/customers/(.*)/subscription',   :update_subscription
+        klass.add_handler 'delete /v1/customers/(.*)/subscription', :cancel_subscription
+        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
+
+      def new_customer(route, method_url, params, headers)
+        params[:id] ||= new_id('cus')
+        cards = []
+        if params[:card]
+          cards << get_card_by_token(params.delete(:card))
+          params[:default_card] = cards.first[:id]
+        end
+        customers[ params[:id] ] = Data.mock_customer(cards, params)
+      end
+
+      def update_subscription(route, method_url, params, headers)
+        route =~ method_url
+
+        customer = customers[$1]
+        assert_existance :customer, $1, customer
+
+        plan = plans[ params[:plan] ]
+        assert_existance :plan, params[:plan], plan
+
+        # Ensure customer has card to charge if plan has no trial and is not free
+        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)
+        end
+
+        sub = Data.mock_subscription id: new_id('su'), plan: plan, customer: $1
+        customer[:subscription] = sub
+      end
+
+      def cancel_subscription(route, method_url, params, headers)
+        route =~ method_url
+
+        customer = customers[$1]
+        assert_existance :customer, $1, customer
+
+        sub = customer[:subscription]
+        assert_existance nil, nil, sub, "No active subscription for customer: #{$1}"
+
+        customer[:subscription] = nil
+
+        plan = plans[ sub[:plan][:id] ]
+        assert_existance :plan, params[:plan], plan
+
+        Data.mock_delete_subscription(id: sub[:id])
+      end
+
+      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.merge!(params)
+
+        if params[:card]
+          new_card = get_card_by_token(params.delete(:card))
+
+          if cus[:cards][:count] == 0
+            cus[:cards][:count] += 1
+          else
+            cus[:cards][:data].delete_if {|card| card[:id] == cus[:default_card]}
+          end
+
+          cus[:cards][:data] << new_card
+          cus[:default_card] = new_card[:id]
+        end
+
+        cus
+      end
+
+      def delete_customer(route, method_url, params, headers)
+        route =~ method_url
+        assert_existance :customer, $1, customers[$1]
+
+        customers[$1] = {
+          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)
+        customers.values
+      end
+
+    end
+  end
+end

data/lib/stripe_mock/request_handlers/invoice_items.rb

@@ -0,0 +1,15 @@
+module StripeMock
+  module RequestHandlers
+    module InvoiceItems
+
+      def InvoiceItems.included(klass)
+        klass.add_handler 'post /v1/invoiceitems',  :new_invoice_item
+      end
+
+      def new_invoice_item(route, method_url, params, headers)
+        Data.mock_invoice(params)
+      end
+
+    end
+  end
+end

data/lib/stripe_mock/request_handlers/plans.rb

@@ -0,0 +1,43 @@
+module StripeMock
+  module RequestHandlers
+    module Plans
+
+      def Plans.included(klass)
+        klass.add_handler 'post /v1/plans',        :new_plan
+        klass.add_handler 'post /v1/plans/(.*)',   :update_plan
+        klass.add_handler 'get /v1/plans/(.*)',    :get_plan
+        klass.add_handler 'delete /v1/plans/(.*)', :delete_plan
+        klass.add_handler 'get /v1/plans',         :list_plans
+      end
+
+      def new_plan(route, method_url, params, headers)
+        params[:id] ||= new_id('plan')
+        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)
+      end
+
+      def list_plans(route, method_url, params, headers)
+        plans.values
+      end
+
+    end
+  end
+end

data/lib/stripe_mock/server.rb

@@ -0,0 +1,58 @@
+require 'jimson-temp'
+
+module StripeMock
+
+  class Server
+    extend Jimson::Handler
+
+    def self.start_new(opts)
+      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
+    end
+
+    def initialize
+      self.clear_data
+    end
+
+    def mock_request(*args)
+      begin
+        @instance.mock_request(*args)
+      rescue Stripe::InvalidRequestError => e
+        {
+          :error_raised => 'invalid_request',
+          :error_params => [e.message, e.param, e.http_status, e.http_body, e.json_body]
+        }
+      end
+    end
+
+    def get_data(key)
+      @instance.send(key)
+    end
+
+    def clear_data
+      @instance = Instance.new
+    end
+
+    def set_debug(toggle)
+      @instance.debug = toggle
+    end
+
+    def set_strict(toggle)
+      @instance.strict = toggle
+    end
+
+    def generate_card_token(card_params)
+      @instance.generate_card_token(card_params)
+    end
+
+    def debug?; @instance.debug; end
+    def strict?; @instance.strict; end
+    def ping; true; end
+  end
+
+end