当前位置:
首页
文章
前端
详情

Mockito单元测试

Mockito简介

Mockito是一个单元测试框架,需要Junit的支持。在我们的项目中,都存在相当多的依赖关系,当我们在测试某一个业务相关的接口或则方法时,绝大多数时候是没有办法或则很难去添加所有的依赖,因为这中间肯定会涉及到别的业务逻辑。而在开发过程中,可能这个模块根本都还没有。那可咋怎啊?这个时候一种叫做mock测试的方式就顺势崛起。通过模拟出依赖对象,并对涉及到的方法设置预期值。这样你就可以只关心依赖方法的结果,从而完成对本模块的单元测试。这种方法还细化了测试粒度。棒棒的。想做更多了解就自行解决了。

Mockito的使用

  1. 首先来一个最基本的Junit测试

    @Test public void stringUtilTest(){ boolean b = StringUtil.isEmpty("good"); Assert.assertTrue("must true",b);//断言 }

  像这种对工具类的测试,一般很少依赖别的类,所以直接断言之。当然断言的类型还有很多,这里就试用一下对boolean的断言。

  2.  mockito对依赖的模拟,并设置预期返回值。针对依赖的方法有返回值。在单元测试时,我们不想也最好不要直接调用依赖的方法的具体实现,因为所依赖的方法可能本来就没有经过测试,还存在bug,难道这时候又要为依赖的方法再写一个test case?或则这个方法由别人开发,但是目前还没有实现,难道要自己去实现?you`d better say NO!看看下面怎么做的。

import org.mockito.Mockito;

import java.util.ArrayList;
import java.util.Date;
import java.util.List;

/**
 * @auther guozg
 */

public class MockTest {
    @Test
    public void mockReturnTest(){
        // mock creation 创建模拟对象
        UserDao mockeDao = Mockito.mock(UserDao.class);
        UserService s = new UserService();//创建被测试类
        s.setDao(mockeDao);//为被测试类添加依赖
        Mockito.when(mockeDao.getData(Mockito.anyString())).thenReturn(4).thenReturn(1);//为模拟对象方法设置预期返回值,
        boolean b = s.checkDate();                          //多个thenReturn表示多次调用时,依次返回
        boolean b1 = s.checkDate();              //如果设置的预期个数少于调用次数,超过的调用都返回最后一个。
        boolean b2 = s.checkDate();              //如果设置的预期个数多于调用次数,任然依次返回相应值
        Mockito.verify(mockeDao, Mockito.times(1)).getData(Mockito.anyString());
        Assert.assertTrue("must true",b);
        Assert.assertFalse("must true",b1);
        Assert.assertFalse("must true",b2);
    }
}

class UserService{
    UserDao dao ;
    public boolean checkDate(){     String id = "123";
        if(dao.getData(id)>3){
            return true;
        }
        return false;
    }

    public UserDao getDao() {
        return dao;
    }

    public void setDao(UserDao dao) {
        this.dao = dao;
    }

}

class UserDao{
    public Integer getData(String id){ return 0; } }

  这里随便举的简单例子,我需要对UserService.checkDate()做一个单元测试。而这时候UserService对UserDao存在依赖关系。所以这时候为了隔离UserDao的实现,通过Mockito.mock模拟出UserDao对象。并为掉用的getData()设置了预期返回值。然后调用要测试的方法、验证设置预期的方法是否被调用,最后对测试方法的返回值断言。这里指的一说的是Mockito.anyString()这个方法,他的主要目的是表示在模拟状态下,getData()的参数可以是任意字符串。当然也可以直接给定一些参数,如果模拟对象指定的参数和实际逻辑给的参数不一致,这时候不会返回实际UserDao中getData()的值也不会返回设置的预期值,而是返回的返回类型的默认值,比如int就返回0,boolean返回false等。Mockito还有很多类似的方法,比如anyInt(),anyBoolen(),anyCollection()等等。

  还有就是看到thenReturn()方法,这是在为依赖方法设置预期返回值,这个就可以有自己控制了。然后看到可以连续多次设置,这个在代码里面有注释了,由于对语言表达能力的不自信,来个表格展示一下。

                Mockito单元测试

  然后还有一个验证方法是否被调用Mockito.verify()这个方法的参数是模拟的对象、VerificationMode。这个VerificationMode就是一个验证模型,可以是代表调用次数、是否调用、超时验证,等等(有些我也不知道干啥的,要进一步研究)。verify()可以理解为断言。

  3.  对于有返回值的方法,我们可以通过设置预期来控制,并且隔离掉具体的实现。但是对于没有返回值得方法呢?不着急,Mockti为我们提供了另一种方式可以控制模拟对象的方法的行为,包括逻辑处理、抛异常、返回值、执行原方法逻辑以及什么都不做。doReturn()|doThrow()| doAnswer()|doNothing()|doCallRealMethod();下面先来测试一下doAnswer();首先,我们会在UserDao 中加一个setUser(User u)方法,为user设置年龄,然后通过UserService的checkData中调用。然后我们通过doAnwser去控制setUser的逻辑。

package com.centnet.train.user.controller;

import org.junit.Assert;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mockito;
import org.mockito.invocation.InvocationOnMock;
import org.mockito.stubbing.Answer;

import java.util.ArrayList;
import java.util.Date;
import java.util.List;

/**
 * @auther guozg
 */

public class MockTest {
   

    @Test
    public void doAnwserTest(){
        // mock creation 创建模拟对象
        UserDao mockeDao = Mockito.mock(UserDao.class);
        UserService s = new UserService();//创建被测试类
        User u = new User();
        s.setDao(mockeDao);//为被测试类添加依赖
        Mockito.doAnswer(new Answer() {
            public Object answer(InvocationOnMock invocationOnMock) throws Throwable {
                User u = invocationOnMock.getArgument(0);
                u.setName("ggg");
                return null;
            }
        }).when(mockeDao).setUser(u);
        Mockito.when(mockeDao.getData()).thenReturn(4);
        boolean b = s.checkDate(u);

        Mockito.verify(mockeDao, Mockito.times(1)).getData();
        Mockito.verify(mockeDao, Mockito.times(1)).setUser(u);
        Assert.assertTrue("must true",b);
        Assert.assertNotNull("控制不成功",u.getName());        Assert.assertNull("原逻辑被执行",u.getAge());
    }
}

class UserService{
    UserDao dao ;
    public boolean checkDate(User u){
        dao.setUser(u);
        if(dao.getData()>3){
            return true;
        }
        return false;
    }

    public UserDao getDao() {
        return dao;
    }

    public void setDao(UserDao dao) {
        this.dao = dao;
    }

    public void setUser(User user){
        user.setAge(12);

    }

}

class UserDao{
    public Integer getData(){
        return 0;
    }

    public void setUser(User user){
        user.setAge(12);
    }
}

class User{

    String name;
    Integer age;

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public Integer getAge() {
        return age;
    }

    public void setAge(Integer age) {
        this.age = age;
    }
}

  我们在UserDao中只设置了age,但是对age为空的断言却成功了。这表示原逻辑没有执行。再看看对Name的不为空断言,也通过了。这表示对setUser的行为被doAnwser()控制了。doAnwser()需要传入对一个Anwser对象,这个对象有点和代理对象类似。如果调用方法有返回值,则anwer()的返回值就是它。invocationOnMock.getArgument(0)获取参数,然后就可以对参数操作啦啦啦啦。其他几个也可以试试,比如改成调用原逻辑,Mockito.doCallRealMethod().when(mockeDao).setUser(u)。哎呀呀,这时候user的age就有了而且还是18.重点:对于在测试;类中的方法不用设置预期,会调用原逻辑,但是对于依赖类的方法就要设置预期了,否则会对有返回值的仅返回类型默认值、无返回值的直接啥都不做走人。

  好了,下班了,先到这里,以后有新的接触,在添加进来,以上例子略显粗犷,如不慎入坑,请包涵!(说得好像有人会看似的!!!要真有人看,基于以上例子可灵活处理。)

免责申明:本站发布的内容(图片、视频和文字)以转载和分享为主,文章观点不代表本站立场,如涉及侵权请联系站长邮箱:xbc-online@qq.com进行反馈,一经查实,将立刻删除涉嫌侵权内容。