Testng+DataProvider+自定义注解实现通用的数据读取

1、正常情况是需要有一个方法提供data provider，然后测试方法调用名字，而excel的路径和sheet表名写在调用的地方，

2、通过注解实现excel文件配置读取，用户无需关心读取过程，只需设置配置路径

3、下面是测试用例

public class Demo extends ExcelDataHeleper {

     //excel文件读取

    @Test(dataProvider="excel")

    @DataFile(path ="excel/addStudentTemplate.xls",sheet ="Sheet0")

    public void testmethod1(Map param){

              System.out.println(param.get("id")+"    "+param.get("phone"));

    }

   //csv文件读取

    @Test(dataProvider="csv")

    @DataFile(path ="csv/a.csv")

    public void testmethodCsv(Map param){
              System.out.println(param.get("id")+"    "+param.get("phone"));

    }

    //yaml文件读取
    @Test(dataProvider="yaml")

    @DataFile(path ="yaml/a.yaml",key="user")

    public void testmethodYaml(Map param){
              System.out.println(param.get("id")+"    "+param.get("phone"));

    }

}

dataProvider:标注读取的是excel文件

DataFile中的path:excel的相对路径，在res下，sheet：表格的sheet名

param：为hashMap，通过表格中的键读取值

DataFile类如下

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface DataFile {
    public String path();  //注解元素 为  id
    public String sheet() default "no description"; //设置默认值，
}

下面介绍继承的基类：ExcelDataHeleper

import org.apache.poi.hssf.usermodel.HSSFDataFormatter;
import org.apache.poi.openxml4j.exceptions.InvalidFormatException;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.ss.usermodel.WorkbookFactory;
import org.testng.annotations.DataProvider;
import java.io.*;
import java.lang.reflect.Method;
import java.util.*;
public class ExcelDataHeleper {

    @DataProvider(name = "excel")
    public Iterator dataMethod(Method m) {
        System.out.println(m.getName());
        DataFile d = m.getAnnotation(DataFile.class);
        System.out.println(d.path() + "   " + d.sheet());
        List