Using JUnit and EasyMock

Project: JUnit_Easymock
IDE: Eclipse
JARs:
junit-4.7.jar
easymock-2.5.2.jar

IncomeCalculator.java
public class IncomeCalculator{
private ICalcMethod calcMethod;
private Position position;
public void setCalcMethod(ICalcMethod calcMethod){
this.calcMethod = calcMethod;
}
public void setPosition(Position position){
this.position = position;
}
public double calc (){
if (calcMethod==null){
throw new CalcMethodException("CalcMethod not yet maintained");
}
if (position==null){
throw new PositionException("Position not yet maintained");
}
return calcMethod.calc(position);
}
}

Position.java
public enum Position {
BOSS, PROGRAMMER, SURFER
}

IncomeCalculatorTest.java
import static org.junit.Assert.*;
import org.junit.Test;
import static org.junit.Assert.assertEquals;import static org.junit.Assert.fail;import com.somellc.income.exceptions.CalcMethodException;import com.somellc.income.exceptions.PositionException;import com.somellc.income.method.ICalcMethod;
import org.easymock.EasyMock;import org.junit.Before;import org.junit.Test;
public class IncomeCalculatorTest {
private ICalcMethod calcMethod;
private IncomeCalculator calc;
@Before
public void setUp() throws Exception {
calcMethod = EasyMock.createMock(ICalcMethod.class); calc = new IncomeCalculator();
}
@Test
public void testCalc1() {
// Setting up the expected value of the method call calc
EasyMock.expect(calcMethod.calc(Position.BOSS)).andReturn(70000.0) .times(2);
EasyMock.expect(calcMethod.calc(Position.PROGRAMMER)) .andReturn(50000.0);
// Setup is finished need to activate the mock EasyMock.replay(calcMethod);
calc.setCalcMethod(calcMethod); try { calc.calc(); fail("Exception did not occur");
} catch (PositionException e) {
}
calc.setPosition(Position.BOSS);
assertEquals(70000.0, calc.calc());
assertEquals(70000.0, calc.calc());
calc.setPosition(Position.PROGRAMMER);
assertEquals(50000.0, calc.calc());
calc.setPosition(Position.SURFER);
EasyMock.verify(calcMethod); }
@Test(expected = CalcMethodException.class)
public void testNoCalc() { calc.setPosition(Position.SURFER); calc.calc(); }
@Test(expected = PositionException.class)
public void testNoPosition() { EasyMock.expect(calcMethod.calc(Position.BOSS)).andReturn(70000.0); EasyMock.replay(calcMethod); calc.setCalcMethod(calcMethod);
calc.calc(); }
@Test(expected = PositionException.class)
public void testCalc2() { // Setting up the expected value of the method call calc EasyMock.expect(calcMethod.calc(Position.SURFER)).andThrow( new PositionException("Don't know this guy")).times(1);
// Setup is finished need to activate the mock
EasyMock.replay(calcMethod); calc.setPosition(Position.SURFER);
calc.setCalcMethod(calcMethod);
calc.calc(); }
}