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,21 @@
 public class InsuranceCalculator {
-
+    private InsuranceStrategy strategy;
     public double calculateInsurance(double income) {
+
         if (income <= 10000) {
-            return income*0.365;
-        } else if (income <= 30000) {
-            return (income-10000)*0.2+35600;
-        } else if (income <= 60000) {
-            return (income-30000)*0.1+76500;
+            strategy = new InsuranceStrategyLow();
+            return strategy.calculate(income);
+        } else if (income <= 36500) {
+            strategy = new InsuranceStrategyMedium();
+            return strategy.calculate(income);
+        } else if (income <= 70000) {
+            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,9 @@
+public abstract class InsuranceStrategy {
+    public double calculate(double income) {
+        return (income - getAdjustment()) * getWeight() + getConstant();
+    }
+
+    public abstract int getConstant();
+    public abstract double getWeight();
+    public abstract int getAdjustment();
+}

src/InsuranceStrategyHigh.java

@@ -0,0 +1,13 @@
+public class InsuranceStrategyHigh extends InsuranceStrategy {
+    public InsuranceStrategyHigh() { }
+    @Override
+    public int getConstant() { return 70000; }
+
+    @Override
+    public double getWeight() { return 0.18; }
+
+    @Override
+    public int getAdjustment() {
+        return 25000;
+    }
+}

src/InsuranceStrategyLow.java

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

src/InsuranceStrategyMedium.java

@@ -0,0 +1,15 @@
+public class InsuranceStrategyMedium extends InsuranceStrategy {
+    public InsuranceStrategyMedium() { }
+    @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,15 @@
+public class InsuranceStrategyVeryHigh extends InsuranceStrategy {
+    public InsuranceStrategyVeryHigh() { }
+    @Override
+    public int getConstant() {
+        return 105600;
+    }
+    @Override
+    public double getWeight() {
+        return 0.02;
+    }
+    @Override
+    public int getAdjustment() {
+        return 60000;
+    }
+}

test/InsuranceTest.java

@@ -0,0 +1,29 @@
+import org.junit.Test;
+
+import static org.junit.Assert.*;
+public class InsuranceTest {
+    @Test
+    public void low() {
+        assertEquals(30000, insuranceFor(5000), 0.01);
+    }
+
+    @Test
+    public void medium() {
+        assertEquals(38600, insuranceFor(25000), 0.01);
+    }
+
+    @Test
+    public void high() {
+        assertEquals(74500, 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);
+    }
+
+}