JAVA利用enum結合testng做數據驅動示例

　　數據驅動是做自動化測試中很重要的一部分，數據源的方案也是百花八門了，比如利用外部文件，直接在@DataProvider中寫死等等，我們今天介紹一下利用enum來做數據源，先來看一下enum的寫法：

public enum TestEnum {
	
	OK(200,"success"),
	FAIL(300,"fail");
	
	private int retCode;
	
	private String msg;
	
	private TestEnum(int retCode, String msg) {
		this.retCode = retCode;
		this.msg = msg;
	}

	public int getRetCode() {
		return retCode;
	}

	public void setRetCode(int retCode) {
		this.retCode = retCode;
	}

	public String getMsg() {
		return msg;
	}

	public void setMsg(String msg) {
		this.msg = msg;
	}

	@Override
	public String toString() {
		return retCode+" "+msg;
	}

	public static void main(String[] args) {
		TestEnum[] vs = TestEnum.values();
		System.out.println(vs[0].getMsg());
		System.out.println(TestEnum.OK.getRetCode());
		System.out.println(TestEnum.OK.getMsg());
		System.out.println(TestEnum.valueOf("OK").getMsg());
	}
	
}

 上述代碼復制后，可直接運行！

上述代碼可理解為有兩組數據，每一組里面有兩個數據，一個是retCode,一個是msg。現在把enum給加載到@DataProvider中去：

public class TestDataDemo {
     
	@DataProvider
    public Object[][] dataProvider(){
		TestEnum[] vs = TestEnum.values();
		Object[][] obj = new Object[vs.length][];
		for (int i = 0; i < vs.length; i++) {
			obj[i] = new Object[]{vs[i]};
		}
        return obj;
    }
     
    @Test(dataProvider="dataProvider")
    public void testDemo(TestEnum te){
        System.out.println(te.getRetCode());
        System.out.println(te.getMsg());
    }  
   
}

 上述的代碼也是可以直接運行的，運行完后，結果如下：

利用上述方式最大的好處是把數據本身與持久化的對象放在一個地方了，且調用時，也很方便。

其實，利用外部文件的話，也可以數據對象持久化，只是原來的示例中我放在了一個Map中，同樣可以把Map改為一個pojo對象，調用時也會很方便。

至於哪種方式好，大家自行的去體驗，自行的去實現吧！