Mockito(一) -- 入门篇

OnlyQi

墨衍会员 · AI 创作全网分发码龄16年

关注

原创最新推荐文章于 2025-01-23 13:50:38 发布 3.1w 阅读 AI 写同款· GEO 优化 ›

Mockito详细教程 Mockito详细教程，介绍了SpringBoot环境下如何使用Mockito辅助单元测试。阅读详情

Mockito是一种mock工具/框架。我理解EasyMock有点过时了，Mockito是现在比较流行的。

一，Mockito是干什么的？

例如现在我们写了一个Java类，该类扫描一个目录并将找到的文件都上传到FTP server。该类对于不同的FTP响应(找不到FTP server 或上传成功，或上传失败)，有一些后续操作。

在写这个类的UT时，我们就必须虚构出来一个FTP对象。这样在UT中，这个虚构的对象能够代替真正的FTP，对被测试类的调用做出一定的响应，从而知道被测试类是否正确的调用了FTP并做出一些正确的期望的响应。例如，我们希望调用一个虚构的FTP对象，强制在调用该对象时返回“找不到指定的FTP Server”，并测试被测试类是否记录了错误日志。

而Mockito就提供了一系列的方法，便于大家在UT中构建虚拟的对象(例如上例中的虚拟的FTP Server)，并使用。

关于mock的概念和EasyMock，大家可以参考：Mock object and EasyMock framework

二，我们使用Mockito做测试时，通常有如下三步。

如前面所说，我们需要创建一个mock对象来代替真的对象。因此，模拟对象是第一步。在Mockito中，模拟对象使用mock()方法。

强制mock对象被调用时的返回值是第二步，比如指定让虚构的FTP对象第一次被调用时返回"找不到FTP server"。这一步一般称为stubbing。一般是when(mockedList.get(0)).thenReturn("first")的样子。

验证被测试类是否正确工作是第三步，使用verify()。例如，验证当虚构的FTP对象返回"找不到FTP server"时，测试被测试类是否再次尝试连接FTP并记录日志。

经过这三步测试过程就完成啦！下面我们对第一步和第二步做说明。第三步我们在后续文章中再做讲解。

a, 模拟对象：

// 模拟LinkedList 的一个对象  
LinkedList mockedList = mock(LinkedList.class);   
  
// 此时调用get方法，会返回null，因为还没有对方法调用的返回值做模拟   
System.out.println(mockedList.get(999));

b, 模拟方法调用的返回值：

// 模拟获取第一个元素时，返回字符串first。  给特定的方法调用返回固定值在官方说法中称为stub。
when(mockedList.get(0)).thenReturn("first");   
  
// 此时打印输出first   
System.out.println(mockedList.get(0));

c, 模拟方法调用抛出异常：

// 模拟获取第二个元素时，抛出RuntimeException  
when(mockedList.get(1)).thenThrow(new RuntimeException());   
  
// 此时将会抛出RuntimeException  
System.out.println(mockedList.get(1));

没有返回值类型的方法也可以模拟异常抛出：

doThrow(new RuntimeException()).when(mockedList).clear();

d, 模拟调用方法时的参数匹配：

// anyInt()匹配任何int参数，这意味着参数为任意值，其返回值均是element  
when(mockedList.get(anyInt())).thenReturn("element");   
  
// 此时打印是element   
System.out.println(mockedList.get(999));

e, 模拟方法调用次数：

// 调用add一次   
mockedList.add("once");   
  
// 下面两个写法验证效果一样，均验证add方法是否被调用了一次  
verify(mockedList).add("once");   
verify(mockedList, times(1)).add("once");

另外还可以通过atLeast(int i)和atMost(int i)来替代time(int i)来验证被调用的次数最小值和最大值。

mock()的基本使用就介绍到这里。接下来在Mockito(二)--实例篇中我们会给出一个用Mockito做测试的完整例子。

收录于