tdd-java-demo

commit c20d13adcae2a0956385d06a6048d4672f1cd751
parent 7dfd9dcd990c09d86580d295367f0a3eb596ecb5
Author: Jul <jul@9o.is>
Date:   Wed, 31 Mar 2021 23:19:44 +0800

use Multiset in Basket methods

Diffstat:
M
src/main/java/takehome/Basket.java
 | 
7
++++---

1 file changed, 4 insertions(+), 3 deletions(-)
diff --git a/src/main/java/takehome/Basket.java b/src/main/java/takehome/Basket.java
@@ -17,16 +17,17 @@ public class Basket {
     }
 
     double getTotal() {
+        Multiset<Item> set = toMultiset(amounts);
         double total = 0;
-        for (Item item : amounts.keySet()) {
-            int amount = amounts.get(item);
+        for (Item item : set.elementSet()) {
+            int amount = getAmount(item);
             total += item.getTotal(amount);
         }
         return total;
     }
 
     Integer getAmount(Item item) {
-        return amounts.get(item);
+        return toMultiset(amounts).count(item);
     }
 
     double getTotal(Discount[] discounts) {