X-Git-Url: https://irasnyder.com/gitweb/?a=blobdiff_plain;f=app%2Fmodels%2Fcoitem.rb;h=b9d902481d62d2ec435820d6fe883e31f7e4c3b6;hb=b12881a928516bca02b2bd8ce1bbe9288840c76f;hp=8513465559ee026914dbbae114b9d9d81bbbdcd2;hpb=0dccca2d3c0e9d5e6eaa1c1ebee3670654fd8c43;p=cs356-p2-videostore.git

diff --git a/app/models/coitem.rb b/app/models/coitem.rb
index 8513465..b9d9024 100644
--- a/app/models/coitem.rb
+++ b/app/models/coitem.rb
@@ -17,6 +17,15 @@ class Coitem < ActiveRecord::Base
   validates_associated :customer
   validates_associated :rentable
 
+  def overdue?
+    return due_date < Time.now.to_date
+  end
+
+  def late_fee
+    days_late = Time.now.to_date - (due_date)
+    return VideoPolicy.find_by_day(8).fee * days_late.to_i
+  end
+
   protected
   def validate
     errors.add(:customer_id, "does not exist is the database") if customer.nil?