Add VideoPolicy to the purchase path

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

diff --git a/app/models/coitem.rb b/app/models/coitem.rb
index 53bf312..b9d9024 100644 (file)
--- a/app/models/coitem.rb
+++ b/app/models/coitem.rb
@@ -22,9 +22,8 @@ class Coitem < ActiveRecord::Base
   end
 
   def late_fee
-    # FIXME: this should be calculated better
     days_late = Time.now.to_date - (due_date)
-    return 3 * days_late.to_i
+    return VideoPolicy.find_by_day(8).fee * days_late.to_i
   end
 
   protected