局部Mock

来源:互联网 发布:动漫模板源码 编辑:程序博客网 时间:2024/06/05 05:35
说明
有时候我们不能mock一个类中的所有方法,而是要真实调用某些方法。这时候就可以使用PowerMockito.spy()来进行mock.
需要注意的是,使用spy来录制行为的时候使用when(...).thenXxx(..)的模式会直接调用原有方法,可能会和预期的结果不符合。正确的使用方式是doXXX(...).when(...).methodToMock(...)。看如下例子:
import org.junit.Test;
import org.powermock.api.mockito.PowerMockito;
 
import java.util.ArrayList;
import java.util.List;
 
public class SpyTest {
 
    @Test
    public void testSpy() {
        List list =  new ArrayList();
        List** = PowerMockito.spy(list);
        // 会抛出IndexOutOfBoundsException异常,因为会去调用spy.get(0)方法。
        PowerMockito.when(spy.get(0)).thenReturn("foo");
 
        // 正确的写法如下
        PowerMockito.doReturn("foo").when(spy).get(0);
    }
}
 
该示例中对List的get方法进行spy操作,并录制行为,但是在执行PowerMockito.when(spy.get(0)).thenReturn("foo");的时候回抛出IndexOutOfBoundsException异常,原因是录制的时候会调用spy.get(0)方法。如何在录制的过程中避免抛出异常呢?正如上面例子中看到的,使用PowerMockito.doReturn("foo").when(spy).get(0);即可。
原创粉丝点击