fake_braintree 0.6.0 → 0.7.0

data/lib/fake_braintree/credit_card_serializer.rb

@@ -0,0 +1,20 @@
+module FakeBraintree
+  class CreditCardSerializer
+    def initialize(credit_card)
+      @credit_card = credit_card
+    end
+
+    def to_h
+      last_2 = @credit_card.last_4[-2..-1]
+      card_type = @credit_card.card_type
+      {
+        type: 'CreditCard',
+        description: "ending in #{last_2}",
+        details: {
+          cardType: card_type,
+          lastTwo: last_2
+        }
+      }
+    end
+  end
+end

data/lib/fake_braintree/customer.rb

@@ -1,3 +1,7 @@
+require 'fake_braintree/helpers'
+require 'fake_braintree/credit_card'
+require 'fake_braintree/valid_credit_cards'
+
 module FakeBraintree
   class Customer
     include Helpers
@@ -39,8 +43,12 @@ module FakeBraintree
     end
 
     def delete
-      delete_customer_with_id(customer_id)
-      deletion_response
+      if customer_exists_in_registry?
+        delete_customer_with_id(customer_id)
+        deletion_response
+      else
+        response_for_customer_not_found
+      end
     end
 
     private
@@ -145,7 +153,7 @@ module FakeBraintree
     end
 
     def delete_customer_with_id(id)
-      FakeBraintree.registry.customers[id] = nil
+      FakeBraintree.registry.customers.delete(id)
     end
 
     def deletion_response

data/lib/fake_braintree/helpers.rb

@@ -1,3 +1,6 @@
+require 'digest/md5'
+require 'active_support/gzip'
+
 module FakeBraintree
   module Helpers
     def gzip(content)

data/lib/fake_braintree/merchant_account.rb

@@ -0,0 +1,95 @@
+module FakeBraintree
+  class MerchantAccount
+    include Helpers
+
+    def initialize(merchant_account_hash_from_params, options)
+      @merchant_account_hash = {
+        'id' => options[:id]
+      }
+      @merchant_account_hash.merge!(merchant_account_hash_from_params)
+      set_merchant_account_id
+    end
+
+    def create
+      if invalid?
+        response_for_invalid_merchant_account
+      else
+        create_merchant_account_with(merchant_account_hash)
+        response_for_created_merchant_account(merchant_account_hash)
+      end
+    end
+
+    def update
+      if merchant_account_exists_in_registry?
+        updates = merchant_account_hash
+        updated_merchant_account = update_existing_merchant_account(updates)
+        response_for_updated_merchant_account(updated_merchant_account)
+      else
+        response_for_merchant_account_not_found
+      end
+    end
+
+    def delete
+      delete_merchant_account_with_id(merchant_account_id)
+      deletion_response
+    end
+
+    private
+
+    def invalid?
+      false
+    end
+
+    def create_merchant_account_with(hash)
+      FakeBraintree.registry.merchant_accounts[hash['id'].to_s] = hash
+    end
+
+    def update_existing_merchant_account(updates_hash)
+      merchant_account_from_registry.merge!(updates_hash)
+    end
+
+    def merchant_account_hash
+      @merchant_account_hash
+    end
+
+    def merchant_account_from_registry
+      FakeBraintree.registry.merchant_accounts[merchant_account_id]
+    end
+
+    def merchant_account_exists_in_registry?
+      FakeBraintree.registry.merchant_accounts.key?(merchant_account_id)
+    end
+
+    def delete_merchant_account_with_id(id)
+      FakeBraintree.registry.merchant_accounts[id] = nil
+    end
+
+    def deletion_response
+      gzipped_response(200, '')
+    end
+
+    def response_for_created_merchant_account(hash)
+      gzipped_response(201, hash.to_xml(root: 'merchant_account'))
+    end
+
+    def response_for_updated_merchant_account(hash)
+      gzipped_response(200, hash.to_xml(root: 'merchant_account'))
+    end
+
+    def response_for_merchant_account_not_found
+      failure_response(404)
+    end
+
+    def failure_response(code)
+      gzipped_response(code, '')
+    end
+
+    def merchant_account_id
+      merchant_account_hash['id']
+    end
+
+    def set_merchant_account_id
+      @merchant_account_hash['id'] ||= create_id(@merchant_account_hash['merchant_id'])
+    end
+  end
+end

data/lib/fake_braintree/payment_method.rb

@@ -0,0 +1,11 @@
+require 'active_support/core_ext/hash/keys'
+
+module FakeBraintree
+  class PaymentMethod
+    def self.tokenize_card(attributes)
+      token = (Time.now.to_f * 1000).round.to_s
+      FakeBraintree.registry.payment_methods[token] = attributes.stringify_keys
+      token
+    end
+  end
+end

data/lib/fake_braintree/redirect.rb

