Java单元测试神器之Mockito

什么是 Mock 测试

Mock测试就是在测试过程当中，对于某些不容易构造或者不容易获取的对象，用一个虚拟的对象来建立以便测试的测试方法。什么是不容易构造的对象呢？例如HttpServletRequest，须要在有servlet容器环境中建立获取。那不容易获取的对象呢？如一个JedisCluster，须要准备redis相关环境，而后设置进去等等。

Mock 能够分解在单元测试中耦合的其余类或者接口，它可以帮你模拟这些依赖，并帮你验证所调用的依赖的行为。

场景事例

 

 

当咱们须要测试OrderService时，按照咱们常规的作法呢，都是要先准备好redis，跟db的环境，而后构造UserService跟CouponService注入进来，此时须要构建完整的依赖树，其过程是比较繁琐的，万一数据库连不上，依赖找不到，服务挂了... 时间一长可能会打击咱们对项目进行单测的积极性，因此这时候颇有必要寻求一种优雅的方式来解决。

铛铛铛~这时候Mockito出现了(java中Mock框架比较多，可是本篇只介绍这个)，它会把那些繁琐的依赖通通转化为Mock Object，以下图，这样咱们就能够专一的进行咱们的单测，减小在解决依赖上浪费的时间了。

 

 

直接开干

关于Mockito的简介这里就不在赘述了，你们有兴趣能够自行去官方文档查阅，这里主要带你们了解一些经常使用的Mock方法。

maven依赖

<dependency>
    <groupId>org.mockito</groupId>
    <artifactId>mockito-core</artifactId>
    <version>2.23.4</version>
    <scope>test</scope>
</dependency>

　　

为了代码测试的方便，直接在测试类中静态导入 import static org.mockito.Mockito.*;

基础方法

@Test
   public void testMockBase(){
       //建立ArrayList的Mock对象
       List mockList = mock(ArrayList.class);
       //pass
       Assert.assertTrue(mockList instanceof ArrayList);

       //当咱们mockList调用方法去add("张三")的时候会返回true
       when(mockList.add("张三")).thenReturn(true);
       //当咱们mockList调用方法size()的时候返回10
       when(mockList.size()).thenReturn(10);
       //pass
       Assert.assertTrue(mockList.add("张三"));
       //pass
       Assert.assertFalse(mockList.add("李四"));
       //pass
       Assert.assertEquals(mockList.size(),10);
       //null
       System.out.println(mockList.get(0));
   }

mock静态方法会建立一个Mock对象，因为 Mock对象 并不会真的执行方法中的代码，因此若是未指定返回值的话会返回默认值（如19行）。第9、十行咱们指定了mockList在执行特定方法后须要返回的值，因此在assertTrue校验是没问题的，可是add("李四")，咱们并没设置，因此是false。

校验方法调用次数

//使用mock
 List mockedList = mock(ArrayList.class);
 mockedList.add("once");

 mockedList.add("twice");
 mockedList.add("twice");

 mockedList.add("three times");
 mockedList.add("three times");
 mockedList.add("three times");
 
 //这里默认是判断该方法调用times(1),同下
 verify(mockedList).add("once");
 verify(mockedList, times(1)).add("once");

 verify(mockedList, times(2)).add("twice");
 verify(mockedList, times(3)).add("three times");
 //从没调用，times(0)
 verify(mockedList, never()).add("never happened");
 //最少一次，最少几回，最多几回
 verify(mockedList, atLeastOnce()).add("three times");
 verify(mockedList, atLeast(2)).add("three times");
 verify(mockedList, atMost(5)).add("three times");

其实在上述的代码中，命名是比较直观的，因此我这边就直接注释在代码中了。

校验方法调用时长

//方法执行在100ms之内的时候能够经过
   verify(mock, timeout(100)).someMethod();
   //同上
   verify(mock, timeout(100).times(1)).someMethod();

   //方法2次调用均没超过100ms
   verify(mock, timeout(100).times(2)).someMethod();

   verify(mock, timeout(100).atLeast(2)).someMethod();

 

经过超时检测能够校验咱们的方法逻辑会不会有出现问题而致使超时的地方。

参数匹配

linkedList.add("element");
// anyInt() 任何整数咱们都返回 element 
when(linkedList.get(anyInt())).thenReturn("element");

System.out.print(linkedList.get(10));//返回element

 

方法抛出异常

 @Test(expected = RuntimeException.class)
    public void doThrow(){
        List list = mock(List.class);
        doThrow(new RuntimeException()).when(list).add(1);
        list.add(1);
    }

 

使用注解注入

  public class ArticleManagerTest {

       @Mock private ArticleCalculator calculator;
       @Mock private ArticleDatabase database;
       @Mock private UserProvider userProvider;

       private ArticleManager manager;

要注意的是，经过注解的方式用使用的话，咱们必须在添加初始化mock的代码，否则即便标注了注解也会是null

MockitoAnnotations.initMocks(testClass);

 

关于Mockito更多详细的用法，你们能够直接参考官方文档，由于各类“奇技淫巧”确实比较多，后面也更新对java8 lambda的支持，不少功能仍是期待你们去挖掘~

 

 

更多详细用法可直接参考官方文档：

static.javadoc.io/org.mockito…

相信当你熟练使用Mockito之后，你会爱上写单测的，也会让你代码健壮性有所增强。有些bug能提早发现的话，总比运行的时候被别人半夜叫起来修复舒服是吧?

 

 

微信公众号《深夜里的程序猿》 -  天天分享最干的干货