跳到主要內容

不同的Expected Result

Problem

為了讓程式夠強健,在發生能夠處理的例外情況可能會做retry。在搞笑談軟工有提到retry好的做法,也可以透過spring的RetryTemplate去實做retry(很大包..)。
但在測試的時候,要怎麼確認你的retry是否可以work呢? 讓我們來看看範例。這個範例在執行時,如果有例外產生的時候,會使用替代方案執行,如果超過重試次數就會拋出一個RuntimeException。
package org.tonylin.powermock;
 
public class RetryExample {
	public User readUser(){
		int attempt = 0;
		int maxAttempt = 2;
		boolean retry = false;
		User user = null;
		do {
			try {
				retry = false;
				if (attempt == 0)
					user = DaoUtil.readUserFromDB();
				else
					user = DaoUtil.readUserFromFile();
			} catch (Exception e) {
				attempt++;
				retry = true;
				if (attempt > maxAttempt)
					throw new RuntimeException(e);
			}
		} while (attempt <= maxAttempt && retry);
		return user;
	}
}

測試案例可以這樣設計:

  • 第一次執行成功。
  • 第一次失敗,改由readUserFromFile執行成功。
  • 第一次執行失敗,第二次執行readUserFromFile失敗,第三次才成功。
  • 執行皆失敗。

該如何透過PowerMock去實做這些測試案例呢?

How to resolve?

第一次執行成功這個案例較基本,我就不特別說明。

宣告

首先我們要先宣告PowerMock相關的annotation:

@RunWith(PowerMockRunner.class)
@PrepareForTest({RetryExample.class, DaoUtil.class})
public class TestRetryExample {
 
}

第一次失敗,改由readUserFromFile執行成功

這個案例不難實做,首先讓readUserFromDB拋exception,readUserFromFile回傳User物件。需要驗證的內容,包含取得的物件必須與我們建立的mock物件吻合,還要確認程式先執行readUserFromDB再執行readUserFromFile。因此我在mock DaoUtil時,使用了mockStaticStrict,這會讓PowerMock.verifyAll()驗證流程必須依照我們所宣告的順序。

@Test
public void testReadUser_failureAtFirstTime(){
	User mockUser = PowerMock.createMock(User.class);
	PowerMock.mockStaticStrict(DaoUtil.class);
 
	DaoUtil.readUserFromDB();
	PowerMock.expectLastCall().andThrow(new RuntimeException()).once();
 
	DaoUtil.readUserFromFile();
	PowerMock.expectLastCall().andReturn(mockUser).once();
 
	PowerMock.replayAll();
 
	RetryExample retryExample = new RetryExample();
	User user = retryExample.readUser();
	assertEquals(mockUser, user);
	PowerMock.verifyAll();
}

執行皆失敗

即所有的請求都拋exception。在驗證的部分要去catch exception,如果沒發生exception就代表有問題。

@Test
public void testReadUser_failure3Times(){
	PowerMock.mockStaticStrict(DaoUtil.class);
 
	DaoUtil.readUserFromDB();
	PowerMock.expectLastCall().andThrow(new RuntimeException()).once();
 
	DaoUtil.readUserFromFile();
	PowerMock.expectLastCall().andThrow(new RuntimeException()).times(2);
 
	PowerMock.replayAll();
 
	RetryExample retryExample = new RetryExample();
 
	try {
		retryExample.readUser();
		fail();
	} catch( Exception e ){
		PowerMock.verifyAll();	
	}
}

第一次執行失敗,第二次執行readUserFromFile失敗,第三次才成功

這才是這篇的重點。為了能模擬同一個method會有不同的執行結果,我使用了PowerMock.expectLastCall().andAnswer()並且override answer()。接著透過一個flag去記錄是第幾次執行以決定要回傳的結果。

@Test
public void testReadUser_failure2Times(){
	final User mockUser = PowerMock.createMock(User.class);
	PowerMock.mockStaticStrict(DaoUtil.class);
 
	DaoUtil.readUserFromDB();
	PowerMock.expectLastCall().andThrow(new RuntimeException()).once();
 
	DaoUtil.readUserFromFile();
	PowerMock.expectLastCall().andAnswer(new IAnswer<User>() {
		private int mTimes = 0;
		@Override
		public User answer() throws Throwable {
			mTimes++;
			if( mTimes == 1 ){
				throw new RuntimeException();
			}
			return mockUser;
		}
	}).times(2);
 
	PowerMock.replayAll();
 
	RetryExample retryExample = new RetryExample();
	User user = retryExample.readUser();
	assertEquals(mockUser, user);
	PowerMock.verifyAll();
}

IAnswer其實還能取得輸入參數,好去回傳不同的結果,這部分等之後有機會再分享給大家。

留言

這個網誌中的熱門文章

PostgreSQL - Unattended installation on windows

Introduction 要將別人軟體包裝到自己軟體中,不可或缺的東西就是Unattended installation。以Unattended installation來說,我們可以選擇透過Installer的silent mode安裝,也可以透過把目標軟體做成portable的版本。本篇文章分享這兩種方法,教導大家如何將PostgreSQL透過Unattended installation方式安裝到目標系統成為service。 Note. 本篇以PostgreSQL 10.7為例。 Install with installer Tips 安裝程式或反安裝程式的參數,除了可以直接上官網搜尋Installation User Guide以外,也可以直接使用help參數查詢: postgresql- 10.7 - 2 -windows-x64.exe --help Windows安裝程式主要有EnterpriseDB與BigSQL兩種。BigSQL版本安裝元件是透過網路下載且支援參數不如EnterpriseDB版本多,以我們需求來說,我們傾向於使用EnterpriseDB版本。接下來分享給大家安裝與反安裝方法。 Installation @ echo off set INSTALL_DIR =C:\postgres10 set INSTALLER =postgresql- 10.7 - 2 -windows-x64.exe   rem options for installation set SSMDB_SERVICE =postgresql- 10 set MODE =--unattendedmodeui none --mode unattended   set DB_PASSWD =--superpassword postgres set DB_PORT =--serverport 5432   set SERVICE_NAME =--servicename % SSMDB_SERVICE %   set PREFIX =--prefix "%INSTALL_DIR%" set DATA_DIR =--datadir "%INSTALL_DIR%\data"   set OPTIONS =

How to install RIDE on Windows?

Introduction 多年沒在Windows上開發RobotFramework,趁著這次整理一下RIDE安裝方法。 目前RIDE最新版本與Python對應版本如下: (3.6 < python <= 3.11) Install current released version (2.0.8.1) with: pip install -U robotframework-ride 安裝Python 直接到Python官網找尋最新的3.11版本,我使用3.11.9: link 。安裝就是一直下一步而已。 安裝wxPython 每次安裝RIDE最困難的都是wxPython。看了一下 官網 描述,我就姑且相信一下: 接著進入下 載頁面 就有安裝教學。基本上就是到Python目錄下的Scripts直接執行以下command: pip install -U wxPython 安裝RIDE 接著就如RIDE官網所說,執行以下command: pip install -U robotframework-ride 啟動RIDE 直接在相同目錄下執行ride就可以啟動了,你也可以直接在桌面建ride連結,加快下次啟動時間。 沒想到這次這麼順利就安裝完成了。因為我是使用java去啟動robot framework,就不特別講要怎麼使用pip安裝robot framework了。

Hello World!

即將要搬家,因此舊網頁內容將慢慢轉移至Blogger。 如果要存取舊網頁,可以使用以下連結: https://wiki.tonylin.idv.tw/dokuwiki/doku.php