[mockito使用教程]部分/内部方法
- 2019 年 12 月 20 日
- 笔记
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打桩,由以上两种方式都能实现。