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()); }  }

上述的代码也是可以直接运行的，运行完后

结果如下：