add Rental#charge method. Only public method necessary to calculate w…

…hat to charge for a rental
stride-nyc · Jul 30, 2018 · e841b02 · medwards1771 · Jul 31, 2018 · e841b02
1 parent 5f27122
commit e841b02
Showing 1 changed file with 7 additions and 12 deletions.
diff --git a/a_first_example/first_example.rb b/a_first_example/first_example.rb
@@ -20,6 +20,12 @@ def initialize(movie, days_rented)
     @movie, @days_rented = movie, days_rented
   end
 
+  def charge
+    base_charge + extra_charge
+  end
+
+  private
+
   def base_charge
     case movie.price_code
     when Movie::REGULAR
@@ -69,7 +75,7 @@ def statement
     @rentals.each do |rental|
       this_amount = 0
 
-      this_amount += charge_for_rental(rental)
+      this_amount += rental.charge
 
       frequent_renter_points += points_for_rental(rental)
 
@@ -83,17 +89,6 @@ def statement
     result
   end
 
-  def charge_for_rental(rental)
-    case rental.movie.price_code
-    when Movie::REGULAR
-      rental.base_charge + rental.extra_charge
-    when Movie::NEW_RELEASE
-      rental.base_charge + rental.extra_charge
-    when Movie::CHILDRENS
-      rental.base_charge + rental.extra_charge
-    end
-  end
-
   def points_for_rental(rental)
     case rental.movie.price_code
     when Movie::CHILDRENS, Movie::REGULAR