spree_gateway 3.7.1 → 3.8.0

data/lib/views/backend/spree/admin/payments/source_forms/_stripe_elements.html.erb

@@ -0,0 +1,79 @@
+<%= render "spree/admin/payments/source_forms/gateway", payment_method: payment_method, previous_cards: payment_method.reusable_sources(@order) %>
+
+<script type="text/javascript" src="https://js.stripe.com/v2/"></script>
+<script type="text/javascript">
+  Stripe.setPublishableKey("<%= payment_method.preferred_publishable_key %>");
+</script>
+
+<script>
+  var mapCC, stripeElementResponseHandler;
+
+  mapCC = function(ccType) {
+    if (ccType === 'MasterCard') {
+      return 'mastercard';
+    } else if (ccType === 'Visa') {
+      return 'visa';
+    } else if (ccType === 'American Express') {
+      return 'amex';
+    } else if (ccType === 'Discover') {
+      return 'discover';
+    } else if (ccType === 'Diners Club') {
+      return 'dinersclub';
+    } else if (ccType === 'JCB') {
+      return 'jcb';
+    }
+  };
+  stripeElementResponseHandler = function(status, response) {
+    var paymentMethodId, token;
+    if (response.error) {
+      $('#stripeError').html(response.error.message);
+      var param_map = {
+        number:    '#card_number',
+        exp_month: '#card_expiry',
+        exp_year:  '#card_expiry',
+        cvc:       '#card_code'
+      }
+      if (response.error.param) {
+        errorField = Spree.stripeElementsPaymentMethod.find(param_map[response.error.param])
+        errorField.addClass('error');
+        errorField.parent().addClass('has-error');
+      }
+      return $('#stripeError').show();
+    } else {
+      Spree.stripeElementsPaymentMethod.find('#card_number<%= payment_method.id %>, #card_expiry<%= payment_method.id %>, #card_code<%= payment_method.id %>').prop("disabled", true);
+      Spree.stripeElementsPaymentMethod.find(".ccType").prop("disabled", false);
+      Spree.stripeElementsPaymentMethod.find(".ccType").val(mapCC(response.card.brand));
+      token = response['id'];
+      paymentMethodId = Spree.stripeElementsPaymentMethod.prop('id').split("_")[2];
+      Spree.stripeElementsPaymentMethod.append("<input type='hidden' class='stripeToken' name='payment_source[" + paymentMethodId + "][gateway_payment_profile_id]' value='" + token + "'/>");
+      Spree.stripeElementsPaymentMethod.append("<input type='hidden' class='stripeToken' name='payment_source[" + paymentMethodId + "][last_digits]' value='" + response.card.last4 + "'/>");
+      Spree.stripeElementsPaymentMethod.append("<input type='hidden' class='stripeToken' name='payment_source[" + paymentMethodId + "][month]' value='" + response.card.exp_month + "'/>");
+      Spree.stripeElementsPaymentMethod.append("<input type='hidden' class='stripeToken' name='payment_source[" + paymentMethodId + "][year]' value='" + response.card.exp_year + "'/>");
+      return Spree.stripeElementsPaymentMethod.parents("form").trigger('submit');
+    }
+  };
+
+  window.addEventListener('DOMContentLoaded', function() {
+    Spree.stripeElementsPaymentMethod = $('#payment_method_' + <%= payment_method.id %>);
+
+    Spree.ready(function() {
+      Spree.stripeElementsPaymentMethod.prepend("<div id='stripeError' class='errorExplanation alert alert-danger' style='display:none'></div>");
+      return $('#new_payment [data-hook=buttons]').click(function() {
+        var expiration, params;
+        $('#stripeError').hide();
+        Spree.stripeElementsPaymentMethod.find('#card_number, #card_expiry, #card_code').removeClass('error');
+        if (Spree.stripeElementsPaymentMethod.is(':visible')) {
+          expiration = $('.cardExpiry:visible').payment('cardExpiryVal');
+          params = $.extend({
+            number: $('.cardNumber:visible').val(),
+            cvc: $('.cardCode:visible').val(),
+            exp_month: expiration.month || 0,
+            exp_year: expiration.year || 0
+          }, Spree.stripeAdditionalInfo);
+          Stripe.card.createToken(params, stripeElementResponseHandler);
+          return false;
+        }
+      });
+    });
+  });
+</script>

