使用Mockito进行单元测试【2】—— stub 和 高级特性[转]
一篇中介绍了Mockito的基本信息,现在接着介绍Mockito强大的stub功能
2. Mockito使用实例
5. 对连续的调用进行不同的返回 (iterator-style stubbing)
还记得在实例2中说道当我们连续两次为同一个方法使用stub的时候,他只会使用最新的一次。但是在某一个方法中我们确实有很多的调用怎么办呢?mockito当然想到这一点了:
- when(mock.someMethod("some arg"))
- .thenThrow(new RuntimeException())
- .thenReturn("foo");
- //First call: throws runtime exception:
- mock.someMethod("some arg");
- //Second call: prints "foo"
- System.out.println(mock.someMethod("some arg"));
- //Any consecutive call: prints "foo" as well (last stubbing wins).
- System.out.println(mock.someMethod("some arg"));
当然我们也可以将第一句写的更简单一些:
- when(mock.someMethod("some arg"))
- .thenReturn("one", "two", "three");
参见网页例子10。
6. 使用回调进行stub【Stubbing with callbacks】
我们可以使用generic的Answer接口来让mock对象执行我们期望它执行的内容。比如我们可以查看调用方法的参数信息,并根据这个信息进行不同的处理,这可以使我们的stub变得十分的灵活。
- when(mock.someMethod(anyString())).thenAnswer(new Answer() {
- Object answer(InvocationOnMock invocation) {
- Object[] args = invocation.getArguments();
- Object mock = invocation.getMock();
- return "called with arguments: " + args;
- }
- });
- //Following prints "called with arguments: foo"
- System.out.println(mock.someMethod("foo"));
参见网页例子11。
7. 使用 doThrow()|doAnswer()|doNothing()|doReturn() 来 stub void方法
void方法也需要stub?呵呵,实际生活中我们应该只会用到doThrow来模拟这个void方法出错的情况吧,anyway,mockito提供了四个方法,发挥你的想象力吧:-)
- doThrow(new RuntimeException()).when(mockedList).clear();
- //following throws RuntimeException:
- mockedList.clear();
参见网页例子12。
8. 在真实的对象上进行spy
spy的意思是你可以修改某个真实对象的某些方法的行为特征,而不改变他的基本行为特征,这种策略的使用跟AOP有点类似。下面举一个例子来说明:
- List list = new LinkedList();
- List spy = spy(list);
- //optionally, you can stub out some methods:
- when(spy.size()).thenReturn(100);
- //using the spy calls <b>real</b> methods
- spy.add("one");
- spy.add("two");
- //prints "one" - the first element of a list
- System.out.println(spy.get(0));
- //size() method was stubbed - 100 is printed
- System.out.println(spy.size());
- //optionally, you can verify
- verify(spy).add("one");
- verify(spy).add("two");
可以看到spy保留了list的大部分功能,只是将它的size方法改写了。不过spy在使用的时候有很多地方需要注意,一不小心就会导致问题,所以不到万不得已还是不要用spy。下面介绍两个spy的陷阱:
【1】有时我们无法使用when的方式来spy,此时我们就需要使用doReturn|Answer|Throw() 等方式来进行spy了:
- List list = new LinkedList();
- List spy = spy(list);
- //Impossible: real method is called so spy.get(0) throws IndexOutOfBoundsException (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的时候实际已经调用了get(0)方法,这个时候将直接抛出异常,所以此时应该使用doReturn来进行spy
【2】spy实际上是对对象做了一个拷贝,就像上面的,如果我们直接看list这个对象,它实际上只执行了这样一句话List list = new LinkedList();
【3】 无法spy final方法。
参见网页例子13。
9. 其他高级特性
当说到高级特性,我们就是说那些基本用不到,但是当我们想用的时候就非常顺手的功能,比如:
# 改变没有stub方法的默认返回值 网页例子14
# 抓取参数进行assert验证 网页例子15
# 重置mock对象 网页例子16
由于浙西而使用机会比较小,这里就不详述了,更多的例子还是参考网页http://docs.mockito.googlecode.com/hg/latest/org/mockito/Mockito.html吧
3.小结
不知看了以上的内容,你是否觉得Mockito是使用mock进行单元测试的理想工具,如果是的话,就猛击
http://code.google.com/p/mockito/的主页下载并使用吧:-)
本文转自:http://qiuguo0205.iteye.com/blog/1456528