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對象，調用時也會很方便。

至於哪一種方式好，你們自行的去體驗，自行的去實現吧！