data/lib/views/backend/spree/admin/payments/source_views/_stripe_ach.html.erb

@@ -0,0 +1,21 @@
+<fieldset data-hook="bank_transfer">
+  <legend><%= Spree.t(:bank_transfer) %></legend>
+  <table class="table table-condensed table-bordered">
+    <tr>
+      <th width="20%"><%= Spree.t('stripe.ach.account_holder_name') %>:</th>
+      <td><%= payment.source.account_holder_name %></td>
+    </tr>
+    <tr>
+      <th><%= Spree.t('stripe.ach.account_holder_type') %>:</th>
+      <td><%= payment.source.account_holder_type %></td>
+    </tr>
+    <tr>
+      <th><%= Spree.t('stripe.ach.routing_number') %>:</th>
+      <td><%= payment.source.routing_number %></td>
+    </tr>
+    <tr>
+      <th><%= Spree.t('stripe.ach.account_number') %>:</th>
+      <td><%= payment.source.account_number %></td>
+    </tr>
+  </table>
+</fieldset>

data/lib/views/backend/spree/admin/payments/source_views/_stripe_apple_pay.html.erb

@@ -0,0 +1 @@
+<%= render "spree/admin/payments/source_views/gateway", payment: payment %>

data/lib/views/backend/spree/admin/payments/source_views/_stripe_elements.html.erb

@@ -0,0 +1 @@
+<%= render "spree/admin/payments/source_views/gateway", payment: payment %>

data/lib/views/frontend/spree/checkout/payment/_stripe_ach.html.erb

