java分享第十八天（ 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对象，调用时也会很方便。 至于哪一种方式好，你们自行的去体验，自行的去实现吧！