tdd-java-demo

commit 35456b5d1c54f5ecd7c414a04ec29769727d3607
parent 294f80bed11d6bc2b1cac1d28c3317605c88b19f
Author: Jul <jul@9o.is>
Date:   Wed, 31 Mar 2021 20:18:13 +0800

move instance method getTotal to Basket

Diffstat:
M
src/main/java/takehome/Basket.java
 | 
8
++++++++
M
src/main/java/takehome/Runner.java
 | 
11
+----------

2 files changed, 9 insertions(+), 10 deletions(-)
diff --git a/src/main/java/takehome/Basket.java b/src/main/java/takehome/Basket.java
@@ -25,4 +25,12 @@ public class Basket {
     Integer getAmount(Item item) {
         return getAmounts().get(item);
     }
+
+    double getTotal(Discount[] discounts) {
+        double total = getTotal();
+        for (Discount discount : discounts) {
+            total -= discount.getDiscount(this);
+        }
+        return total;
+    }
 }
diff --git a/src/main/java/takehome/Runner.java b/src/main/java/takehome/Runner.java
@@ -31,21 +31,12 @@ public class Runner {
         };
 
         Basket basket = new Basket(getItemAmounts(items));
-
-        double total = getTotal(basket, discounts);
+        double total = basket.getTotal(discounts);
 
         DecimalFormat df = new DecimalFormat("0.00");
         out.println("Total: $" + df.format(total));
     }
 
-    private double getTotal(Basket basket, Discount[] discounts) {
-        double total = basket.getTotal();
-        for (Discount discount : discounts) {
-            total -= discount.getDiscount(basket);
-        }
-        return total;
-    }
-
     private Map<Item, Integer> soupBreadRequirement() {
         Map<Item, Integer> soupBreadRequirement = new HashMap<>();
         soupBreadRequirement.put(SOUP, 2);