@@ -1,3 +1,9 @@
+require 'uri'
+require 'rack/utils'
+require 'fake_braintree/helpers'
+require 'fake_braintree/credit_card'
+require 'fake_braintree/customer'
+
 module FakeBraintree
   class Redirect
     include Helpers

data/lib/fake_braintree/registry.rb

@@ -4,16 +4,18 @@ class FakeBraintree::Registry
   end
 
   attr_accessor :customers,:subscriptions, :failures, :transactions, :redirects,
-    :credit_cards, :addresses
+    :credit_cards, :addresses, :payment_methods, :merchant_accounts
 
   def clear!
-    @addresses     = {}
-    @customers     = {}
-    @subscriptions = {}
-    @failures      = {}
-    @transactions  = {}
-    @redirects     = {}
-    @credit_cards  = {}
+    @addresses       = {}
+    @customers       = {}
+    @subscriptions   = {}
+    @failures        = {}
+    @transactions    = {}
+    @redirects       = {}
+    @credit_cards    = {}
+    @payment_methods = {}
+    @merchant_accounts = {}
   end
 
   def failure?(card_number)

data/lib/fake_braintree/server.rb

@@ -1,7 +1,15 @@
+require 'forwardable'
+require 'capybara'
+require 'fake_braintree/sinatra_app'
+
 class FakeBraintree::Server
-  def boot
-    server = Capybara::Server.new(FakeBraintree::SinatraApp)
-    server.boot
-    ENV['GATEWAY_PORT'] = server.port.to_s
+  SERVER_HOST = '127.0.0.1'
+
+  extend Forwardable
+  def_delegators :@server, :port, :boot
+
+  def initialize(options = {})
+    app = FakeBraintree::SinatraApp
+    @server = Capybara::Server.new(app, options.fetch(:port, nil), SERVER_HOST)
   end
 end

data/lib/fake_braintree/sinatra_app.rb

@@ -1,8 +1,23 @@
+require 'sinatra/base'
+require 'active_support/core_ext/hash/conversions'
+require 'fake_braintree/customer'
+require 'fake_braintree/subscription'
+require 'fake_braintree/redirect'
+require 'fake_braintree/credit_card'
+require 'fake_braintree/address'
+require 'fake_braintree/payment_method'
+require 'fake_braintree/transaction'
+require 'fake_braintree/client_token'
+require 'fake_braintree/credit_card_serializer'
+require 'fake_braintree/merchant_account'
+
 module FakeBraintree
   class SinatraApp < Sinatra::Base
     set :show_exceptions, false
     set :dump_errors, true
     set :raise_errors, true
+    set :public_folder, File.dirname(__FILE__) + '/braintree_assets'
+    set :protection, except: :frame_options
     disable :logging
 
     include Helpers
@@ -18,6 +33,52 @@ module FakeBraintree
       end
     end
 
+    # braintree.api.Client.prototype.tokenizeCard()
+    get '/merchants/:merchant_id/client_api/v1/payment_methods/credit_cards' do
+      request_hash = params
+
+      callback = request_hash.delete('callback')
+      nonce = FakeBraintree::PaymentMethod.tokenize_card(request_hash['creditCard'])
+
+      headers = {
+        'Content-Encoding' => 'gzip',
+        'Content-Type' => 'application/javascript; charset=utf-8'
+      }
+      json = {
+        creditCards: [nonce: nonce],
+        status: 201
+      }.to_json
+      response = "#{callback}(#{json})"
+      [200, headers, gzip(response)]
+    end
+
+    # braintree.api.Client.prototype.getCreditCards()
+    get '/merchants/:merchant_id/client_api/v1/payment_methods' do
+      request_hash = params
+
+      callback = request_hash.delete('callback')
+      customer_id = request_hash['authorizationFingerprint']
+      begin
+        customer = Braintree::Customer.find(customer_id)
+        credit_cards = customer.credit_cards.collect do |card|
+          FakeBraintree::CreditCardSerializer.new(card).to_h
+        end
+      rescue Braintree::NotFoundError, ArgumentError
+        credit_cards = []
+      end
+
+      headers = {
+        'Content-Encoding' => 'gzip',
+        'Content-Type' => 'application/javascript; charset=utf-8'
+      }
+      json = {
+        paymentMethods: credit_cards,
+        status: 200
+      }.to_json
+      response = "#{callback}(#{json})"
+      [200, headers, gzip(response)]
+    end
+
     # Braintree::Customer.create
     post '/merchants/:merchant_id/customers' do
       customer_hash = hash_from_request_body_with_key('customer')
@@ -82,9 +143,27 @@ module FakeBraintree
 
     # Braintree::Subscription.cancel
     put '/merchants/:merchant_id/subscriptions/:id/cancel' do
-      updates = {'status' => Braintree::Subscription::Status::Canceled}
       options = {id: params[:id], merchant_id: params[:merchant_id]}
