X-Git-Url: https://irasnyder.com/gitweb/?a=blobdiff_plain;f=app%2Fmodels%2Fpurchase.rb;h=26cf518a7458ff2019fbfe3cdbf1075c487d2594;hb=53634274872ef0ba36cbf94e23df4443ea5613d5;hp=ff07e41c2916691cb06073e760e49972ce5c1dca;hpb=ebb2b4d40713d535c06970b2a17aace4458ecf25;p=cs356-p2-videostore.git

diff --git a/app/models/purchase.rb b/app/models/purchase.rb
index ff07e41..26cf518 100644
--- a/app/models/purchase.rb
+++ b/app/models/purchase.rb
@@ -6,6 +6,14 @@ class Purchase < ActiveRecord::Base
   validates_presence_of :price
   validates_numericality_of :price
 
+  def title
+    if type == MerchandisePurchase
+      return merchandise.title
+    else
+      return rentable.title
+    end
+  end
+
   protected
   def validate
     errors.add(:price, "cannot be negative") if price < 0