读取Excel使TestNG完成数据驱动

Testng中数据驱动常用的注解是 @DataProvider，但是这个方法必须返回一个Object[][]。
简单思路就是，读取excel数据，excel数据第一行作为map的Key其它行为值，放入map并返回。 @DataProvider 注解对应方法去获取调读取excel方法，拿到返回的Object[][]，其余都一样使用。
1、导入maven依赖

        <dependency>
            <groupId>org.apache.poi</groupId>
            <artifactId>poi-ooxml</artifactId>
            <version>3.15</version>
        </dependency>
        <dependency>
            <groupId>org.apache.poi</groupId>
            <artifactId>poi</artifactId>
            <version>3.15</version>
        </dependency>

2、读取Excel数据，并放入Map中返回

public class DateCenter {
    /**
     * @param file 读取某个excel文件
     * @return Object
     */
    public Object[][] testData(String file) {
        ArrayList<String> arrkey = new ArrayList<String>();
        Workbook workbook = getWorkbook(file);
        Sheet sheet = workbook.getSheetAt(0);
        // 获取总行数
        int rowTotalNum = sheet.getLastRowNum() + 1;
        // 总列数
        int columns = sheet.getRow(0).getPhysicalNumberOfCells();
        HashMap<String, String>[][] map = new HashMap[rowTotalNum - 1][1];
        // 对数组中所有元素hashmap进行初始化
        if (rowTotalNum > 1) {
            for (int i = 0; i < rowTotalNum - 1; i++) {
                map[i][0] = new HashMap();
            }
        } else {
            System.out.println("测试的Excel" + file + "中没有数据");
        }
        // 获得首行的列名，作为hashmap的key值
        for (int c = 0; c < columns; c++) {
            String cellvalue = CellUnit.getCellValue(sheet, 0, c);
            arrkey.add(cellvalue);
        }
        // 遍历所有的单元格的值添加到hashmap中
        for (int r = 1; r < rowTotalNum; r++) {
            for (int c = 0; c < columns; c++) {
                String cellvalue = CellUnit.getCellValue(sheet, r, c);
                map[r - 1][0].put(arrkey.get(c), cellvalue);
            }
        }
        return map;
    }
    /**
     * 创建 workbook
     *
     * @param filePath excel文件路径
     * @return Workbook 对象
     * @throws IOException
     */
    public static Workbook getWorkbook(String filePath) {
        Workbook wb = null;
        try {
            if (filePath.endsWith(".xls")) {
                File file = new File(filePath);
                InputStream is = new FileInputStream(file);
                wb = new HSSFWorkbook(is);
            } else if (filePath.endsWith(".xlsx") || filePath.endsWith(".xlsm")) {
                wb = new XSSFWorkbook(filePath);
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
        return wb;
    }
    /**
     * 单元格处理内部类
     */
    private static class CellUnit {
        /**
         * 通过sheet 行号和列返回值
         *
         * @param sheet   sheet name
         * @param rowNum  行号
         * @param cellNum 列号
         * @return
         */
        public static String getCellValue(Sheet sheet, int rowNum, int cellNum) {
            String value;
            Cell cell = sheet.getRow(rowNum).getCell(cellNum);
            if (cell != null) {
                // 有值时，转换成字符串
                value = CellUnit.getCellValue2(cell);
            } else {
                // 当单元格没有值时，设为空串
                value = "";
            }
            return value;
        }
        /**
         * 把不同类型的单元格转换成字符串，并返回
         *
         * @param cell cell
         * @return 单个单元格值
         */
        public static String getCellValue2(Cell cell) {
            String value = "";
            switch (cell.getCellTypeEnum()) {
                case STRING:
                    value = String.valueOf(cell.getRichStringCellValue());
                    return value;
                case NUMERIC:
                    // 处理数值类型自动转成科学计数法问题
                    DecimalFormat df = new DecimalFormat("0");
                    value = String.valueOf(df.format(cell.getNumericCellValue()));
                    return value;
                case BOOLEAN:
                    value = String.valueOf(cell.getBooleanCellValue());
                    return value;
                case FORMULA:
                    value = String.valueOf(cell.getCellFormula());
                    return value;
                case ERROR:
                    value = String.valueOf(cell.getErrorCellValue());
                    return value;
                case BLANK:
                    return value;
                default:
                    System.out.println("单元格异常");
                    return value;
            }
        }
    }
}

3、TestNG使用数据

    @DataProvider(name = "testData")
    public Object[][] data() {
        TestCaseExcel testcase = new TestCaseExcel();
        return testcase.testData(file);
    }
    @Test(dataProvider = "testData")
    public void testCase(HashMap<String, String> data) {
        String fileName = data.get("excelName");
        String bpSheetName = data.get("Benefits Package Sheet");
        int bpRowNum = Integer.parseInt(data.get("BP sheet RowNum"));
        String csvSheetName = data.get("Cost Share Variances Sheet");
        int csvRowNum = Integer.parseInt(data.get("CSV Sheet RowNum"));
        String hiosPlanID = data.get("HIOS Plan ID");
        String isPass = data.get("isPass");