-      Subscription.new(updates, options).update
+      Subscription.new({}, options).cancel
+    end
+
+    # Braintree::PaymentMethod.find
+    get '/merchants/:merchant_id/payment_methods/any/:token' do
+      credit_card = FakeBraintree.registry.credit_cards[params[:token]]
+      if credit_card
+        gzipped_response(200, credit_card.to_xml(root: 'credit_card'))
+      else
+        gzipped_response(404, {})
+      end
+    end
+
+    # Braintree::PaymentMethod.update
+    put '/merchants/:merchant_id/payment_methods/any/:token' do
+      credit_card = FakeBraintree.registry.credit_cards[params[:token]]
+      updates     = hash_from_request_body_with_key('payment_method')
+      options     = {token: params[:token], merchant_id: params[:merchant_id]}
+
+      CreditCard.new(updates, options).update
     end
 
     # Braintree::CreditCard.find
@@ -106,9 +185,28 @@ module FakeBraintree
       CreditCard.new(updates, options).update
     end
 
+    # Braintree::CreditCard.delete
+    delete '/merchants/:merchant_id/payment_methods/credit_card/:credit_card_token' do
+      cc_hash     = {}
+      options     = {token: params[:credit_card_token], merchant_id: params[:merchant_id]}
+
+      CreditCard.new(cc_hash, options).delete
+    end
+
+    # Braintree::PaymentMethod.create
     # Braintree::CreditCard.create
     post '/merchants/:merchant_id/payment_methods' do
-      credit_card_hash = hash_from_request_body_with_key('credit_card')
+      request_hash = Hash.from_xml(request.body)
+      request.body.rewind
+
+      credit_card_hash =
+        if request_hash.key?('credit_card')
+          hash_from_request_body_with_key('credit_card')
+        else
+          payment_method_hash = hash_from_request_body_with_key('payment_method')
+          nonce = payment_method_hash.delete('payment_method_nonce')
+          FakeBraintree.registry.payment_methods[nonce].merge(payment_method_hash)
+        end
       options = {merchant_id: params[:merchant_id]}
 
       if credit_card_hash['options']
@@ -122,18 +220,13 @@ module FakeBraintree
     # Braintree::CreditCard.sale
     post '/merchants/:merchant_id/transactions' do
       if FakeBraintree.decline_all_cards?
-        gzipped_response(422, FakeBraintree.create_failure.to_xml(root: 'api_error_response'))
+        gzipped_response(422, FakeBraintree.create_failure.to_xml(root: "api_error_response"))
       else
-        transaction = hash_from_request_body_with_key('transaction')
+        data = hash_from_request_body_with_key("transaction")
         transaction_id = md5("#{params[:merchant_id]}#{Time.now.to_f}")
-        options = transaction["options"] || {}
-        status = "authorized"
-        if options.fetch("submit_for_settlement", false) == true
-          status = "submitted_for_settlement"
-        end
-        transaction_response = {'id' => transaction_id, 'amount' => transaction['amount'], 'status' => status, 'type' => 'sale'}
-        FakeBraintree.registry.transactions[transaction_id] = transaction_response
-        gzipped_response(200, transaction_response.to_xml(root: 'transaction'))
+        transaction = FakeBraintree::Transaction.new(data, transaction_id)
+        response = transaction.create
+        gzipped_response(200, response.to_xml(root: "transaction"))
       end
     end
 
@@ -150,7 +243,7 @@ module FakeBraintree
     # Braintree::Transaction.refund
     post '/merchants/:merchant_id/transactions/:transaction_id/refund' do
       transaction          = hash_from_request_body_with_key('transaction')
-      transaction_id       = md5('#{params[:merchant_id]}#{Time.now.to_f}')
+      transaction_id       = md5("#{params[:merchant_id]}#{Time.now.to_f}")
       transaction_response = {'id' => transaction_id, 'amount' => transaction['amount'], 'type' => 'credit'}
       FakeBraintree.registry.transactions[transaction_id] = transaction_response
       gzipped_response(200, transaction_response.to_xml(root: 'transaction'))
@@ -194,5 +287,46 @@ module FakeBraintree
       redirect = FakeBraintree.registry.redirects[params[:id]]
       redirect.confirm
     end
