Java中用enum結合testng實現數據驅動的方法示例

2019-11-26 13:14:39

字體：大中小

來源：轉載

供稿：網友

前言

數據驅動測試是相同的測試腳本使用不同的測試數據執行，測試數據和測試行為完全分離。數據驅動是做自動化測試中很重要的一部分，數據源的方案也是百花八門了，比如利用外部文件，直接在@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()); }  }

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

結果如下：