@@ -0,0 +1,81 @@
+<div class="payment-gateway">
+  <% param_prefix = "payment_source[#{payment_method.id}]" %>
+  <div class="payment-gateway-half-fields">
+    <div class="mb-4 payment-gateway-field checkout-content-inner-field">
+      <%= text_field_tag "#{param_prefix}[account_holder_name]", "#{@order.bill_address_firstname} #{@order.bill_address_lastname}", { id: "account_holder_name", class: 'spree-flat-input', placeholder: Spree.t('stripe.ach.account_holder_name')} %>
+    </div>
+    <div class="mb-4 payment-gateway-field checkout-content-inner-field">
+      <%= select_tag "#{param_prefix}[account_holder_type]", options_for_select(%w(Individual Company)), {id: "account_holder_type", class: 'spree-flat-input', placeholder: Spree.t('stripe.ach.account_holder_type')} %>
+    </div>
+
+    <div class="mb-4 payment-gateway-field checkout-content-inner-field" data-hook="account_number">
+      <%= text_field_tag "#{param_prefix}[routing_number]", '', {id: 'routing_number', class: 'spree-flat-input', autocomplete: "off", placeholder: Spree.t('stripe.ach.routing_number'), maxlength: 9} %>
+    </div>
+    <div class="mb-4 payment-gateway-field checkout-content-inner-field" data-hook="account_number">
+      <%= text_field_tag "#{param_prefix}[account_number]", '', {id: 'account_number', class: 'spree-flat-input', autocomplete: "off", placeholder: Spree.t('stripe.ach.account_number')} %>
+    </div>
+    <div class="mb-4 payment-gateway-field checkout-content-inner-field" data-hook="account_number">
+      <%= text_field_tag "#{param_prefix}[verify_account_number]", '', {id: 'verify_account_number', class: 'spree-flat-input', autocomplete: "off", placeholder: Spree.t('stripe.ach.verify_account_number')} %>
+    </div>
+  </div>
+</div>
+
+<script type="text/javascript" src="https://js.stripe.com/v3/"></script>
+<script type="text/javascript">
+    var stripe = Stripe("<%= payment_method.preferred_publishable_key %>");
+</script>
+
+<script>
+    stripeResponseHandler = function(response) {
+        var paymentMethodId, token, bank_acc_status;
+        if (response.error) {
+            $('#stripeError').html(response.error.message);
+            var param_map = {
+                account_holder_name: '#account_holder_name',
+                account_holder_type: '#account_holder_type',
+                account_number: '#account_number',
+                routing_number: '#routing_number'
+            }
+            if (response.error.param){
+                errorField = Spree.stripePaymentMethod.find(param_map[response.error.param])
+                errorField.addClass('error');
+                errorField.parent().addClass('has-error');
+            }
+            return $('#stripeError').show();
+        } else {
+            token = response.token['id'];
+            bank_acc_status = response.token.bank_account['status'];
+            paymentMethodId = Spree.stripePaymentMethod.prop('id').split("_")[2];
+            Spree.stripePaymentMethod.append("<input type='hidden' class='stripeToken' name='payment_source[" + paymentMethodId + "][gateway_payment_profile_id]' value='" + token + "'/>");
+            Spree.stripePaymentMethod.append("<input type='hidden' class='stripeStatus' name='payment_source[" + paymentMethodId + "][status]' value='" + bank_acc_status + "'/>")
+            return Spree.stripePaymentMethod.parents("form").trigger('submit');
+        }
+    };
+
+    window.addEventListener('DOMContentLoaded', function () {
+        Spree.stripePaymentMethod = $('#payment_method_' + <%= payment_method.id %>);
+        Spree.stripePaymentMethod.prepend("<div id='stripeError' class='errorExplanation alert alert-danger' style='display:none'></div>");
+
+        $('#checkout_form_payment [data-hook=buttons]').click(function (e) {
+            var params;
+            $('#stripeError').hide();
+            Spree.stripePaymentMethod.find('#account_holder_name, #account_holder_type, #account_number, #routing_number').removeClass('error');
+            if (Spree.stripePaymentMethod.is(':visible')) {
+                e.preventDefault();
+
+                params = $.extend({
+                    country: 'US',
+                    currency: 'usd',
+                    account_holder_name: $('#account_holder_name:visible').val(),
+                    account_holder_type: $('#account_holder_type:visible').val(),
+                    routing_number: $('#routing_number:visible').val(),
+                    account_number: $('#account_number:visible').val()
+                }, Spree.stripeAdditionalInfo);
+
+                stripe.createToken('bank_account', params).then(stripeResponseHandler);
+
+                return false;
+            }
+        });
+    });
+</script>

data/lib/views/frontend/spree/checkout/payment/_stripe_ach_verify.html.erb

@@ -0,0 +1,16 @@
+<p class="payment-type checkout-content-header">
+  <%= Spree.t('stripe.ach.verify_bank_account').upcase %>
+</p>
+
+<div class="payment-gateway">
+  <div class="payment-gateway-half-fields">
+    <input type="hidden" name="order[payment_id]" value= <%= payment.id %> />
+    <input type='hidden' name='verifyAch' value=true />
+    <div class="mb-4 payment-gateway-field checkout-content-inner-field">
+      <%= number_field_tag "amounts[]", nil, { id: "deposit1", class: 'spree-flat-input', placeholder: Spree.t('stripe.ach.first_deposit') } %>
+    </div>
+    <div class="mb-4 payment-gateway-field checkout-content-inner-field">
+      <%= number_field_tag "amounts[]", nil, { id: "deposit2", class: 'spree-flat-input', placeholder: Spree.t('stripe.ach.second_deposit') } %>
+    </div>
+  </div>
+</div>

data/lib/views/frontend/spree/checkout/payment/_stripe_apple_pay.html.erb

