Java单元测试神器之Mockito

什么是 Mock 测试 ？

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

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

场景事例

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

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

直接开干

关于Mockito的简介这里就不在赘述了，大家有兴趣可以自行去官方文档查阅，这里主要带大家了解一些常用的Mock方法。

maven依赖

<dependency>

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

基础方法

1   @Test
2   public void testMockBase(){
3       //创建ArrayList的Mock对象
4       List mockList = mock(ArrayList.class);
5       //pass
6       Assert.assertTrue(mockList instanceof ArrayList);
7
8       //当我们mockList调用方法去add("张三")的时候会返回true
9       when(mockList.add("张三")).thenReturn(true);
10       //当我们mockList调用方法size()的时候返回10
11       when(mockList.size()).thenReturn(10);
12       //pass
13       Assert.assertTrue(mockList.add("张三"));
14       //pass
15       Assert.assertFalse(mockList.add("李四"));
16       //pass
17       Assert.assertEquals(mockList.size(),10);
18       //null
19       System.out.println(mockList.get(0));
20   }

mock静态方法会创建一个Mock对象，由于 Mock对象 并不会真的执行方法中的代码，所以如果未指定返回值的话会返回默认值。我们指定了mockList在执行特定方法后需要返回的值，所以在assertTrue校验是没问题的，when().thenReturn() 表示当执行到某个指定的方法时，返回特定的内容，但是add("李四")，我们并没设置，所以是false。

校验方法调用次数

1 //使用mock
2 List mockedList = mock(ArrayList.class);
3 mockedList.add("once");
4
5 mockedList.add("twice");
6 mockedList.add("twice");
7
8 mockedList.add("three times");
9 mockedList.add("three times");
10 mockedList.add("three times");
11 
12 //这里默认是判断该方法调用times(1),同下
13 verify(mockedList).add("once");
14 verify(mockedList, times(1)).add("once");
15
16 verify(mockedList, times(2)).add("twice");
17 verify(mockedList, times(3)).add("three times");
18 //从没调用，times(0)
19 verify(mockedList, never()).add("never happened");
20 //最少一次，最少几次，最多几次
21 verify(mockedList, atLeastOnce()).add("three times");
22 verify(mockedList, atLeast(2)).add("three times");
23 verify(mockedList, atMost(5)).add("three times");

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

校验方法调用时长

1   //方法执行在100ms以内的时候可以通过
2   verify(mock, timeout(100)).someMethod();
3   //同上
4   verify(mock, timeout(100).times(1)).someMethod();
5
6   //方法2次调用均没超过100ms
7   verify(mock, timeout(100).times(2)).someMethod();
8
9   verify(mock, timeout(100).atLeast(2)).someMethod();

通过超时检测可以校验我们的方法逻辑会不会有出现问题而导致超时的地方。

参数匹配

1linkedList.add("element");
2// anyInt() 任何整数我们都返回 element 
3when(linkedList.get(anyInt())).thenReturn("element");
4
5System.out.print(linkedList.get(10));//返回element

方法抛出异常​​​​​​

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

使用注解注入

1  public class ArticleManagerTest {
2
3       @Mock private ArticleCalculator calculator;
4       @Mock private ArticleDatabase database;
5       @Mock private UserProvider userProvider;
6
7       private ArticleManager manager;

要注意的是，通过注解的方式用使用的话，我们必须在添加初始化mock的代码，不然即使标注了注解也会是null

MockitoAnnotations.initMocks(testClass);

关于Mockito更多详细的用法，大家可以直接参考官方文档，因为各种“骚操作”确实比较多，后面也更新对java8 lambda的支持，很多功能还是期待大家去挖掘~

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

https://static.javadoc.io/org.mockito/mockito-core/2.25.1/org/mockito/Mockito.html#0

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

喜欢的话，关注微信公众号《深夜里的程序员》，每天发布高质量IT干货，还有惊喜等着你~