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.重点:对于在测试;类中的方法不用设置预期,会调用原逻辑,但是对于依赖类的方法就要设置预期了,否则会对有返回值的仅返回类型默认值、无返回值的直接啥都不做走人。
好了,下班了,先到这里,以后有新的接触,在添加进来,以上例子略显粗犷,如不慎入坑,请包涵!(说得好像有人会看似的!!!要真有人看,基于以上例子可灵活处理。)
原文地址:https://www.cnblogs.com/guozhigang/p/9884969.html