easymock教程-運行時返回值或者異常

來至:http://skydream.iteye.com/blog/834158

?

?

??????? 前面的教程中，我們看到easymock可以通過expect方法來設定mock方法的返回值或者異常，但是注意這些案例中設置的返回值都是在調用被測試 的類的方法前就已經確定下來的，即我們其實在測試類的代碼運行前(實際是在EasyMock.replay()方法調用前)就已經"預知"了返回結果。

?

??? 但是在某些情況下，我們可能無法預知返回值，比如我們需要根據輸入的參數值來決定返回什么，而這個參數可能無法在record階段獲得。因此在mock方法中我們無法在record階段就決定應該返回什么。

?

??? 對于這種場景，easymock提供了IAnswer接口和andAnswer()方法來提供運行時決定返回值或者異常的機制。

?

??? 我們來看一個簡單的例子：

???? public ? class ?Business? {
???????? private ?Service?service;

???????? public ? void ?execute()? {
???????????? int ?count? = ?ramdonInt();
???????????? int ?result? = ?service.execute(count);
????????}

???????? public ? void ?setService(Service?service)? {
???????????? this .service? = ?service;
????????}

???????? private ? int ?ramdonInt()? {
????????????Random?random? = ? new ?Random();
???????????? return ?random.nextInt()? / ? 10000 ;
????????}
????}

???? public ? interface ?Service? {
???????? public ? int ?execute( int ?count);
????}

??? 在Business的execute()方法中，需要調用service.execute(int count)方法，而傳入的參數count是需要運行時才能確定的，這里為了簡單我們random了一個int來模擬這種情況。

?

??? 然后看測試案例

????@Test
???? public ? void ?testRuntimeReturn()? {
????????Business?business? = ? new ?Business();
????????Service?service? = ?EasyMock.createMock(Service. class );
????????business.setService(service);
????????EasyMock.expect(service.execute(EasyMock.anyInt())).andAnswer( new ?IAnswer < Integer > ()? {
???????????? public ?Integer?answer()? throws ?Throwable? {
????????????????Integer?count? = ?(Integer)?EasyMock.getCurrentArguments()[ 0 ];
???????????????? return ?count? * ? 2 ;
????????????}
????????} );

????????EasyMock.replay(service);
????????business.execute();
????????EasyMock.verify(service);
????}

??? 這里我們通過EasyMock.expect(service.execute(EasyMock.anyInt()))來接受任意值的count參數輸 入，andAnswer(new IAnswer<Integer>() {}) 讓我們可以指定一個IAnswer的實現類來給出返回值。在這個IAnswer的實現類中，我們通過 EasyMock.getCurrentArguments()[0]獲取到service.execute()方法的第一個參數，然后簡單的運用 count*2規則給出返回值。這里的EasyMock.getCurrentArguments()方法可以獲取到運行時的參數列表，不過注意這個方法 對重構不夠友好，如果參數列表發生變化則必須手工修改對象的獲取參數的代碼。

?

??? 下面是一個運行時拋出異常的例子，簡單起見我們通過設置exception的message來在錯誤信息中傳遞運行時的count值。

???@Test
???? public ? void ?testRuntimeException()? {
????????Business?business? = ? new ?Business();
????????Service?service? = ?EasyMock.createMock(Service. class );
????????business.setService(service);
????????EasyMock.expect(service.execute(EasyMock.anyInt())).andAnswer( new ?IAnswer < Integer > ()? {
???????????? public ?Integer?answer()? throws ?Throwable? {
????????????????Integer?count? = ?(Integer)?EasyMock.getCurrentArguments()[ 0 ];
???????????????? throw ? new ?RuntimeException( " count= " ? + ?count);
????????????}
????????} );

????????EasyMock.replay(service);
???????? try ? {
????????????business.execute();
????????????fail( " should?throw?RuntimeException " );
????????} ? catch ?(RuntimeException?e)? {
????????????assertTrue(e.getMessage().indexOf( " count= " )? != ? - 1 );
???????????? // get?count?from?message
????????????EasyMock.verify(service);
????????}
????}

??? 除了IAnswer接口外，easymock中還有另外一個方式可以完成類似的功能，就是使用andDelegate()方法，

???? public ? class ?ServiceStub? implements ?Service? {
???????? public ? int ?execute( int ?count)? {
???????????? return ?count? * ? 2 ;
????????}
????}

????@Test
???? public ? void ?testRuntimeReturn()? {
????????Business?business? = ? new ?Business();
????????Service?service? = ?EasyMock.createMock(Service. class );
????????business.setService(service);
????????EasyMock.expect(service.execute(EasyMock.anyInt())).andDelegateTo( new ?ServiceStub());

????????EasyMock.replay(service);
????????business.execute();
????????EasyMock.verify(service);
????}

??? 這里需要先創建一個Service類的實現類和一個實例，然后通過andDelegateTo()將這個stub的實例傳進去，注意這里delegate進去的實例必須是mock對象接口相同。

?

??? delegateTo方式實際上是我們手工創建了stub(mock和stub的概念及差別請參考本教程的"mock和stub"一文)，這和我們使用 easymock的初衷有所違背。而且當這個接口有眾多方法時，創建這樣一個stub會顯得很痛苦，不如使用IAnswer方便直接。

EasyMock教程-運行時返回值或者異常 文章分類:軟件開發管理