Mockito - 回调

Mockito提供了一个Answer接口，允许使用通用接口进行存根.

语法

//添加数字
时的行为(calcService.add(20.0,10.0)).thenAnswer(new Answer< Double>(){
 @Override 
 public double answer(InvocationOnMock invocation)throws Throwable {
//获取传递给mock的参数
 Object [] args = invocation.getArguments(); 
//获取模拟
 Object mock = invocation.getMock() ; 
//返回结果
return 30.0; 
} 
});

示例

步骤1号;创建一个名为CalculatorService的接口来提供数学函数

文件: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);
}

第2步去;创建一个JAVA类来表示MathApplication

文件: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);
   }
}

第3步和第3步;测试MathApplication类

让我们通过在其中注入一个模拟的calculatorService来测试MathApplication类.模拟将由Mockito创建.

这里我们通过when()添加了一个模拟方法调用，add()到模拟对象.但是在测试期间，我们在调用add()之前调用了subtract().当我们使用Mockito.createStrictMock()创建一个模拟对象时，该方法的执行顺序很重要.

文件:MathApplicationTester.java

import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;
import static org.mockito.Mockito.inOrder;

import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.InOrder;
import org.mockito.runners.MockitoJUnitRunner;

// @RunWith attaches a runner with the test class to initialize the test data
@RunWith(MockitoJUnitRunner.class)
public class MathApplicationTester {
	
   private MathApplication mathApplication;
   private CalculatorService calcService;

   @Before
   public void setUp(){
      mathApplication = new MathApplication();
      calcService = mock(CalculatorService.class);
      mathApplication.setCalculatorService(calcService);
   }

   @Test
   public void testAdd(){

      //add the behavior to add numbers
      when(calcService.add(20.0,10.0)).thenAnswer(new Answer<Double>() {

         @Override
         public Double answer(InvocationOnMock invocation) throws Throwable {
            //get the arguments passed to mock
            Object[] args = invocation.getArguments();
			
            //get the mock 
            Object mock = invocation.getMock();	
			
            //return the result
            return 30.0;
         }
      });

      //test the add functionality
      Assert.assertEquals(mathApplication.add(20.0, 10.0),30.0,0);
   }
}

第4步去;执行测试用例

在 C:\>中创建名为TestRunner的java类文件Mockito_WORKSPACE 执行测试用例.

文件: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());
   }
}  	

第5步去;验证结果

使用 javac 编译器编译类，如下所示 :

 
 C:\ Mockito_WORKSPACE> javac CalculatorService.java MathApplication. 
 java MathApplicationTester.java TestRunner.java

现在运行Test Runner查看结果 :

 
 C:\ Mockito_WORKSPACE> java TestRunner

验证输出.

 
 true