数据驱动测试之—— Excel+TestNG

对于利用Webdriver做自动化的童鞋，对于如何将元素或者输入数据如何和编码分离都应该不会太陌生，本着一边学习一边分享的心态，大概总结了一下关于利用CSV、XML以及Excel来存放数据，然后在结合TestNG来执行数据驱动测试。

需要的Jar包文件：

poi-3.11-20141221.jar

poi-ooxml-3.11-20141221.jar

poi-ooxml-schemas-3.11-20141221.jar

xmlbeans-2.6.0.jar

下面是关于Excel+TestNG进行数据驱动的代码：

package com.util.datadriver;

import java.io.File;

import java.io.FileInputStream;

import java.util.ArrayList;

import java.util.List;

import org.apache.poi.hssf.usermodel.HSSFWorkbook;

import org.apache.poi.ss.usermodel.Row;

import org.apache.poi.ss.usermodel.Sheet;

import org.apache.poi.ss.usermodel.Workbook;

import org.apache.poi.xssf.usermodel.XSSFWorkbook;

import org.testng.annotations.DataProvider;

import org.testng.annotations.Test;

/**

 *

 * 利用TestNG和Excel（poi）实现数据的驱动

 *

 * 需要导入额jar包： poi-3.11-20141221.jar、 poi-ooxml-3.11-20141221.jar、

 *         poi-ooxml-schemas-3.11-20141221.jar、 xmlbeans-2.6.0.jar

 */

public class DataProviderExcel {

    @DataProvider(name = "excelData")

    public static Object[][] excelData() throws Exception {

        return getExcelData("date", "excelData.xlsx", "testSheet1");

    }

    @Test(dataProvider = "excelData")

    public void testExcelData(String input1, String input2, String input3) {

        System.out.println(input1);

        System.out.println(input2);

        System.out.println(input3);

    }

    @SuppressWarnings("resource")

    public static Object[][] getExcelData(String filePath, String fileName,

            String sheetName) throws Exception {

        Workbook workbook = null;

        File file = new File(filePath + "/" + fileName);

        FileInputStream fis = new FileInputStream(file);

        String type = fileName.substring(fileName.indexOf("."));

        // Excel2007 以后的格式为.xlsx的Excel文件，需要调用XSSFWorkbook

        // Excel2007以前版本的格式为.xls的Excel文件，需要调用HSSFWorkbook

        if (type.equals(".xlsx")) {

            workbook = new XSSFWorkbook(fis);

        } else if (type.equals(".xls")) {

            workbook = new HSSFWorkbook(fis);

        }

        Sheet sheet = workbook.getSheet(sheetName);

        // 获得最后一行的行数

        int lastRowCount = sheet.getLastRowNum();

        // 获得第一行的行数

        int firstRowCount = sheet.getFirstRowNum();

        // Excel行数从第0行开始

        int sumRowCount = lastRowCount - firstRowCount + 1;

        List<Object[]> list = new ArrayList<Object[]>();

        // 获取每行的行对象,第一行为信息栏，不计入，所以从1开始

        for (int i = 1; i < sumRowCount; i++) {

            Row row = sheet.getRow(i);

            // 获得一行中最后单元格的count

            int lastCellCount = row.getLastCellNum();

            // 定义一个数组来存放cell中值，根据cell的长度来定义数组的长度

            String[] fileds = new String[lastCellCount];

            for (int j = 0; j < lastCellCount; j++) {

                String cellValue = row.getCell(j).getStringCellValue();

                fileds[j] = cellValue;

            }

            list.add(fileds);

        }

        fis.close();

        // 定义一个object[][] 的二维数组，存放list中的值

        Object[][] results = new Object[list.size()][];

        // 设置二维数组每行的值，

        for (int a = 0; a < list.size(); a++) {

            results[a] = list.get(a);

        }

        return results;

    }

}

秒客网

数据驱动测试之—— Excel+TestNG

相关文章