读书人

学习Mockito - Mock对象的重置跟Answe

发布时间: 2012-12-25 16:18:28 作者: rapoo

学习Mockito - Mock对象的重置和Answer接口(方法预期回调接口)的应用

Mock对象的重置

?

Mockito提供了reset(mock1,mock2……)方法,用来重置mock对象。当mock对象被重置后,它将回到刚创建完的状态,没有任何stubbing和方法调用。这个特性平时是很少用到的,因为我们大都为每个test方法创建mock,所以没有必要对它进行重置。

?

官方提供这个特性的唯一目的是使得我们能在有容器注入的mock对象中工作更为方便。所以,当决定要使用这个方法的时候,首先应该考虑一下我们的测试代码是否简洁和专注,测试方法是否已经超长了。

?

?

Answer接口(方法预期回调接口)的应用

?

Answer接口说明

对mock对象的方法进行调用预期的设定,可以通过thenReturn()来指定返回值,thenThrow()指定返回时所抛异常,通常来说这两个方法足以应对一般的需求。但有时我们需要自定义方法执行的返回结果,Answer接口就是满足这样的需求而存在的。另外,创建mock对象的时候所调用的方法也可以传入Answer的实例mock(java.lang.Class<T> classToMock, Answer defaultAnswer),它可以用来处理那些mock对象没有stubbing的方法的返回值。

?

InvocationOnMock对象的方法

Answer接口定义了参数为InvocationOnMock对象的answer方法,利用InvocationOnMock提供的方法可以获取mock方法的调用信息。下面是它提供的方法:

getArguments() 调用后会以Object数组的方式返回mock方法调用的参数。

getMethod() 返回java.lang.reflect.Method 对象

getMock() 返回mock对象

callRealMethod() 真实方法调用,如果mock的是接口它将会抛出异常

?

通过一个例子来看一下Answer的使用。我们自定义CustomAnswer类,它实现了Answer接口,返回值为String类型。

@Testpublic void customAnswerTest(){    List<String> mock = mock(List.class);    when(mock.get(4)).thenAnswer(new Answer(){        public String answer(InvocationOnMock invocation) throws Throwable {            Object[] args = invocation.getArguments();            Integer num = (Integer)args[0];            if( num>3 ){                return "yes";            } else {                throw new RuntimeException();            }        }    });    System.out.println(mock.get(4));}
?

?

?

读书人网 >编程

热点推荐