背景

数据驱动是我们写自动化脚本非常常用的技术，而Testng中数据驱动常用的注解是 @DataProvider，但是这个方法必须返回一个Object[][]。最近常有学生问起，如果通过外部文件作为数据源，从而实现数据驱动。例如数据源为Excel时，如何读取excel完成数据驱动呢？简单思路就是，读取excel数据，excel数据第一行作为map的Key其它行为值，放入map并返回。 @DataProvider 注解对应方法去获取调读取excel方法，拿到返回的Object[][]，其余都一样使用。
ps.代码是不值钱的，关键是解决问题的思路。

Demo

话不多说，直接上Demo。

1.准备Excel数据

Paste_Image.png

新建maven工程，并导入 poi相关包：

  <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>

3.读取Excel数据，并放入Map中返回。

 /**

     * @param file 读取某个excel文件

     * @return Object

     */

    public Object[][] testData(String file) {

        ArrayList<String> arrkey = new ArrayList<String>();

        Workbook workbook = WorkBookEhi.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 {

            log.error("测试的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;

    }

当然这个这里面的getWorkbook()方法我也是经过封装了，代码如下：

  /**

     * 创建 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;

    }

CellUnit.getCellValue()方法封装如下：

  /**

     * 通过sheet 行号和列返回值

     *

     * @param sheet   sheet name

     * @param rowNum  行号

     * @param cellNum 列号

     * @return

     */

    public static String getCellValue(Sheet sheet, int rowNum, int cellNum) {

        Cell cell = sheet.getRow(rowNum).getCell(cellNum);

        String value = CellUnit.getCellValue(cell);

        return value;

    }

    CellUnit.getCellValue() 方法封装如下：

     /**

     * 把不同类型的单元格转换成字符串，并返回

     *

     * @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:

                value = String.valueOf(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:

                log.warn("未知该单元格类型");

                return value;

        }

    }

4.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");

作者：博客已迁移I米阳
链接：https://www.jianshu.com/p/895e4c118db9
来源：简书

巴特西

读取Excel，通过Testng完成数据驱动

背景

Demo

最新文章

热门文章