Java中用enum結合testng實現(xiàn)數(shù)據(jù)驅動的方法示例

更新時間：2017年01月17日 11:02:07 作者：張飛_

TestNG數(shù)據(jù)驅動提供的參數(shù)化讓我們在測試項目可以靈活根據(jù)需求建立不同的dataprovider來提供數(shù)據(jù)，而真正實現(xiàn)數(shù)據(jù)，頁面，測試彼此獨立而又有機結合的可能性。下面這篇文章主要給大家介紹了Java中用enum和testng做數(shù)據(jù)驅動的方法示例,需要的朋友可以參考借鑒。

前言

數(shù)據(jù)驅動測試是相同的測試腳本使用不同的測試數(shù)據(jù)執(zhí)行，測試數(shù)據(jù)和測試行為完全分離。數(shù)據(jù)驅動是做自動化測試中很重要的一部分，數(shù)據(jù)源的方案也是百花八門了，比如利用外部文件，直接在@DataProvider中寫死等等。

我們今天介紹一下利用enum來做數(shù)據(jù)源，先來看一下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());
 }
 
}

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

上述代碼可理解為有兩組數(shù)據(jù)，每一組里面有兩個數(shù)據(jù)，一個是retCode,一個是msg。

現(xiàn)在把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());
 } 
 
}

上述的代碼也是可以直接運行的，運行完后

結果如下：