Mockito(二)--实例篇
编程技术 / houtizong 发布于 3年前 56
学习了基本知识后，就可以实战了，Mockito的实际使用还是比较麻烦的。因为在实际使用中，最常遇到的就是需要模拟第三方类库的行为。
比如现在有一个类FTPFileTransfer，实现了向FTP传输文件的功能。这个类中使用了apache的ftp类org.apache.commons.net.ftp.FTPClient；现在测试FTPFileTransfer 这个类中的isFTPConnected方法, 希望模拟无法连接ftp的情况，测试是否记录了错误log。
public class FTPFileTransfer { //为了测试当ftp链接不上时，是否真的会记log，我们必须mock一个假的FTPClient对象，用该对象传递/覆盖掉真实的FTPClient对象ftp，并强制让这个假对象返回"无法连接",然后看是否会记log. private FTPClient ftp; private boolean isFTPConnected(){ if (!ftp.isConnected()) { logger.error("Disconnected from FTP."); }} 因此使用mock需要解决的问题是： 如何用mock的FTP对象覆盖掉真实代码中调用的FTPClient。因此需要将mock对象传递进去。
这里对源代码有一个限制：
源代码中必须使用set和get方法来设置/获得ftp对象，这样测试代码可以使用set来传递mock对象。或者测试代码中写一个方法覆盖掉源代码中实例化ftp对象的方法，且测试代码中使用mock ftp对象。
现假定源代码中使用了get/set，有如下方法进行单元测试：
一.创建一个返回FTPFileTransfer instance的方法
//使用mock()方法创建一个FTP的mock对象mockedFTP。FTPClient mockedFTP = mock(FTPClient.class);//Stub “无法连接”when(mockedFTP.isConnected())).thenReturn("无法连接"); //写一个get方法返回FTPFileTransfer的实例用来测试，将mockedFTP作为参数传递进去。同时在这个方法内部用set方法将FTPFileTransfer类中的成员变量FTPClient ftp更改为mockFTP。//这样我们就得到了一个FTPFileTransfer的实例，同时里面的ftp已经变成了我们希望的mockFTP。private FTPFileTransfer getMockTaskFileTransfer(final FTPClientmockedFTP) { FTPFileTransfer test = new FTPFileTransfer("127.0.0.1", 8888, "//usr", "username", "password"); test.setFTPClient(mockedFTP); return test;}@Testpublic void testTransfer() throws SocketException, IOException{ FTPFileTransfer test = getMockTaskFileTransfer(); //得到这个实例以后，就直接调用这个实例的isFTPConnected方法，然后去log文件里找有没有我们希望的log就行了。注意此时，mockedFTP一定会返回"无法连接"，所以isFTPConnected一定会记log。 test.isFTPConnected();} 二.用subclass-and-override实现
从名字就可以看出，通过创建被测试类的子类，覆盖掉被测试类的getFTPClient()方法，将mock对象传递进去。
class MockFTPFileTransfer extends FTPFileTransfer{ public MockFTPFileTransfer(){ super("127.0.0.1", 8888, "//usr", "username", "password"); } //源代码中必须使用get来获得ftp对象，否则mock不会生效 @Override public FTPClient getFTPClient(){ FTPClient mockedFTP = mock(FTPClient.class); when(mockedFTP.isConnected()).thenReturn(true); return mockedFTP; }}
@Testpublic void testTransfer() throws SocketException, IOException{ FTPFileTransfer test = new MockFTPFileTransfer(); test.isFTPConnected();} 三.用partial mock实现
partial mock是1.8之后的新功能。通常情况下会使用mock出来的对象完全覆盖掉被模拟的对象，对于那些没有stub的方法，则会返回build-in 类型的默认值。
@Testpublic void testTransfer() throws SocketException, IOException{ //模拟一个FTPClient对象，同时stub行为。 FTPClient mockedFTP = mock(FTPClient.class); when(mockedFTP.isConnected()).thenReturn(true); //spy可以模拟一个real object //这里的可以认为是spy在real object上包了一层，除了getFTPClient()被覆盖掉以外，其他方法仍然是真实对象的。 //这就是partial mock的概念： 仅仅用mock对象覆盖源对象的一部分，而不是全部。 FTPFileTransfer spyFTP= spy(new FTPFileTransfer("127.0.0.1", "//usr", "username", "password")); when(spyFTP.getFTPClient()).thenReturn(mockedFTP); spyFTP.setFTPClient(mockedFTP); spyFTP.isFTPConnected();} 请勿发布不友善或者负能量的内容。与人为善，比聪明更重要！
技术博客集 - 网站简介：
前后端技术：
后端基于Hyperf2.1框架开发,前端使用Bootstrap可视化布局系统生成
网站主要作用：
1.编程技术分享及讨论交流，内置聊天系统;
2.测试交流框架问题，比如：Hyperf、Laravel、TP、beego;
3.本站数据是基于大数据采集等爬虫技术为基础助力分享知识，如有侵权请发邮件到站长邮箱，站长会尽快处理;
4.站长邮箱：[email protected];
文章归档
文章标签
友情链接
