diff --git a/src/main/java/video/domain/Customer.java b/src/main/java/video/domain/Customer.java index f0e7881..d8b13d2 100644 --- a/src/main/java/video/domain/Customer.java +++ b/src/main/java/video/domain/Customer.java @@ -24,15 +24,28 @@ public Rentals getRentals() { } public RentalResult rentalMovies() { - int totalAmount = 0; + int totalAmount = totalAmount(); + + int frequentRenterPoints = frequentRenterPoints(); + + return new RentalResult(totalAmount, frequentRenterPoints); + } + + private int frequentRenterPoints() { int frequentRenterPoints = 0; for (Rental each : rentals.list()) { - int thisAmount = each.amount(); frequentRenterPoints += each.frequentRenterPoints(); - totalAmount += thisAmount; } + return frequentRenterPoints; + } - return new RentalResult(totalAmount, 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