Mockito Tutorial on Mockito Varying Calls

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