Mockito - 预期调用

Mockito 提供了对特定方法可进行的调用次数的特殊检查。假设 MathApplication 应仅调用 CalculatorService.serviceUsed() 方法一次，那么它就不应多次调用 CalculatorService.serviceUsed()。

//添加计算服务的行为以将两个数字相加
when(calcService.add(10.0,20.0)).thenReturn(30.00);

//限制方法调用次数为 1，不允许调用次数过多或过少
verify(calcService, times(1)).add(10.0, 20.0);

按如下方式创建 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);
}

示例

文件：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 − 测试 MathApplication 类

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

文件：MathApplicationTester.java

import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.never;

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 将运行器附加到测试类以初始化测试数据
@RunWith(MockitoJUnitRunner.class)
public class MathApplicationTester {
	
   //@InjectMocks 注解用于创建并注入mock对象
   @InjectMocks 
   MathApplication mathApplication = new MathApplication();

   //@Mock注解用于创建需要注入的模拟对象
   @Mock
   CalculatorService calcService;

   @Test
   public void testAdd(){
        //添加计算服务的行为以将两个数字相加
        when(calcService.add(10.0,20.0)).thenReturn(30.00);
                
        //添加 calc 服务的行为以减去两个数字
        when(calcService.subtract(20.0,10.0)).thenReturn(10.00);
        
        //测试添加功能
        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);
        
        //测试减法功能
        Assert.assertEquals(mathApplication.subtract(20.0, 10.0),10.0,0.0);
        
        //默认调用次数为 1
        verify(calcService).subtract(20.0, 10.0);
        
        //检查 add 函数是否被调用了三次
        verify(calcService, times(3)).add(10.0, 20.0);
        
        //验证该方法从未在模拟中被调用过
        verify(calcService, never()).multiply(10.0,20.0);
   }
}

步骤 4 − 执行测试用例

在 C:\> Mockito_WORKSPACE 中创建一个名为 TestRunner 的 Java 类文件来执行测试用例。

文件：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

现在运行测试运行器来查看结果 −

C:\Mockito_WORKSPACE>java TestRunner

验证输出。

true