Add make_default method on AddPaymentSourcesToWallet class

Move the logic to make the last payment source as default outside the 
add_to_wallet method to take advantage if someone wants to change only 
the make default behavior.

core/app/models/spree/wallet/add_payment_sources_to_wallet.rb

@@ -24,16 +24,19 @@ def add_to_wallet
       # add valid sources to wallet and optionally set a default
       if sources.any?
         # arbitrarily sort by id for picking a default
-        wallet_payment_sources = sources.sort_by(&:id).map do |source|
+        sources.sort_by(&:id).each do |source|
           order.user.wallet.add(source)
         end
 
-        order.user.wallet.default_wallet_payment_source =
-          wallet_payment_sources.last
+        make_default
       end
     end
   end
 
+  def make_default
+    order.user.wallet.default_wallet_payment_source = order.user.wallet_payment_sources.last
+  end
+
   private
 
   attr_reader :order

core/spec/models/spree/wallet/add_payment_sources_to_wallet_spec.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+require 'rails_helper'
+
+RSpec.describe Spree::Wallet::AddPaymentSourcesToWallet, type: :model do
+  let(:order) { create(:order_ready_to_complete) }
+
+  describe '#add_to_wallet' do
+    subject { described_class.new(order) }
+
+    it 'saves the payment source and calls make_default' do
+      expect(subject).to receive(:make_default)
+      subject.add_to_wallet
+      expect(order.user.wallet.wallet_payment_sources.count).to eq(1)
+    end
+  end
+
+  describe '#make_default' do
+    subject(:call) { -> { described_class.new(order).make_default } }
+
+    let(:payment) { create(:payment) }
+
+    before do
+      order.user.wallet.add(payment.source)
+    end
+
+    it 'makes the user payment source as default' do
+      subject.call
+      expect(order.user.wallet.wallet_payment_sources.any?(&:default)).to be_truthy
+    end
+  end
+end