Java中用enum结合testng实现数据驱动的方法示例

Z时代
2024-01-10
分类：IT

前言

数据驱动测试是相同的测试脚本使用不同的测试数据执行，测试数据和测试行为完全分离。数据驱动是做自动化测试中很重要的一部分，数据源的方案也是百花八门了，比如利用外部文件，直接在@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());
 } 
}