使用 spy() 監視真實的對象
在使用 when(foo.call()).thenReturn("foo"); 的時候,事實上并沒有真的調用 foo.Call(),而是在運行測試的時候,在調用 foo.call() 的時候都給你返回 foo 這個返回值:當然這個前提是 foo 是一個 Mock 對象(意味著加了 @Mock 注解)。
如果要調用一個真實的對象的方法,同時又要控制它的返回值,那么可以這樣做:
doReturn("foo").when(foo).call();
當然,這個 foo 需要是一個 spy 對象:
foo=spy(fooBase);
拿官方的例子做栗子:
List list = new LinkedList();
List spy = spy(list);
// Right
doReturn("foo").when(spy).get(0); 【0】
// Wrong
when(spy.get(0)).thenReturn("foo"); 【1】
此處,spy 是一個監視 list 的對象。
【0】當調用 spy.get(0) 的時候,并不是真的去調用這個方法,而是直接返回 foo。 【1】當調用 spy.get(0) 的時候,會直接調用 list.get(0),因為它是一個真實的對象,然后因為這個 list 的 size()==0 因此會拋出一個數組越界的異常。