From fc2844a3d5e1fcf97e389c2ef4d76382f8e2b556 Mon Sep 17 00:00:00 2001
From: Rob Harrington <oeoeaio@gmail.com>
Date: Fri, 4 May 2018 11:38:35 +1000
Subject: [PATCH] Add default_card method to user model

---
 app/models/spree/user_decorator.rb |  4 ++++
 spec/models/spree/user_spec.rb     | 28 ++++++++++++++++++++++++++++
 2 files changed, 32 insertions(+)

diff --git a/app/models/spree/user_decorator.rb b/app/models/spree/user_decorator.rb
index 3877428c86..8ae76a11b4 100644
--- a/app/models/spree/user_decorator.rb
+++ b/app/models/spree/user_decorator.rb
@@ -73,6 +73,10 @@ Spree.user_class.class_eval do
     owned_enterprises(:reload).size < enterprise_limit
   end
 
+  def default_card
+    credit_cards.where(is_default: true).first
+  end
+
   private
 
   def limit_owned_enterprises
diff --git a/spec/models/spree/user_spec.rb b/spec/models/spree/user_spec.rb
index 39a1ce530e..3cd4da93af 100644
--- a/spec/models/spree/user_spec.rb
+++ b/spec/models/spree/user_spec.rb
@@ -126,4 +126,32 @@ describe Spree.user_class do
       end
     end
   end
+
+  describe "default_card" do
+    let(:user) { create(:user) }
+
+    context "when the user has no credit cards" do
+      it "returns nil" do
+        expect(user.default_card).to be nil
+      end
+    end
+
+    context "when the user has one credit card" do
+      let!(:card) { create(:credit_card, user: user) }
+
+      it "should be assigned as the default and be returned" do
+        expect(card.reload.is_default).to be true
+        expect(user.default_card.id).to be card.id
+      end
+    end
+
+    context "when the user has more than one card" do
+      let!(:non_default_card) { create(:credit_card, user: user) }
+      let!(:default_card) { create(:credit_card, user: user, is_default: true) }
+
+      it "returns the card which is specified as the default" do
+        expect(user.default_card.id).to be default_card.id
+      end
+    end
+  end
 end