[mockito使用教程]部分/内部方法

mockito有两种API实现部分/内部方法mock，分别是spy与callRealMethod()。

spy

spy会创建一个真实的对象，对象的方法都会被调用，除非你将某个方法打桩(stage)，这个方法才不执行，走mock数据，下面是例子。

List list = new LinkedList();
List spy = spy(list);

//对size打桩:
when(spy.size()).thenReturn(100);

//执行真实的方法
spy.add("one");
spy.add("two");

//打印 "one"
System.out.println(spy.get(0));

//size() 被打桩 - 打印100
System.out.println(spy.size());

//可以做以下校验
verify(spy).add("one");
verify(spy).add("two");

注意when(spy.size()).thenReturn(100);这个打桩方法会执行这段代码，如果程序不要执行这个方法，直接打桩，可以使用doReturn(100).when(spy).size()。

List list = new LinkedList();
List spy = spy(list);

//会执行spy.get(0)方法导致 throws IndexOutOfBoundsException
when(spy.get(0)).thenReturn("foo");

//采用 doReturn() 打桩
doReturn("foo").when(spy).get(0);

doCallRealMethod

对于mock的对象，它的所有方法都会被打桩，不执行真实方法，除非使用doCallRealMethod。例子

Foo mock = mock(Foo.class);
doCallRealMethod().when(mock).someVoidMethod();

// 会执行实现方法 Foo.someVoidMethod()
mock.someVoidMethod();

内部方法

如果想mock一个类的内部方法，实际就是该类某些方法执行真实逻辑，某些方法执行mock打桩，由以上两种方式都能实现。