Mockito單元測試框架學習
一、問題:如何將mock的類自動注入到待測類,特別是在沒有setter方法的情況下。
解答:
前提:待測的service類及其依賴的其他類都是處在被spring管理中的。
做法:在測試類中,只要將待測的類標注為@InjectMocks,將其依賴的其他類標注為 @Mock,
就可以使用MockitoAnnotations.initMocks(this);這句話自動將依賴的類注入待測類,如果依賴類在spring的管理下有自己的name,那么甚至在待測類中都不需要寫setter方法。
例:
1、待測類
@Component("abcService") public class AbcService { @Resource(name="aaaDao") private AaaDao aaaDao; @Resource(name="bbbDao") private BbbDao bbbDao; ......//注:此處省略的代碼中并不包含aaaDao和bbbDao的setter方法。 } |
2、測試類
public class AbcServiceTest{ @InjectMocks AbcService abcService; @Mock AaaDao aaaDao; @Mock BbbDao bbbDao; @Before public void setup(){ MockitoAnnotations.initMocks(this);//這句話執行以后,aaaDao和bbbDao自動注入到abcService中。 //在這之后,你就可以放心大膽地使用when().then()等進行更詳細的設置。 } } |
二、問題:如何對連續的調用進行不同的返回
對連續的調用進行不同的返回 (iterator-style stubbing)
還記得在實例2中說道當我們連續兩次為同一個方法使用stub的時候,他只會使用最新的一次。但是在某一個方法中我們確實有很多的調用怎么辦呢?mockito當然想到這一點了:
when(mock.someMethod("some arg")) .thenThrow(new RuntimeException()) .thenReturn("foo"); //First call: throws runtime exception: mock.someMethod("some arg"); //Second call: prints "foo" System.out.println(mock.someMethod("some arg")); //Any consecutive call: prints "foo" as well (last stubbing wins). System.out.println(mock.someMethod("some arg")); |
當然我們也可以將第一句寫的更簡單一些:
when(mock.someMethod("some arg"))
.thenReturn("one", "two", "three");
posted on 2014-11-03 09:10 順其自然EVO 閱讀(309) 評論(0) 編輯 收藏 所屬分類: 測試學習專欄