当前位置:首页 > 开发 > 编程语言 > 编程 > 正文

Mockito(二)--实例篇

发表于: 2013-12-09   作者:bijian1013   来源:转载   浏览:
摘要:         学习了基本知识后,就可以实战了,Mockito的实际使用还是比较麻烦的。因为在实际使用中,最常遇到的就是需要模拟第三方类库的行为。         比如现在有一个类FTPFileTransfer,实现了向FTP传输文件的功能。这个类中使用了a

        学习了基本知识后,就可以实战了,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;
}

@Test
public 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;
    }
}

 

@Test
public void testTransfer() throws SocketException, IOException{       
    FTPFileTransfer test = new MockFTPFileTransfer();
    test.isFTPConnected();
}

三.用partial mock实现

        partial mock是1.8之后的新功能。通常情况下会使用mock出来的对象完全覆盖掉被模拟的对象,对于那些没有stub的方法,则会返回build-in 类型的默认值。

@Test
public 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();
}

Mockito(二)--实例篇

  • 0

    开心

    开心

  • 0

    板砖

    板砖

  • 0

    感动

    感动

  • 0

    有用

    有用

  • 0

    疑问

    疑问

  • 0

    难过

    难过

  • 0

    无聊

    无聊

  • 0

    震惊

    震惊

编辑推荐
基本类型实例 //1.定义一个一维数组,先声明,在分配空间 int []number;//生命,没有初始化,number=
基本类型实例 //1.定义一个一维数组,先声明,在分配空间 int []number;//生命,没有初始化,number=
基本类型实例 //1.定义一个一维数组,先声明,在分配空间 int []number;//生命,没有初始化,number=
为什么需要Mock 测试驱动的开发(Test Driven Design, TDD)要求我们先写单元测试,再写实现代码。在
1引言 在标题的取名上,不敢说颇费心机,也算得上花费了一点功夫的。首先想到的是“架构设计过程”
1引言 在标题的取名上,不敢说颇费心机,也算得上花费了一点功夫的。首先想到的是“架构设计过程”
1引言 在标题的取名上,不敢说颇费心机,也算得上花费了一点功夫的。首先想到的是“架构设计过程”
转自:http://www.ruanyifeng.com/blog/2015/07/flex-examples.html 作者: 阮一峰 日期: 2015年7
indexedDB 相对于html5 中提到 web SQL Database,w3c已经明确声明放弃对其的继续支持,开始支持新
还有另一个问题就是所有的内容都放在同一个地方。如果我们的服务器在芝加哥,那么美国中西部的人们
版权所有 IT知识库 CopyRight © 2009-2015 IT知识库 IT610.com , All Rights Reserved. 京ICP备09083238号