tdd-java-demo

Basket.java

(1433B)

 package takehome.model;
 
 import com.google.common.collect.HashMultiset;
 import com.google.common.collect.Multiset;
 
 import java.time.LocalDate;
 import java.util.Map;
 
 public class Basket {
     private final Multiset<Item> items;
     private LocalDate purchaseDate;
 
     public Basket(Map<Item, Integer> amounts) {
         this.items = toMultiset(amounts);
         this.purchaseDate = LocalDate.MAX;
     }
 
     public Multiset<Item> getItems() {
         return items;
     }
 
     public double getTotal() {
         double total = 0;
         for (Item item : items.elementSet()) {
             int amount = getAmount(item);
             total += item.getTotal(amount);
         }
         return total;
     }
 
     public Integer getAmount(Item item) {
         return items.count(item);
     }
 
     public double getTotal(Discount[] discounts) {
         double total = getTotal();
         for (Discount discount : discounts) {
             if (discount.isValid(purchaseDate)) {
                 total -= discount.getDiscount(this);
             }
         }
         return total;
     }
 
     private Multiset<Item> toMultiset(Map<Item, Integer> map) {
         Multiset<Item> multiset = HashMultiset.create();
         for (Item item : map.keySet()) {
             multiset.add(item, map.get(item));
         }
         return multiset;
     }
 
     public void setPurchaseDate(LocalDate purchaseDate) {
         this.purchaseDate = purchaseDate;
     }
 }