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

Posted 2022-05-02 xinxin1994

背景

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

Demo

话不多说，直接上Demo。

1.准备Excel数据

 

Paste_Image.png

2. 新建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
来源：简书