一、前言
数据驱动是我们写自动化脚本非常常用的技术,而Testng中数据驱动常用的注解,如何读取excel完成数据驱动呢? 简单思路就是,读取excel数据,放入map并返回
![d1a367830e24d0c8c324f277b4282645.png](https://i-blog.csdnimg.cn/blog_migrate/8768d59c8f26ea7961e892de25571b69.png)
二、Demo
1、准备Excel数据
![0907231f71394bef7497b73cbc23b472.png](https://i-blog.csdnimg.cn/blog_migrate/0e45961260ca16faf850adcd8c8ed6ea.jpeg)
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;
}
it.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");
笔者创建了一个测试交流群,如果对软件测试、接口测试、自动化测试、面试经验交流感兴趣可以加测试交流群(文章开头图片),还会有同行一起技术交流