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

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

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