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

  • 2019 年 12 月 20 日
  • 笔记

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打桩,由以上两种方式都能实现。