mockito provides the following additional methods to vary the expected call counts.
atleast (int min) − expects min calls.
atleastonce () − expects at least one call.
atmost (int max) − expects max calls.
example
step 1 − create an interface calculatorservice to provide mathematical functions
file: calculatorservice.java
public interface calculatorservice { public double add(double input1, double input2); public double subtract(double input1, double input2); public double multiply(double input1, double input2); public double divide(double input1, double input2); }
step 2 − create a java class to represent mathapplication
file: mathapplication.java
public class mathapplication { private calculatorservice calcservice; public void setcalculatorservice(calculatorservice calcservice){ this.calcservice = calcservice; } public double add(double input1, double input2){ return calcservice.add(input1, input2); } public double subtract(double input1, double input2){ return calcservice.subtract(input1, input2); } public double multiply(double input1, double input2){ return calcservice.multiply(input1, input2); } public double divide(double input1, double input2){ return calcservice.divide(input1, input2); } }
step 3 − test the mathapplication class
let's test the mathapplication class, by injecting in it a mock of calculatorservice. mock will be created by mockito.
file: mathapplicationtester.java
import static org.mockito.mockito.verify; import static org.mockito.mockito.when; import static org.mockito.mockito.atleastonce; import static org.mockito.mockito.atleast; import static org.mockito.mockito.atmost; import org.junit.assert; import org.junit.test; import org.junit.runner.runwith; import org.mockito.injectmocks; import org.mockito.mock; import org.mockito.runners.mockitojunitrunner; // @runwith attaches a runner with the test class to initialize the test data @runwith(mockitojunitrunner.class) public class mathapplicationtester { //@injectmocks annotation is used to create and inject the mock object @injectmocks mathapplication mathapplication = new mathapplication(); //@mock annotation is used to create the mock object to be injected @mock calculatorservice calcservice; @test public void testadd(){ //add the behavior of calc service to add two numbers when(calcservice.add(10.0,20.0)).thenreturn(30.00); //add the behavior of calc service to subtract two numbers when(calcservice.subtract(20.0,10.0)).thenreturn(10.00); //test the add functionality assert.assertequals(mathapplication.add(10.0, 20.0),30.0,0); assert.assertequals(mathapplication.add(10.0, 20.0),30.0,0); assert.assertequals(mathapplication.add(10.0, 20.0),30.0,0); //test the subtract functionality assert.assertequals(mathapplication.subtract(20.0, 10.0),10.0,0.0); //check a minimum 1 call count verify(calcservice, atleastonce()).subtract(20.0, 10.0); //check if add function is called minimum 2 times verify(calcservice, atleast(2)).add(10.0, 20.0); //check if add function is called maximum 3 times verify(calcservice, atmost(3)).add(10.0,20.0); } }
step 4 − execute test cases
create a java class file named testrunner in c:\> mockito_workspace to execute test case(s)
file: testrunner.java
import org.junit.runner.junitcore; import org.junit.runner.result; import org.junit.runner.notification.failure; public class testrunner { public static void main(string[] args) { result result = junitcore.runclasses(mathapplicationtester.class); for (failure failure : result.getfailures()) { system.out.println(failure.tostring()); } system.out.println(result.wassuccessful()); } }
step 5 − verify the result
compile the classes using javac compiler as follows −
c:\mockito_workspace>javac calculatorservice.java mathapplication. java mathapplicationtester.java testrunner.java
now run the test runner to see the result −
c:\mockito_workspace>java testrunner
verify the output.
true