Mockito实战 博客分类: 技术总结单元测试 Mockito实战
Mockito,测试框架,语法简单,功能强大!
静态、私有、构造等方法测试需要配合PowerMock,PowerMock有Mockito和EasyMock两个版本,语法相同,本文只介绍Mockito.
测试对像:ArticleManager
ArticleManager 类
publicclassArticleManager{void setDatabase(ArticleDatabase database){...}Stringget(String val){
database.get(val);}String add(String val){
database.set(val);}}
ArticleDatabase 类
publicclassArticleDatabase{Stringget(String val){...}String add(String val){...}}
注解
@Mock
创建Mock对像
@MockprivateArticleDatabase ds;
实例ds是一个虚拟对像,ds对像本身的所有方法返回值都为Null。
@InjectMocks
@InjectMocksprivateArticleManager articleManager;
将现有Mock对像(ds)注入到articleManager,支持方法包括constructor, setter, property 和 Spring的@Autowired !
@Captor
参数调用截取器
//定义@Captor,如果要获取多个参数, 需要定义多个@Captor@CaptorprivateArgumentCaptor<String> argumentString;
articleManager.add("content");
verify(ds, times(1)).add(argumentString.capture());//打印参数值
logger.debug(argumentString.getValue());
@Spy
Mock类的部分真实或定制后调用。没有定制的方法就调用真实的方法。
@SpyprivateArticleManager articleManager2;//定制方法,这个地方用`when`还会调用真实的方法后返回定制结果//见:http://docs.mockito.googlecode.com/hg/latest/org/mockito/Spy.html
doReturn("0").when(articleManager2).get(0);
articleManager2.get(0);
verify(articleManager2).get(0);//调用真实的方法
articleManager2.get(1)
verify(articleManager2).get(1);
@Spy
真实调用也可以用@Mock
对像thenCallRealMethod
方法调用:
when(articleManager2.get(2)).thenCallRealMethod();
Stubbing
特别制定Mock对像某方法的返回值、类型或异常等,默认是Null,
when(ds.get(0)).thenReturn("first");when(ds.get(1)).thenThrow(newRuntimeException());
verify
对某方法返回结果的校验,包括调用次数,返回值等
articleManager.add("one");
verify(ds).add("one");
参考:@Captor
模糊匹配
在Stubbing或verify里可用模糊匹配来适配多种情况的调用,Mockito在包org.mockito.Matchers下提供有anyString(), anyInt()或者直接anyObject()等大量模糊匹配函数,如果有需要还可自己写,实现(org.hamcrest.Matcher)。 例:
when(articleManager.get(anyString())).toReturn("element");System.out.println(articleManager.get("1"));
verify(articleManager).get(anyString());
在使用校验函数verify时,参数如果有模糊匹配则所有参数都要用模糊匹配,如:
//正确的写法:
verify(mock).articleManager(anyInt(), anyString(), eq("third argument"));//错误用法:
verify(mock).articleManager(anyInt(), anyString(),"third argument");
更多模糊匹配函数请参考:http://mockito.googlecode.com/svn/branches/1.8.5/javadoc/org/mockito/Matchers.html
1自动生成Mock类
在需要Mock的属性上标记@Mock注解,然后@RunWith(MockitoJUnitRunner.class)或者在setUp()方法中显示调用MockitoAnnotations.initMocks(this);生成Mock类即可。
2自动注入Mock类到被测试类
只要在被测试类上标记@InjectMocks,Mockito就会自动将标记@Mock、@Spy等注解的属性值注入到被测试类中。
@InjectMocks的问题是:如果被测试类是代理类,那么注入会失效。比如上面的UserService如果是事务或者其他AOP代理类,那么进入@Test方法时UserService中的DAO属性不会被Mock类替换。
3 Mock方法定制再也不用录制、播放了
Mockito的Mock方法定制可读性很强,而且也不需要像EasyMock那样录制播放,定制后就可以使用。
例如:
when(userDao.selectAll()).
thenReturn(Collections.<UserDomain>emptyList());
4有些方法想Mock定制,有些想调用真实方法
因为@Mock针对接口生成Mock类,所以我们是没法调用到真实的实现类的方法。可以使用@Spy注解标注属性,并且标注@Resource注解让Spring注入真实实现类,那么Mockito就会自动生成Spy类。
例如:
@InjectMocks
@Resource(name ="userService")
privateUserService userService;
@Spy
@Resource
privateUserDao userDao;
Spy类就可以满足我们的要求。如果一个方法定制了返回值或者异常,那么就会按照定制的方式被调用执行;如果一个方法没被定制,那么调用的就是真实类的方法。
如果我们定制了一个方法A后,再下一个测试方法中又想调用真实方法,那么只需在方法A被调用前,调用Mockito.reset(spyObject);就行了。
输出为:
0
0
Real Test1()!!! - 1
Real Test2()!!! - 2
Real Test1()!!! - 100
Real Test1()!!! - 1
Real Test2()!!! - 2
Real Test1()!!! - 104
要注意的是,对Spy对象的方法定制有时需要用另一种方法:
===============================================================================
Importantgotcha on spying real objects!
Sometimes it's impossible to usewhen(Object) for stubbing spies. Example:
List list = new LinkedList();
List spy = spy(list);
//Impossible: real method is called so spy.get(0) throwsIndexOutOfBoundsException (the list is yet empty)
when(spy.get(0)).thenReturn("foo");
//You have to use doReturn() for stubbing
doReturn("foo").when(spy).get(0);
===============================================================================
因为用when(spy.f1())会导致f1()方法被真正执行,所以就需要另一种写法。
参考资料
Mockito文档
http://mockito.googlecode.com/svn/branches/1.6/javadoc/org/mockito/Mockito.html