Add Income tracking to the Purchase System

[cs356-p2-videostore.git] / app / models / purchase.rb

diff --git a/app/models/purchase.rb b/app/models/purchase.rb
index 26cf518..b596aa2 100644 (file)
--- a/app/models/purchase.rb
+++ b/app/models/purchase.rb
@@ -9,8 +9,10 @@ class Purchase < ActiveRecord::Base
   def title
     if type == MerchandisePurchase
       return merchandise.title
-    else
+    elsif type == RentablePurchase
       return rentable.title
+    else
+      return 'Late Fees'
     end
   end