@@ -9,7 +9,6 @@
 <script>
   var stripeApplePay = Stripe("<%= payment_method.preferred_publishable_key %>");
   var elements = stripeApplePay.elements();
-
   var paymentRequest = stripeApplePay.paymentRequest({
     country: '<%= payment_method.preferred_country_code.try(:upcase) %>',
     currency: '<%= @order.currency.downcase %>',
@@ -64,6 +63,15 @@
     form.appendChild(hiddenInput);
   };
 
+  function hideApplePayRadioButtonForNonAppleDevices() {
+    var isSafari = !!navigator.userAgent.match(/Version\/[\d\.]+.*Safari/);
+    var isiOS = /iPad|iPhone|iPod/.test(navigator.userAgent) && !window.MSStream;
+
+    if (!isSafari && !isiOS) {
+      $('input[name$="order[payments_attributes][][payment_method_id]"][value="<%= payment_method.id %>"]').closest('li').hide()
+    }
+  }
+
   paymentRequest.on('token', function(ev) {
     var form = document.getElementById('checkout_form_payment');
     var token = ev.token;
@@ -83,6 +91,7 @@
 
   window.addEventListener('DOMContentLoaded', function() {
     Spree.stripeApplePayPaymentMethod = $('#payment_method_' + <%= payment_method.id %>);
+    hideApplePayRadioButtonForNonAppleDevices();
 
     prButton.on('ready', function () {
       Spree.stripeApplePayPaymentMethod.prepend("<div id='stripeApplePayError' class='errorExplanation alert alert-danger' style='display:none'></div>");

data/lib/views/frontend/spree/checkout/payment/_stripe_elements.html.erb

@@ -1,24 +1,15 @@
 <% param_prefix = "payment_source[#{payment_method.id}]" %>
 <div class="well clearfix">
   <p class="field">
-    <%= label_tag "name_on_card_#{payment_method.id}" do %>
-      <%= Spree.t(:name_on_card) %><abbr class="required" title="required">*</abbr>
-    <% end %>
-    <%= text_field_tag "#{param_prefix}[name]", "#{@order.bill_address_firstname} #{@order.bill_address_lastname}", { id: "name_on_card_#{payment_method.id}", class: 'form-control required'} %>
-    </p>
-    <div class="form-row">
-      <%= label_tag "card_number" do %>
-        <%= Spree.t(:card_number) %><abbr class="required" title="required">*</abbr>
-      <% end %>
-      <div class="form-control required cardNumber">
-        <div id="card-element">
-          <!-- a Stripe Element will be inserted here. -->
-        </div>
-      </div>
-      <!-- Used to display form errors -->
-      <div id="card-errors" role="alert"></div>
-    </div>
+    <%= text_field_tag "#{param_prefix}[name]", "#{@order.bill_address_firstname} #{@order.bill_address_lastname}", { id: "name_on_card_#{payment_method.id}", class: 'spree-flat-input', placeholder: Spree.t(:name_on_card)} %>
   </p>
+  <div class="form-control required cardNumber spree-flat-input">
+    <div id="card-element">
+      <!-- a Stripe Element will be inserted here. -->
+    </div>
+  </div>
+  <!-- Used to display form errors -->
+  <div id="card-errors" role="alert"></div>
 </div>
 
 <script type="text/javascript" src="https://js.stripe.com/v3/"></script>
@@ -26,20 +17,24 @@
 <script>
   var stripeElements = Stripe("<%= payment_method.preferred_publishable_key %>");
   var elements = stripeElements.elements();
+  var spreeFlatInputStyle = getComputedStyle(document.querySelector('input.spree-flat-input'));
 
   var card = elements.create('card', {
     iconStyle: 'solid',
     hidePostalCode: true,
     style: {
       base: {
-        iconColor: '#555555',
-        lineHeight: '24px',
-        fontWeight: 300,
-        fontFamily: '"Helvetica Neue", Helvetica, sans-serif',
-        fontSize: '14px',
+        color: spreeFlatInputStyle['color'],
+        iconColor: spreeFlatInputStyle['color'],
+        lineHeight: spreeFlatInputStyle['line-height'],
+        fontWeight: spreeFlatInputStyle['font-weight'],
+        fontFamily: spreeFlatInputStyle['font-family'],
+        fontSize: spreeFlatInputStyle['font-size'],
 
         '::placeholder': {
-          color: '#555555',
+          color: '#757575',
+          fontWeight: 300,
+          textTransform: 'uppercase'
         },
       },
       invalid: {

data/spec/factories/check_factory.rb

@@ -0,0 +1,10 @@
+FactoryBot.define do
+  factory :check, class: Spree::Check do
+    account_holder_name { 'John Doe' }
+    account_holder_type { 'Individual' }
+    account_type { 'checking' }
+    routing_number { '110000000' }
+    account_number { '000123456789' }
+    association(:payment_method, factory: :credit_card_payment_method)
+  end
+end

data/spec/features/admin/stripe_elements_payment_spec.rb

@@ -0,0 +1,109 @@
+require 'spec_helper'
+
+describe 'Admin Panel Stripe elements payment', type: :feature, :js => true do
+  stub_authorization!
+
+  let!(:country)         { create(:country, :states_required => true) }
+  let!(:state)           { create(:state, :country => country) }
+  let!(:shipping_method) { create(:shipping_method) }
+  let!(:stock_location)  { create(:stock_location) }
+  let!(:mug)             { create(:product, :name => 'RoR Mug') }
+  let!(:zone)            { create(:zone) }
+  let!(:stripe_elements_payment_method) do
+    Spree::Gateway::StripeElementsGateway.create!(
+      :name => 'Stripe Element',
+      :preferred_secret_key => 'sk_test_VCZnDv3GLU15TRvn8i2EsaAN',
+      :preferred_publishable_key => 'pk_test_Cuf0PNtiAkkMpTVC2gwYDMIg'
+    )
+  end
+  
+  let!(:order) { OrderWalkthrough.up_to(:payment) }
+  before { visit spree.new_admin_order_payment_path(order.number) }
+
+  it 'can process a valid payment' do
+    fill_in_stripe_payment
+    wait_for { !page.has_current_path?(spree.admin_order_payments_path(order.number)) }
+    
+    expect(page.body).to have_content('Payment has been successfully created!')
+    expect(page).to have_current_path spree.admin_order_payments_path(order.number)
+  end
+  
+  if Spree.version.to_f >= 4.1
+    it 'shows an error with an invalid card name' do
+      fill_in_stripe_payment(true)
+
+      expect(page).to have_content("Credit card Name can't be blank")
+      expect(page).to have_current_path spree.admin_order_payments_path(order.number)
+    end
+  else
+    it 'can proces valid payment with invalid card name' do
+      fill_in_stripe_payment(true)
+      wait_for { !page.has_current_path?(spree.admin_order_payments_path(order.number)) }
+
+      expect(page.body).to have_content('Payment has been successfully created!')
+      expect(page).to have_current_path spree.admin_order_payments_path(order.number)
+    end
+  end
+
+  it 'shows an error with an invalid card number' do
+    fill_in_stripe_payment(false, true)
+    
+    expect(page).to have_content('The card number is not a valid credit card number.')
+    expect(page).to have_current_path spree.new_admin_order_payment_path(order.number)
+  end
+
+  it 'shows an error with an invalid card code' do
+    fill_in_stripe_payment(false, false, true)
+
+    expect(page).to have_content("Your card's security code is invalid.")
+    expect(page).to have_current_path spree.new_admin_order_payment_path(order.number)
+  end
+
+  it 'shows an error with an invalid card expiration' do
+    fill_in_stripe_payment(false, false, false, true)
+
+    if Spree.version.to_f >= 4.1
+      expect(page).to have_content('Credit card Month is not a number')
+      expect(page).to have_content('Credit card Year is not a number')
+      expect(page).to have_current_path spree.admin_order_payments_path(order.number)
+    else
+      expect(page).to have_content("Your card's expiration year is invalid.")
+      expect(page).to have_current_path spree.new_admin_order_payment_path(order.number)
+    end
+  end
+
+  def fill_in_stripe_payment(invalid_name = false, invalid_number = false, invalid_code = false, invalid_expiration = false)
+    fill_in 'Name *', with: invalid_name ? '' : 'Stripe Elements Gateway Payment'
+    fill_in_card_number(invalid_number)
+    fill_in_cvc(invalid_code)
+    fill_in_card_expiration(invalid_expiration)
+
+    click_button 'Update'
+  end
+
+  def fill_in_card_number(invalid_number)
+    number = invalid_number ? '123' : '4242 4242 4242 4242'
+    fill_in_field('Card Number *', '#card_number1', number)
+  end
+
+  def fill_in_card_expiration(invalid_expiration)
+    valid_expiry = Spree.version.to_f >= 4.2 ? "01/#{Time.current.year + 1}" : "01 / #{Time.current.year + 1}"
+    invalid_expiry = Spree.version.to_f >= 4.2 ? '01/' : '01 / '
+
+    card_expiry = invalid_expiration ? invalid_expiry : valid_expiry
+    fill_in_field('Expiration *', '#card_expiry1', card_expiry)
+  end
+
+  def fill_in_cvc(invalid_code)
+    value = invalid_code ? '1' : '123'
+    label = Spree.version.to_f >= 4.2 ? 'Card Varification Code (CVC) *' : 'Card Code *'
+
+    fill_in label, with: value
+  end
+
+  def fill_in_field(field_name, field_id, number)
+    until page.find(field_id).value == number
+      fill_in field_name, with: number
+    end
+  end
+end

data/spec/features/stripe_checkout_spec.rb

@@ -91,7 +91,10 @@ describe "Stripe checkout", type: :feature do
     expect(page).to have_css('.has-error #card_code.error')
   end
 
+  # this scenario will not occur on Spree 4.2 due to swapping jquery.payment to cleave
+  # see https://github.com/spree/spree/pull/10363
   it "shows an error with invalid expiry month field", :js => true do
+    skip if Spree.version.to_f >= 4.2 
     fill_in 'card_number', with: '4242 4242 4242 4242'
     fill_in 'card_expiry', :with => "00 / #{Time.now.year + 1}"
     fill_in 'card_code', with: '123'

data/spec/models/gateway/stripe_ach_gateway_spec.rb

@@ -0,0 +1,185 @@
+require 'spec_helper'
+
+describe Spree::Gateway::StripeAchGateway do
+  let(:secret_key) { 'key' }
+  let(:email) { 'customer@example.com' }
+  let(:source) { Spree::Check.new }
+  let(:payment) {
+    double('Spree::Payment',
+           source: source,
+           order: double('Spree::Order',
+                         email: email,
+                         bill_address: bill_address,
+                         user: double('Spree::User',
+                                      email: email)
+           )
+    )
+  }
+  let(:provider) do
+    double('provider').tap do |p|
+      p.stub(:purchase)
+      p.stub(:authorize)
+      p.stub(:capture)
+      p.stub(:verify)
+    end
+  end
+
+  before do
+    subject.preferences = { secret_key: secret_key }
+    subject.stub(:options_for_purchase_or_auth).and_return(['money','check','opts'])
+    subject.stub(:provider).and_return provider
+  end
+
+  describe '#create_profile' do
+    before do
+      payment.source.stub(:update!)
+    end
+
+    context 'with an order that has a bill address' do
+      let(:bill_address) {
+        double('Spree::Address',
+               address1: '123 Happy Road',
+               address2: 'Apt 303',
+               city: 'Suzarac',
+               zipcode: '95671',
+               state: double('Spree::State', name: 'Oregon'),
+               country: double('Spree::Country', name: 'United States')
+        )
+      }
+
+      it 'stores the bill address with the provider' do
+        subject.provider.should_receive(:store).with(payment.source, {
+            email: email,
+            login: secret_key,
+
+            address: {
+                address1: '123 Happy Road',
+                address2: 'Apt 303',
+                city: 'Suzarac',
+                zip: '95671',
+                state: 'Oregon',
+                country: 'United States'
+            }
+        }).and_return double.as_null_object
+
+        subject.create_profile payment
+      end
+    end
+
+    context 'with an order that does not have a bill address' do
+      let(:bill_address) { nil }
+
+      it 'does not store a bill address with the provider' do
+        subject.provider.should_receive(:store).with(payment.source, {
+            email: email,
+            login: secret_key
+        }).and_return double.as_null_object
+
+        subject.create_profile payment
+      end
+
+    end
+
+    context 'with a check represents payment_profile' do
+      let(:source) { Spree::Check.new(gateway_payment_profile_id: 'tok_profileid') }
+      let(:bill_address) { nil }
+
+      it 'stores the profile_id as a check' do
+        subject.provider.should_receive(:store).with(source.gateway_payment_profile_id, anything).and_return double.as_null_object
+
+        subject.create_profile payment
+      end
+    end
+  end
+
+  context 'purchasing' do
+    after do
+      subject.purchase(19.99, 'check', {})
+    end
+
+    it 'send the payment to the provider' do
+      provider.should_receive(:purchase).with('money', 'check', 'opts')
+    end
+  end
+
+  context 'authorizing' do
+    after do
+      subject.authorize(19.99, 'check', {})
+    end
+
+    it 'send the authorization to the provider' do
+      provider.should_receive(:authorize).with('money', 'check', 'opts')
+    end
+  end
+
+  context 'verifying' do
+    after do
+      subject.verify(source, {})
+    end
+
+    it 'send the verify to the provider' do
+      provider.should_receive(:verify).with(source, anything)
+    end
+  end
+
+  context 'capturing' do
+
+    after do
+      subject.capture(1234, 'response_code', {})
+    end
+
+    it 'convert the amount to cents' do
+      provider.should_receive(:capture).with(1234, anything, anything)
+    end
+
+    it 'use the response code as the authorization' do
+      provider.should_receive(:capture).with(anything, 'response_code', anything)
+    end
+  end
+
+  context 'capture with payment class' do
+    let(:gateway) do
+      gateway = described_class.new(active: true)
+      gateway.set_preference :secret_key, secret_key
+      gateway.stub(:options_for_purchase_or_auth).and_return(['money', 'check', 'opts'])
+      gateway.stub(:provider).and_return provider
+      gateway.stub source_required: true
+      gateway
+    end
+
+    let(:order) { Spree::Order.create }
+
+    let(:check) do
+      # mock_model(Spree::Check, :gateway_customer_profile_id => 'cus_abcde',
+      # :imported => false)
+      create :check, gateway_customer_profile_id: 'cus_abcde', imported: false
+    end
+
+    let(:payment) do
+      payment = Spree::Payment.new
+      payment.source = check
+      payment.order = order
+      payment.payment_method = gateway
+      payment.amount = 98.55
+      payment.state = 'pending'
+      payment.response_code = '12345'
+      payment
+    end
+
+    after do
+      payment.capture!
+    end
+
+    let!(:success_response) do
+      double('success_response',
+             success?: true,
+             authorization: '123',
+             avs_result: { 'code' => 'avs-code' },
+             cvv_result: { 'code' => 'cvv-code', 'message' => 'CVV Result' })
+    end
+
+    it 'gets correct amount' do
+      provider.should_receive(:capture).with(9855, '12345', anything).and_return(success_response)
+    end
+  end
+end