tdd-java-demo

commit 7d1bf70b0b0c18e79593bb3372fe8e807e5a8d46
parent a386721f53b218b7cc8b91b160305324d949756e
Author: Jul <jul@9o.is>
Date:   Wed, 31 Mar 2021 16:54:26 +0800

move instance method getTotal to Basket

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

2 files changed, 10 insertions(+), 10 deletions(-)
diff --git a/src/main/java/takehome/Basket.java b/src/main/java/takehome/Basket.java
@@ -12,4 +12,13 @@ public class Basket {
     public Map<Item, Integer> getAmounts() {
         return amounts;
     }
+
+    double getTotal() {
+        double total = 0;
+        for (Item item : getAmounts().keySet()) {
+            int amount = getAmounts().get(item);
+            total += item.getTotal(amount);
+        }
+        return total;
+    }
 }
diff --git a/src/main/java/takehome/Runner.java b/src/main/java/takehome/Runner.java
@@ -29,21 +29,12 @@ public class Runner {
             amounts.put(item, amount);
         }
 
-        double total = getTotal(new Basket(amounts));
+        double total = new Basket(amounts).getTotal();
 
         DecimalFormat df = new DecimalFormat("0.00");
         out.println("Total: $" + df.format(total));
     }
 
-    private double getTotal(Basket basket) {
-        double total = 0;
-        for (Item item : basket.getAmounts().keySet()) {
-            int amount = basket.getAmounts().get(item);
-            total += item.getTotal(amount);
-        }
-        return total;
-    }
-
     private int getItemAmount(Item item) {
         out.println("How much " + item.getName() + "?");
         return in.nextInt();