单元测试mock之mockito使用

先来一个简单的例子来感受一下

外部接口类：TestService.java

 package com.yzl.mock;

 /**
  * 测试用服务
  *
  * @author yangzhilong
  */
 public interface TestService {
     /**
      *
      * 简单的测试服务，实现两个数相加
      *
      * @param a
      * @param b
      * @return
      */
     public int add(int a, int b);
 }

正常流程的JUnit单元测试方法类似如下：

 @Test
 public void test0(){
     //从spring容器里取到该接口的实现类
     TestService service = new ClassPathXmlApplicationContext("xxx.xml").getBean(TestService.class);
     int a=2, b=3;
     int result = service.add(a, b);
     Assert.assertSame("计算结果预期", 5, result);

     //拿到这个result然后调用其他的业务逻辑
 }

但是目前该接口的实现类(服务)目前是调用不通的，或者是压根还没做，但是你又不想因为这个接口而影响自己的代码的测试，就可以使用Mockito来模拟接口的调用。

首先引入mockito的jar

 <dependency>
     <groupId>org.mockito</groupId>
     <artifactId>mockito-core</artifactId>
     <version>2.1.0-RC.1</version>
     <scope>test</scope>
 </dependency>

然后编写代码如下：

 package com.yzl.mock;

 import org.junit.Assert;
 import org.junit.Test;
 import org.mockito.Mockito;

 public class MockTest {
     /*
      * 使用mockito来模拟测试
      */
     @Test
     public void test1(){
         TestService service = Mockito.mock(TestService.class);
         int a=2, b=3;
         //使用mockito模拟接口返回
         Mockito.when(service.add(a, b)).thenReturn(5);
         int result = service.add(a, b);
         System.out.println(a + "+" + b + "=" + result);
         Assert.assertSame("计算结果预期", 5, service.add(a, b));
         //使用result继续做后面的业务
     }
 }

上面的代码中Mockito的静态方法可以通过静态Import的形式导入，然后使用里面的静态方法就可以不带类名了，例如：

//非静态import的写法
import org.mockito.Mockito;
Mockito.when(service.add(a, b)).thenReturn(5);

//静态import的写法
import static org.mockito.Mockito.*;
when(service.add(a, b)).thenReturn(5);

运行junit单元测试方法将输出：

2+3=5