Fix the late fee calculations

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

diff --git a/app/models/coitem.rb b/app/models/coitem.rb
index 8513465..5d0a212 100644 (file)
--- a/app/models/coitem.rb
+++ b/app/models/coitem.rb
@@ -17,6 +17,19 @@ 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)
+    if rentable.class == Video
+      return VideoPolicy.find_by_day(7).fee * days_late.to_i
+    else
+      return GamePolicy.find_by_day(7).fee * days_late.to_i
+    end
+  end
+
   protected
   def validate
     errors.add(:customer_id, "does not exist is the database") if customer.nil?