Mockito单元测试

2023-02-19,

Mockito简介

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

Mockito的使用

    首先来一个最基本的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.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.重点:对于在测试;类中的方法不用设置预期,会调用原逻辑,但是对于依赖类的方法就要设置预期了,否则会对有返回值的仅返回类型默认值、无返回值的直接啥都不做走人。

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

Mockito单元测试的相关教程结束。

《Mockito单元测试.doc》

下载本文的Word格式文档,以方便收藏与打印。