Happy End

RefactoringAssignment.iml

@@ -24,5 +24,15 @@
         <SOURCES />
       </library>
     </orderEntry>
+    <orderEntry type="module-library" scope="TEST">
+      <library name="JUnit4">
+        <CLASSES>
+          <root url="jar://$MAVEN_REPOSITORY$/junit/junit/4.13.1/junit-4.13.1.jar!/" />
+          <root url="jar://$MAVEN_REPOSITORY$/org/hamcrest/hamcrest-core/1.3/hamcrest-core-1.3.jar!/" />
+        </CLASSES>
+        <JAVADOC />
+        <SOURCES />
+      </library>
+    </orderEntry>
   </component>
 </module>

src/InsuranceCalculator.java

@@ -1,15 +1,19 @@
 public class InsuranceCalculator {
 
+    private InsuranceStrategy strategy;
     public double calculateInsurance(double income) {
         if (income <= 10000) {
-            return income*0.365;
+            strategy = new InsuranceStrategyLow();
+            return strategy.calculate(income);
         } else if (income <= 30000) {
-            return (income-10000)*0.2+35600;
+            strategy = new InsuranceStrategyMedium();
+            return strategy.calculate(income);
         } else if (income <= 60000) {
-            return (income-30000)*0.1+76500;
+            strategy = new InsuranceStrategyHigh();
+            return strategy.calculate(income);
         } else {
-            return (income-60000)*0.02+105600;
+            strategy = new InsuranceStrategyVeryHigh();
+            return strategy.calculate(income);
         }
     }
-
 }

src/InsuranceStrategy.java

@@ -0,0 +1,17 @@
+public abstract class InsuranceStrategy {
+
+    public double calculate(double income) {
+        return (income - getAdjustment()) * getWeight() + getConstant();
+    }
+    public double getWeight() {
+        return 0.02;
+    }
+
+    public int getAdjustment() {
+        return 60000;
+    }
+
+    public int getConstant() {
+        return 105600;
+    }
+}

src/InsuranceStrategyHigh.java

@@ -0,0 +1,17 @@
+public class InsuranceStrategyHigh extends InsuranceStrategy {
+
+    @Override
+    public int getConstant() {
+        return 76500;
+    }
+
+    @Override
+    public double getWeight() {
+        return 0.1;
+    }
+
+    @Override
+    public int getAdjustment() {
+        return 30000;
+    }
+}

src/InsuranceStrategyLow.java

@@ -0,0 +1,16 @@
+public class InsuranceStrategyLow extends InsuranceStrategy {
+    @Override
+    public int getConstant() {
+        return 0;
+    }
+
+    @Override
+    public double getWeight() {
+        return 0.365;
+    }
+
+    @Override
+    public int getAdjustment() {
+        return 0;
+    }
+}

src/InsuranceStrategyMedium.java

@@ -0,0 +1,16 @@
+public class InsuranceStrategyMedium extends InsuranceStrategy {
+    @Override
+    public int getConstant() {
+        return 35600;
+    }
+
+    @Override
+    public double getWeight() {
+        return 0.2;
+    }
+
+    @Override
+    public int getAdjustment() {
+        return 10000;
+    }
+}

src/InsuranceStrategyVeryHigh.java

@@ -0,0 +1,3 @@
+public class InsuranceStrategyVeryHigh extends InsuranceStrategy {
+
+}

test/InsuranceCalculatorTest.java

@@ -0,0 +1,29 @@
+import org.junit.Test;
+
+import static org.junit.Assert.*;
+
+public class InsuranceCalculatorTest {
+    @Test
+    public void low() {
+        assertEquals(1825, insuranceFor(5000), 0.01);
+    }
+
+    @Test
+    public void medium() {
+        assertEquals(38600, insuranceFor(25000), 0.01);
+    }
+
+    @Test
+    public void high() {
+        assertEquals(78500, insuranceFor(50000), 0.01);
+    }
+
+    @Test
+    public void veryHigh() {
+        assertEquals(106400, insuranceFor(100_000), 0.01);
+    }
+
+    private double insuranceFor(double income) {
+        return new InsuranceCalculator().calculateInsurance(income);
+    }
+}