+
+    # Braintree::ClientToken.generate
+    post '/merchants/:merchant_id/client_token' do
+      client_token_hash = hash_from_request_body_with_key('client_token')
+      token = FakeBraintree::ClientToken.generate(client_token_hash)
+      response = { value: token }.to_xml(root: :client_token)
+      gzipped_response(200, response)
+    end
+
+    get '/config' do
+      headers = {
+        'Content-Encoding' => 'gzip',
+        'Content-Type' => 'application/javascript; charset=utf-8'
+      }
+      response = "#{params['callback']}(#{{ status: 200 }.to_json})"
+      [200, headers, gzip(response)]
+    end
+
+    #Braintree::MerchantAccount.find
+    get '/merchants/:merchant_id/merchant_accounts/:merchant_account_id' do
+      merchant_account = FakeBraintree.registry.merchant_accounts[params[:merchant_account_id]]
+      if merchant_account
+        gzipped_response(200, merchant_account.to_xml(root: 'merchant_account'))
+      else
+        gzipped_response(404, {})
+      end
+    end
+
+    # Braintree::MerchantAccount.update
+    put '/merchants/:merchant_id/merchant_accounts/:merchant_account_id/update_via_api' do
+      merchant_account_hash = hash_from_request_body_with_key('merchant_account')
+      options = {id: params[:merchant_account_id], merchant_id: params[:merchant_id]}
+      MerchantAccount.new(merchant_account_hash, options).update
+    end
+
+    # Braintree::MerchantAccount.create
+    post '/merchants/:merchant_id/merchant_accounts/create_via_api' do
+      merchant_account_hash = hash_from_request_body_with_key('merchant_account')
+      options = {merchant_id: params[:merchant_id]}
+      MerchantAccount.new(merchant_account_hash, options).create
+    end
   end
 end

data/lib/fake_braintree/subscription.rb

@@ -1,3 +1,5 @@
+require 'fake_braintree/helpers'
+
 module FakeBraintree
   class Subscription
     include Helpers
@@ -14,6 +16,10 @@ module FakeBraintree
 
     def create
       create_subscription_with(subscription_hash)
+      if credit_card = FakeBraintree.registry.credit_cards[payment_method_token]
+        credit_card['subscriptions'] ||= []
+        credit_card['subscriptions'] << subscription_hash
+      end
       response_for_created_subscription(subscription_hash)
     end
 
@@ -26,13 +32,22 @@ module FakeBraintree
       end
     end
 
+    def cancel
+      if subscription_exists_in_registry?
+        canceled_subscription = update_existing_subscription('status' => canceled_status)
+        response_for_canceled_subscription(canceled_subscription)
+      else
+        response_for_subscription_not_found
+      end
+    end
+
     private
 
     def subscription_hash
       @subscription_hash.merge(
         'transactions' => [],
-        'add_ons' => added_add_ons,
-        'discounts' => added_discounts,
+        'add_ons' => add_ons,
+        'discounts' => discounts,
         'next_billing_date' => braintree_formatted_date(next_billing_date),
         'billing_day_of_month' => billing_day_of_month,
         'billing_period_start_date' => braintree_formatted_date(billing_period_start_date),
@@ -61,17 +76,33 @@ module FakeBraintree
       date.strftime('%Y-%m-%d')
     end
 
-    def added_add_ons
-      if @subscription_hash['add_ons'].is_a?(Hash) && @subscription_hash['add_ons']['add']
-        @subscription_hash['add_ons']['add'].map { |add_on| { 'id' => add_on['inherited_from_id'] } }
-      else
-        []
-      end
+    def add_ons
+      discounts_or_add_ons(@subscription_hash['add_ons'])
     end
 
-    def added_discounts
-      if @subscription_hash['discounts'].is_a?(Hash) && @subscription_hash['discounts']['add']
-        @subscription_hash['discounts']['add'].map { |discount| { 'id' => discount['inherited_from_id'], 'amount' => discount['amount'] } }
+    def discounts
+      discounts_or_add_ons(@subscription_hash['discounts'])
+    end
+
+    def discounts_or_add_ons(discount_or_add_on)
+      return [] unless discount_or_add_on.is_a?(Hash)
+
+      if discount_or_add_on['add']
+        discount_or_add_on['add'].map do |hsh|
+          {
+            'id'       => hsh['inherited_from_id'],
+            'quantity' => hsh['quantity'],
+            'amount'   => hsh['amount']
+          }
+        end
+      elsif discount_or_add_on['update']
+        discount_or_add_on['update'].map do |hsh|
+          {
+            'id'       => hsh['existing_id'],
+            'quantity' => hsh['quantity'],
+            'amount'   => hsh['amount']
+          }
+        end
       else
         []
       end
@@ -117,6 +148,10 @@ module FakeBraintree
       Braintree::Subscription::Status::Active
     end
 
+    def canceled_status
+      Braintree::Subscription::Status::Canceled
+    end
+
     def response_for_created_subscription(hash)
       gzipped_response(201, hash.to_xml(root: 'subscription'))
     end
@@ -125,8 +160,8 @@ module FakeBraintree
       gzipped_response(404, {})
     end
 
-    def response_for_created_subscription(hash)
-      gzipped_response(201, hash.to_xml(root: 'subscription'))
+    def response_for_canceled_subscription(hash)
+      gzipped_response(200, hash.to_xml(root: 'subscription'))
     end
   end
 end