diff --git a/src/main/java/video/domain/Customer.java b/src/main/java/video/domain/Customer.java index d8b13d2..4de5e45 100644 --- a/src/main/java/video/domain/Customer.java +++ b/src/main/java/video/domain/Customer.java @@ -24,28 +24,11 @@ public Rentals getRentals() { } public RentalResult rentalMovies() { - int totalAmount = totalAmount(); + int totalAmount = rentals.totalAmount(); - int frequentRenterPoints = frequentRenterPoints(); + int frequentRenterPoints = rentals.totalFrequentRenterPoints(); return new RentalResult(totalAmount, frequentRenterPoints); } - private int frequentRenterPoints() { - int frequentRenterPoints = 0; - for (Rental each : rentals.list()) { - frequentRenterPoints += each.frequentRenterPoints(); - } - return frequentRenterPoints; - } - - private int totalAmount() { - int totalAmount = 0; - for (Rental each : rentals.list()) { - int thisAmount = each.amount(); - totalAmount += thisAmount; - } - return totalAmount; - } - } \ No newline at end of file diff --git a/src/main/java/video/domain/rental/Rentals.java b/src/main/java/video/domain/rental/Rentals.java index 3e6aca3..548865f 100644 --- a/src/main/java/video/domain/rental/Rentals.java +++ b/src/main/java/video/domain/rental/Rentals.java @@ -13,4 +13,21 @@ public void add(Rental rental) { public List list() { return list; } + + public int totalAmount() { + int totalAmount = 0; + for (Rental each : list()) { + int thisAmount = each.amount(); + totalAmount += thisAmount; + } + return totalAmount; + } + + public int totalFrequentRenterPoints() { + int frequentRenterPoints = 0; + for (Rental each : list()) { + frequentRenterPoints += each.frequentRenterPoints(); + } + return frequentRenterPoints; + } }